Skip to content

Commit e37c625

Browse files
committed
TEMP: Alter manual activity tests to not require worker
1 parent e069732 commit e37c625

File tree

1 file changed

+27
-18
lines changed

1 file changed

+27
-18
lines changed

tests/test_activity.py

Lines changed: 27 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,8 @@
1111
from temporalio.exceptions import ApplicationError, CancelledError
1212
from temporalio.worker import Worker
1313

14+
STANDALONE_ACTIVITY_MATCHING_TASKS_IMPLEMENTED = False
15+
1416

1517
@activity.defn
1618
async def increment(input: int) -> int:
@@ -105,12 +107,13 @@ async def test_manual_completion(client: Client):
105107
activities=[async_activity],
106108
workflows=[WaitForSignalWorkflow],
107109
):
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+
)
114117
# Complete activity manually
115118
async_activity_handle = client.get_async_activity_handle(
116119
activity_id=activity_id,
@@ -144,11 +147,14 @@ async def test_manual_cancellation(client: Client):
144147
activities=[async_activity],
145148
workflows=[WaitForSignalWorkflow],
146149
):
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
152158
async_activity_handle = client.get_async_activity_handle(
153159
activity_id=activity_id,
154160
run_id=activity_handle.run_id,
@@ -183,11 +189,14 @@ async def test_manual_fail(client: Client):
183189
activities=[async_activity],
184190
workflows=[WaitForSignalWorkflow],
185191
):
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
191200
async_activity_handle = client.get_async_activity_handle(
192201
activity_id=activity_id,
193202
run_id=activity_handle.run_id,
@@ -242,13 +251,13 @@ async def test_manual_heartbeat(client: Client):
242251
activities=[activity_for_testing_heartbeat],
243252
workflows=[WaitForSignalWorkflow],
244253
):
254+
# Send heartbeat manually
245255
async_activity_handle = client.get_async_activity_handle(
246256
activity_id=activity_id,
247257
run_id=activity_handle.run_id,
248258
)
249259
await async_activity_handle.heartbeat("Test heartbeat details")
250-
await client.get_workflow_handle_for(
251-
WaitForSignalWorkflow.run,
260+
await client.get_workflow_handle(
252261
workflow_id=wait_for_signal_workflow_id,
253262
).signal(WaitForSignalWorkflow.signal)
254263
assert await activity_handle.result() == "Test heartbeat details"

0 commit comments

Comments
 (0)