Cleanup

Author: dandavison

tests/nexus/test_timeout.py

@@ -13,8 +13,13 @@
 from tests.helpers.nexus import create_nexus_endpoint, make_nexus_endpoint_name
 
 
+@nexusrpc.service
+class Service:
+    op_that_never_returns: nexusrpc.Operation[None, None]
+
+
 @nexusrpc.handler.service_handler
-class NexusService:
+class ServiceHandler:
     @nexusrpc.handler.sync_operation
     async def op_that_never_returns(
         self, _ctx: nexusrpc.handler.StartOperationContext, id: None
@@ -28,10 +33,10 @@ class NexusCallerWorkflow:
     async def run(self) -> None:
         nexus_client = workflow.create_nexus_client(
             endpoint=make_nexus_endpoint_name(workflow.info().task_queue),
-            service=NexusService,
+            service=ServiceHandler,
         )
         await nexus_client.execute_operation(
-            NexusService.op_that_never_returns,
+            Service.op_that_never_returns,
             None,
             schedule_to_close_timeout=timedelta(seconds=10),
         )
@@ -44,7 +49,7 @@ async def test_nexus_timeout(env: WorkflowEnvironment):
     async with new_worker(
         env.client,
         NexusCallerWorkflow,
-        nexus_service_handlers=[NexusService()],
+        nexus_service_handlers=[ServiceHandler()],
     ) as worker:
         await create_nexus_endpoint(worker.task_queue, env.client)