|
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 | + |
14 | 16 |
|
15 | 17 | @activity.defn |
16 | 18 | async def increment(input: int) -> int: |
@@ -105,12 +107,13 @@ async def test_manual_completion(client: Client): |
105 | 107 | activities=[async_activity], |
106 | 108 | workflows=[WaitForSignalWorkflow], |
107 | 109 | ): |
108 | | - # Wait for activity to start |
109 | | - await client.execute_workflow( |
110 | | - WaitForSignalWorkflow.run, |
111 | | - id=wait_for_signal_workflow_id, |
112 | | - task_queue=task_queue, |
113 | | - ) |
| 110 | + if STANDALONE_ACTIVITY_MATCHING_TASKS_IMPLEMENTED: |
| 111 | + # Wait for activity to start |
| 112 | + await client.execute_workflow( |
| 113 | + WaitForSignalWorkflow.run, |
| 114 | + id=wait_for_signal_workflow_id, |
| 115 | + task_queue=task_queue, |
| 116 | + ) |
114 | 117 | # Complete activity manually |
115 | 118 | async_activity_handle = client.get_async_activity_handle( |
116 | 119 | activity_id=activity_id, |
@@ -144,11 +147,14 @@ async def test_manual_cancellation(client: Client): |
144 | 147 | activities=[async_activity], |
145 | 148 | workflows=[WaitForSignalWorkflow], |
146 | 149 | ): |
147 | | - await client.execute_workflow( |
148 | | - WaitForSignalWorkflow.run, |
149 | | - id=wait_for_signal_workflow_id, |
150 | | - task_queue=task_queue, |
151 | | - ) |
| 150 | + if STANDALONE_ACTIVITY_MATCHING_TASKS_IMPLEMENTED: |
| 151 | + # Wait for activity to start |
| 152 | + await client.execute_workflow( |
| 153 | + WaitForSignalWorkflow.run, |
| 154 | + id=wait_for_signal_workflow_id, |
| 155 | + task_queue=task_queue, |
| 156 | + ) |
| 157 | + # Cancel activity manually |
152 | 158 | async_activity_handle = client.get_async_activity_handle( |
153 | 159 | activity_id=activity_id, |
154 | 160 | run_id=activity_handle.run_id, |
@@ -183,11 +189,14 @@ async def test_manual_fail(client: Client): |
183 | 189 | activities=[async_activity], |
184 | 190 | workflows=[WaitForSignalWorkflow], |
185 | 191 | ): |
186 | | - await client.execute_workflow( |
187 | | - WaitForSignalWorkflow.run, |
188 | | - id=wait_for_signal_workflow_id, |
189 | | - task_queue=task_queue, |
190 | | - ) |
| 192 | + if STANDALONE_ACTIVITY_MATCHING_TASKS_IMPLEMENTED: |
| 193 | + # Wait for activity to start |
| 194 | + await client.execute_workflow( |
| 195 | + WaitForSignalWorkflow.run, |
| 196 | + id=wait_for_signal_workflow_id, |
| 197 | + task_queue=task_queue, |
| 198 | + ) |
| 199 | + # Fail activity manually |
191 | 200 | async_activity_handle = client.get_async_activity_handle( |
192 | 201 | activity_id=activity_id, |
193 | 202 | run_id=activity_handle.run_id, |
@@ -242,13 +251,13 @@ async def test_manual_heartbeat(client: Client): |
242 | 251 | activities=[activity_for_testing_heartbeat], |
243 | 252 | workflows=[WaitForSignalWorkflow], |
244 | 253 | ): |
| 254 | + # Send heartbeat manually |
245 | 255 | async_activity_handle = client.get_async_activity_handle( |
246 | 256 | activity_id=activity_id, |
247 | 257 | run_id=activity_handle.run_id, |
248 | 258 | ) |
249 | 259 | await async_activity_handle.heartbeat("Test heartbeat details") |
250 | | - await client.get_workflow_handle_for( |
251 | | - WaitForSignalWorkflow.run, |
| 260 | + await client.get_workflow_handle( |
252 | 261 | workflow_id=wait_for_signal_workflow_id, |
253 | 262 | ).signal(WaitForSignalWorkflow.signal) |
254 | 263 | assert await activity_handle.result() == "Test heartbeat details" |
|
0 commit comments