style: fix ruff lint and format issues in integration tests

Author: gk16

tests/integration/conftest.py

@@ -55,14 +55,17 @@ def pytest_configure(config: pytest.Config) -> None:
 
 from agentanycast._generated.agentanycast.v1 import (  # noqa: E402
     a2a_models_pb2 as a2a_pb2,
-    agent_card_pb2 as card_pb2,
+)
+from agentanycast._generated.agentanycast.v1 import (  # noqa: E402
     node_service_pb2 as ns_pb2,
+)
+from agentanycast._generated.agentanycast.v1 import (  # noqa: E402
     node_service_pb2_grpc as ns_grpc,
-    registry_service_pb2 as reg_pb2,
+)
+from agentanycast._generated.agentanycast.v1 import (  # noqa: E402
     registry_service_pb2_grpc as reg_grpc,
 )
 
-
 # ── Docker Compose gRPC fixtures ────────────────────────────────────
 
 
@@ -150,43 +153,59 @@ def _sub():
 
 def send_task_to_peer(sender_client, peer_id: str, text: str = "hello") -> str:
     """Send a text task from sender to peer, return task_id."""
-    resp = sender_client.SendTask(ns_pb2.SendTaskRequest(
-        peer_id=peer_id,
-        message=a2a_pb2.Message(
-            role=a2a_pb2.MESSAGE_ROLE_USER,
-            parts=[a2a_pb2.Part(text_part=a2a_pb2.TextPart(text=text))],
-        ),
-    ))
+    resp = sender_client.SendTask(
+        ns_pb2.SendTaskRequest(
+            peer_id=peer_id,
+            message=a2a_pb2.Message(
+                role=a2a_pb2.MESSAGE_ROLE_USER,
+                parts=[a2a_pb2.Part(text_part=a2a_pb2.TextPart(text=text))],
+            ),
+        )
+    )
     return resp.task.task_id
 
 
 def complete_task(client, task_id: str, result_text: str = "done") -> None:
     """Transition task to WORKING then COMPLETED."""
-    client.UpdateTaskStatus(ns_pb2.UpdateTaskStatusRequest(
-        task_id=task_id, status=a2a_pb2.TASK_STATUS_WORKING,
-    ))
-    client.CompleteTask(ns_pb2.CompleteTaskRequest(
-        task_id=task_id,
-        artifacts=[a2a_pb2.Artifact(
-            artifact_id=unique_id("art"),
-            name="result",
-            parts=[a2a_pb2.Part(text_part=a2a_pb2.TextPart(text=result_text))],
-        )],
-    ))
+    client.UpdateTaskStatus(
+        ns_pb2.UpdateTaskStatusRequest(
+            task_id=task_id,
+            status=a2a_pb2.TASK_STATUS_WORKING,
+        )
+    )
+    client.CompleteTask(
+        ns_pb2.CompleteTaskRequest(
+            task_id=task_id,
+            artifacts=[
+                a2a_pb2.Artifact(
+                    artifact_id=unique_id("art"),
+                    name="result",
+                    parts=[a2a_pb2.Part(text_part=a2a_pb2.TextPart(text=result_text))],
+                )
+            ],
+        )
+    )
 
 
 def fail_task(client, task_id: str, error_message: str = "test error") -> None:
     """Transition task to WORKING then FAILED."""
-    client.UpdateTaskStatus(ns_pb2.UpdateTaskStatusRequest(
-        task_id=task_id, status=a2a_pb2.TASK_STATUS_WORKING,
-    ))
-    client.FailTask(ns_pb2.FailTaskRequest(
-        task_id=task_id, error_message=error_message,
-    ))
+    client.UpdateTaskStatus(
+        ns_pb2.UpdateTaskStatusRequest(
+            task_id=task_id,
+            status=a2a_pb2.TASK_STATUS_WORKING,
+        )
+    )
+    client.FailTask(
+        ns_pb2.FailTaskRequest(
+            task_id=task_id,
+            error_message=error_message,
+        )
+    )
 
 
 def wait_task_status(client, task_id: str, status, timeout: float = 15):
     """Poll until task reaches the given status."""
+
     def _check():
         resp = client.GetTask(ns_pb2.GetTaskRequest(task_id=task_id))
         return resp.task.status == status
@@ -207,10 +226,12 @@ def connect_nodes(client_a, client_b):
     peer_id_b = resp_b.node_info.peer_id
     addrs_b = list(resp_b.node_info.listen_addresses)
 
-    client_a.ConnectPeer(ns_pb2.ConnectPeerRequest(
-        peer_id=peer_id_b,
-        addresses=addrs_b,
-    ))
+    client_a.ConnectPeer(
+        ns_pb2.ConnectPeerRequest(
+            peer_id=peer_id_b,
+            addresses=addrs_b,
+        )
+    )
     return peer_id_a, peer_id_b
 
 

tests/integration/test_anycast.py

@@ -27,10 +27,12 @@
 
 from agentanycast._generated.agentanycast.v1 import (  # noqa: E402
     node_service_pb2 as ns_pb2,
+)
+from agentanycast._generated.agentanycast.v1 import (  # noqa: E402
     registry_service_pb2 as reg_pb2,
 )
 
-from .conftest import unique_id, wait_for  # noqa: E402
+from .conftest import unique_id  # noqa: E402
 
 
 class TestAnycastSkillRouting:
@@ -45,19 +47,25 @@ def test_anycast_skill_routing(self, grpc_registry, grpc_node_a, grpc_node_b) ->
         peer_id_b = resp_b.node_info.peer_id
 
         # Register the skill at the relay for node B.
-        grpc_registry.RegisterSkills(reg_pb2.RegisterSkillsRequest(
-            peer_id=peer_id_b,
-            skills=[reg_pb2.SkillInfo(
-                skill_id=skill_id,
-                description=f"Test skill {skill_id}",
-            )],
-            agent_name=f"Anycast Agent ({skill_id})",
-        ))
+        grpc_registry.RegisterSkills(
+            reg_pb2.RegisterSkillsRequest(
+                peer_id=peer_id_b,
+                skills=[
+                    reg_pb2.SkillInfo(
+                        skill_id=skill_id,
+                        description=f"Test skill {skill_id}",
+                    )
+                ],
+                agent_name=f"Anycast Agent ({skill_id})",
+            )
+        )
 
         # Discover via the registry.
-        resp = grpc_registry.DiscoverBySkill(reg_pb2.DiscoverBySkillRequest(
-            skill_id=skill_id,
-        ))
+        resp = grpc_registry.DiscoverBySkill(
+            reg_pb2.DiscoverBySkillRequest(
+                skill_id=skill_id,
+            )
+        )
         assert len(resp.agents) >= 1
         peer_ids = [a.peer_id for a in resp.agents]
         assert peer_id_b in peer_ids
@@ -77,20 +85,26 @@ def test_anycast_tag_filtering(self, grpc_registry) -> None:
 
         # Register three agents with same skill but different tags.
         for peer_id, lang in [(peer_en, "en"), (peer_zh, "zh"), (peer_ja, "ja")]:
-            grpc_registry.RegisterSkills(reg_pb2.RegisterSkillsRequest(
-                peer_id=peer_id,
-                skills=[reg_pb2.SkillInfo(
-                    skill_id=skill_id,
-                    tags={"lang": lang},
-                )],
-                agent_name=f"Agent-{lang}",
-            ))
+            grpc_registry.RegisterSkills(
+                reg_pb2.RegisterSkillsRequest(
+                    peer_id=peer_id,
+                    skills=[
+                        reg_pb2.SkillInfo(
+                            skill_id=skill_id,
+                            tags={"lang": lang},
+                        )
+                    ],
+                    agent_name=f"Agent-{lang}",
+                )
+            )
 
         # Filter by lang=zh.
-        resp = grpc_registry.DiscoverBySkill(reg_pb2.DiscoverBySkillRequest(
-            skill_id=skill_id,
-            tags={"lang": "zh"},
-        ))
+        resp = grpc_registry.DiscoverBySkill(
+            reg_pb2.DiscoverBySkillRequest(
+                skill_id=skill_id,
+                tags={"lang": "zh"},
+            )
+        )
         peer_ids = [a.peer_id for a in resp.agents]
         assert peer_zh in peer_ids
         assert peer_en not in peer_ids

tests/integration/test_api.py

@@ -40,11 +40,13 @@ def test_list_agents(self, grpc_registry) -> None:
         peer_id = unique_id("12D3KooW-api")
         skill_id = unique_id("api-skill")
 
-        grpc_registry.RegisterSkills(reg_pb2.RegisterSkillsRequest(
-            peer_id=peer_id,
-            skills=[reg_pb2.SkillInfo(skill_id=skill_id, description="API test")],
-            agent_name=agent_name,
-        ))
+        grpc_registry.RegisterSkills(
+            reg_pb2.RegisterSkillsRequest(
+                peer_id=peer_id,
+                skills=[reg_pb2.SkillInfo(skill_id=skill_id, description="API test")],
+                agent_name=agent_name,
+            )
+        )
 
         resp = httpx.get(f"{RELAY_API}/api/v1/agents", timeout=10)
         assert resp.status_code == 200
@@ -60,16 +62,20 @@ def test_filter_by_skill(self, grpc_registry) -> None:
         peer_match = unique_id("12D3KooW-match")
         peer_other = unique_id("12D3KooW-other")
 
-        grpc_registry.RegisterSkills(reg_pb2.RegisterSkillsRequest(
-            peer_id=peer_match,
-            skills=[reg_pb2.SkillInfo(skill_id=skill_id)],
-            agent_name="Match Agent",
-        ))
-        grpc_registry.RegisterSkills(reg_pb2.RegisterSkillsRequest(
-            peer_id=peer_other,
-            skills=[reg_pb2.SkillInfo(skill_id=unique_id("other-skill"))],
-            agent_name="Other Agent",
-        ))
+        grpc_registry.RegisterSkills(
+            reg_pb2.RegisterSkillsRequest(
+                peer_id=peer_match,
+                skills=[reg_pb2.SkillInfo(skill_id=skill_id)],
+                agent_name="Match Agent",
+            )
+        )
+        grpc_registry.RegisterSkills(
+            reg_pb2.RegisterSkillsRequest(
+                peer_id=peer_other,
+                skills=[reg_pb2.SkillInfo(skill_id=unique_id("other-skill"))],
+                agent_name="Other Agent",
+            )
+        )
 
         resp = httpx.get(
             f"{RELAY_API}/api/v1/agents",

tests/integration/test_bridge.py

@@ -27,6 +27,8 @@
 
 from agentanycast._generated.agentanycast.v1 import (  # noqa: E402
     agent_card_pb2 as card_pb2,
+)
+from agentanycast._generated.agentanycast.v1 import (  # noqa: E402
     node_service_pb2 as ns_pb2,
 )
 
@@ -40,12 +42,14 @@ def test_bridge_card_endpoint(self, grpc_node_a) -> None:
         """GET /.well-known/agent.json returns a valid AgentCard JSON."""
         # First set a card on node-a so the endpoint has something to return.
         card_name = unique_id("bridge-agent")
-        grpc_node_a.SetAgentCard(ns_pb2.SetAgentCardRequest(
-            card=card_pb2.AgentCard(
-                name=card_name,
-                skills=[card_pb2.Skill(id="bridge-test", description="Bridge test skill")],
-            ),
-        ))
+        grpc_node_a.SetAgentCard(
+            ns_pb2.SetAgentCardRequest(
+                card=card_pb2.AgentCard(
+                    name=card_name,
+                    skills=[card_pb2.Skill(id="bridge-test", description="Bridge test skill")],
+                ),
+            )
+        )
 
         resp = httpx.get(f"{NODE_A_BRIDGE}/.well-known/agent.json", timeout=10)
         assert resp.status_code == 200
@@ -57,12 +61,14 @@ def test_bridge_card_endpoint(self, grpc_node_a) -> None:
     def test_bridge_inbound_endpoint(self, grpc_node_a) -> None:
         """POST /a2a endpoint should accept requests (or return a structured error)."""
         # Ensure there's a card set.
-        grpc_node_a.SetAgentCard(ns_pb2.SetAgentCardRequest(
-            card=card_pb2.AgentCard(
-                name=unique_id("bridge-inbound"),
-                skills=[card_pb2.Skill(id="inbound-test", description="Inbound test")],
-            ),
-        ))
+        grpc_node_a.SetAgentCard(
+            ns_pb2.SetAgentCardRequest(
+                card=card_pb2.AgentCard(
+                    name=unique_id("bridge-inbound"),
+                    skills=[card_pb2.Skill(id="inbound-test", description="Inbound test")],
+                ),
+            )
+        )
 
         # Send a minimal A2A JSON-RPC request.
         payload = {