add test to hello_child sample (temporalio#77)

antmendoza · web-flow · commit 9a5424cec12e · 2023-07-10T19:01:49.000+02:00
diff --git a/tests/hello/hello_child_test.py b/tests/hello/hello_child_test.py
@@ -0,0 +1,48 @@
+import uuid
+
+from temporalio import workflow
+from temporalio.client import Client
+from temporalio.worker import Worker
+
+from hello.hello_child_workflow import (
+    ComposeGreetingInput,
+    ComposeGreetingWorkflow,
+    GreetingWorkflow,
+)
+
+
+async def test_child_workflow(client: Client):
+    task_queue_name = str(uuid.uuid4())
+    async with Worker(
+        client,
+        task_queue=task_queue_name,
+        workflows=[GreetingWorkflow, ComposeGreetingWorkflow],
+    ):
+        assert "Hello, World!" == await client.execute_workflow(
+            GreetingWorkflow.run,
+            "World",
+            id=str(uuid.uuid4()),
+            task_queue=task_queue_name,
+        )
+
+
+@workflow.defn(name="ComposeGreetingWorkflow")
+class MockedComposeGreetingWorkflow:
+    @workflow.run
+    async def run(self, input: ComposeGreetingInput) -> str:
+        return f"{input.greeting}, {input.name} from mocked child!"
+
+
+async def test_mock_child_workflow(client: Client):
+    task_queue_name = str(uuid.uuid4())
+    async with Worker(
+        client,
+        task_queue=task_queue_name,
+        workflows=[GreetingWorkflow, MockedComposeGreetingWorkflow],
+    ):
+        assert "Hello, World from mocked child!" == await client.execute_workflow(
+            GreetingWorkflow.run,
+            "World",
+            id=str(uuid.uuid4()),
+            task_queue=task_queue_name,
+        )
