Refactor: Correct polling infrequent usecase

temporalio · Jul 8, 2024 · af2c5c2 · af2c5c2
1 parent ba5fd0f
commit af2c5c2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/polling/infrequent/activities.py b/polling/infrequent/activities.py
@@ -13,7 +13,8 @@ class ComposeGreetingInput:
 
 @activity.defn
 async def compose_greeting(input: ComposeGreetingInput) -> str:
-    test_service = TestService()
+    attempt = activity.info().attempt - 1
+    test_service = TestService(attempt=attempt)
     # If this raises an exception because it's not done yet, the activity will
     # continually be scheduled for retry
     return await test_service.get_service_result(input)
diff --git a/polling/test_service.py b/polling/test_service.py
@@ -1,7 +1,7 @@
 class TestService:
-    def __init__(self):
-        self.try_attempts = 0
-        self.error_attempts = 5
+    def __init__(self, attempt=0, error_attempts=5):
+        self.try_attempts = attempt
+        self.error_attempts = error_attempts
 
     async def get_service_result(self, input):
         print(