|
11 | 11 | from temporalio.exceptions import ApplicationError, CancelledError |
12 | 12 | from temporalio.worker import Worker |
13 | 13 |
|
14 | | -STANDALONE_ACTIVITY_MATCHING_TASKS_IMPLEMENTED = False |
15 | | - |
16 | | -pytest.skip("Standalone activity is not implemented in server", allow_module_level=True) |
17 | | - |
18 | 14 |
|
19 | 15 | @activity.defn |
20 | 16 | async def increment(input: int) -> int: |
@@ -106,13 +102,12 @@ async def test_manual_completion(client: Client): |
106 | 102 | activities=[async_activity], |
107 | 103 | workflows=[WaitForSignalWorkflow], |
108 | 104 | ): |
109 | | - if STANDALONE_ACTIVITY_MATCHING_TASKS_IMPLEMENTED: |
110 | | - # Wait for activity to start |
111 | | - await client.execute_workflow( |
112 | | - WaitForSignalWorkflow.run, |
113 | | - id=wait_for_signal_workflow_id, |
114 | | - task_queue=task_queue, |
115 | | - ) |
| 105 | + # Wait for activity to start |
| 106 | + await client.execute_workflow( |
| 107 | + WaitForSignalWorkflow.run, |
| 108 | + id=wait_for_signal_workflow_id, |
| 109 | + task_queue=task_queue, |
| 110 | + ) |
116 | 111 | # Complete activity manually |
117 | 112 | async_activity_handle = client.get_async_activity_handle( |
118 | 113 | activity_id=activity_id, |
@@ -143,14 +138,11 @@ async def test_manual_cancellation(client: Client): |
143 | 138 | activities=[async_activity], |
144 | 139 | workflows=[WaitForSignalWorkflow], |
145 | 140 | ): |
146 | | - if STANDALONE_ACTIVITY_MATCHING_TASKS_IMPLEMENTED: |
147 | | - # Wait for activity to start |
148 | | - await client.execute_workflow( |
149 | | - WaitForSignalWorkflow.run, |
150 | | - id=wait_for_signal_workflow_id, |
151 | | - task_queue=task_queue, |
152 | | - ) |
153 | | - # Cancel activity manually |
| 141 | + await client.execute_workflow( |
| 142 | + WaitForSignalWorkflow.run, |
| 143 | + id=wait_for_signal_workflow_id, |
| 144 | + task_queue=task_queue, |
| 145 | + ) |
154 | 146 | async_activity_handle = client.get_async_activity_handle( |
155 | 147 | activity_id=activity_id, |
156 | 148 | run_id=activity_handle.run_id, |
@@ -182,14 +174,11 @@ async def test_manual_fail(client: Client): |
182 | 174 | activities=[async_activity], |
183 | 175 | workflows=[WaitForSignalWorkflow], |
184 | 176 | ): |
185 | | - if STANDALONE_ACTIVITY_MATCHING_TASKS_IMPLEMENTED: |
186 | | - # Wait for activity to start |
187 | | - await client.execute_workflow( |
188 | | - WaitForSignalWorkflow.run, |
189 | | - id=wait_for_signal_workflow_id, |
190 | | - task_queue=task_queue, |
191 | | - ) |
192 | | - # Fail activity manually |
| 177 | + await client.execute_workflow( |
| 178 | + WaitForSignalWorkflow.run, |
| 179 | + id=wait_for_signal_workflow_id, |
| 180 | + task_queue=task_queue, |
| 181 | + ) |
193 | 182 | async_activity_handle = client.get_async_activity_handle( |
194 | 183 | activity_id=activity_id, |
195 | 184 | run_id=activity_handle.run_id, |
@@ -241,13 +230,13 @@ async def test_manual_heartbeat(client: Client): |
241 | 230 | activities=[activity_for_testing_heartbeat], |
242 | 231 | workflows=[WaitForSignalWorkflow], |
243 | 232 | ): |
244 | | - # Send heartbeat manually |
245 | 233 | async_activity_handle = client.get_async_activity_handle( |
246 | 234 | activity_id=activity_id, |
247 | 235 | run_id=activity_handle.run_id, |
248 | 236 | ) |
249 | 237 | await async_activity_handle.heartbeat("Test heartbeat details") |
250 | | - await client.get_workflow_handle( |
| 238 | + await client.get_workflow_handle_for( |
| 239 | + WaitForSignalWorkflow.run, |
251 | 240 | workflow_id=wait_for_signal_workflow_id, |
252 | 241 | ).signal(WaitForSignalWorkflow.signal) |
253 | 242 | assert await activity_handle.result() == "Test heartbeat details" |
|
0 commit comments