fixed tests

Author: GSmithApps

tests/custom_metric/workflow_test.py

@@ -1,4 +1,5 @@
 import uuid
+from concurrent.futures import ThreadPoolExecutor
 
 from temporalio import activity
 from temporalio.client import Client
@@ -22,6 +23,7 @@ async def print_message_mock():
         task_queue=_TASK_QUEUE,
         workflows=[StartTwoActivitiesWorkflow],
         activities=[print_message_mock],
+        activity_executor=ThreadPoolExecutor(5),
     ):
         result = await client.execute_workflow(
             StartTwoActivitiesWorkflow.run,

tests/hello/hello_activity_choice_test.py

@@ -20,13 +20,12 @@
 ]
 
 
-@pytest.mark.asyncio
 @pytest.mark.parametrize(
     "activity_func, order_amount, expected_result", activity_test_data
 )
-async def test_order_fruit(activity_func, order_amount, expected_result):
+def test_order_fruit(activity_func, order_amount, expected_result):
     activity_environment = ActivityEnvironment()
 
-    result = await activity_environment.run(activity_func, order_amount)
+    result = activity_environment.run(activity_func, order_amount)
 
     assert result == expected_result

tests/hello/hello_activity_test.py

@@ -1,4 +1,5 @@
 import uuid
+from concurrent.futures import ThreadPoolExecutor
 
 from temporalio import activity
 from temporalio.client import Client
@@ -18,6 +19,7 @@ async def test_execute_workflow(client: Client):
         client,
         task_queue=task_queue_name,
         workflows=[GreetingWorkflow],
+        activity_executor=ThreadPoolExecutor(5),
         activities=[compose_greeting],
     ):
         assert "Hello, World!" == await client.execute_workflow(

tests/hello/hello_cancellation_test.py

@@ -1,5 +1,6 @@
 import asyncio
 import uuid
+from concurrent.futures import ThreadPoolExecutor
 
 import pytest
 from temporalio.client import Client, WorkflowExecutionStatus, WorkflowFailureError
@@ -21,6 +22,7 @@ async def test_cancel_workflow(client: Client):
         task_queue=task_queue_name,
         workflows=[CancellationWorkflow],
         activities=[cleanup_activity, never_complete_activity],
+        activity_executor=ThreadPoolExecutor(5),
     ):
         handle = await client.start_workflow(
             CancellationWorkflow.run,