fixing minor bugs in the original sample and adding the mysterious missing tests that are in the slides but not the repo

Author: MasonEgger

samples/age-estimation/activities.py

@@ -1,7 +1,7 @@
 import urllib.parse
 
 import aiohttp
-from models import EstimatorResponse
+from shared import EstimatorResponse
 from temporalio import activity
 
 

samples/age-estimation/tests/test_activities.py

@@ -0,0 +1,19 @@
+import aiohttp
+import pytest
+from activities import AgeEstimationActivities
+from temporalio.testing import ActivityEnvironment
+
+
+@pytest.mark.asyncio
+@pytest.mark.parametrize(
+    "input, output",
+    [
+        ("Mason", 40),
+    ],
+)
+async def test_retrieve_estimate(input, output):
+    async with aiohttp.ClientSession() as session:
+        activity_environment = ActivityEnvironment()
+        activities = AgeEstimationActivities(session)
+        result = await activity_environment.run(activities.retrieve_estimate, input)
+        assert output == result.age

samples/age-estimation/tests/test_workflow.py

@@ -0,0 +1,28 @@
+import aiohttp
+import pytest
+from activities import AgeEstimationActivities
+from temporalio.testing import WorkflowEnvironment
+from temporalio.worker import Worker
+from workflow import EstimateAgeWorkflow
+
+
+@pytest.mark.asyncio
+async def test_successful_age_estimation():
+    async with await WorkflowEnvironment.start_time_skipping() as env:
+        tq = "age-estimation-workflow"
+        async with aiohttp.ClientSession() as session:
+            activities = AgeEstimationActivities(session)
+            async with Worker(
+                env.client,
+                task_queue=tq,
+                workflows=[EstimateAgeWorkflow],
+                activities=[activities.retrieve_estimate],
+            ):
+                input = "Mason"
+                output = await env.client.execute_workflow(
+                    EstimateAgeWorkflow.run,
+                    input,
+                    id="test-age-estimation-workflow-mason",
+                    task_queue=tq,
+                )
+                assert f"{input} has an estimated age of 40" == output

samples/age-estimation/tests/test_workflow_with_mocks.py

@@ -0,0 +1,38 @@
+import aiohttp
+import pytest
+from activities import AgeEstimationActivities
+from temporalio.testing import WorkflowEnvironment
+from temporalio.worker import Worker
+from temporalio import activity
+from workflow import EstimateAgeWorkflow
+from shared import EstimatorResponse
+
+
+@activity.defn(name="retrieve_estimate")
+async def estimate_age_mocked(input: str) -> EstimatorResponse:
+    if input == "Stanislav":
+        return EstimatorResponse(name="Stanislav", count=8928, age=21)
+    else:
+        return EstimatorResponse(name="Mason", count=1487, age=40)
+
+
+@pytest.mark.asyncio
+async def test_successful_age_estimation():
+    async with await WorkflowEnvironment.start_time_skipping() as env:
+        tq = "age-estimation-workflow"
+        async with aiohttp.ClientSession() as session:
+            activities = AgeEstimationActivities(session)
+            async with Worker(
+                env.client,
+                task_queue=tq,
+                workflows=[EstimateAgeWorkflow],
+                activities=[estimate_age_mocked],
+            ):
+                input = "Stanislav"
+                output = await env.client.execute_workflow(
+                    EstimateAgeWorkflow.run,
+                    input,
+                    id="test-age-estimation-workflow-mason",
+                    task_queue=tq,
+                )
+                assert f"{input} has an estimated age of 21" == output

samples/age-estimation/workflow.py

@@ -5,17 +5,16 @@
 # Import activity, passing it through the sandbox without reloading the module
 with workflow.unsafe.imports_passed_through():
     from activities import AgeEstimationActivities
-    from shared import EstimatorResponse
 
 
 @workflow.defn
-class EstimateAge:
+class EstimateAgeWorkflow:
     @workflow.run
-    async def run(self, name: str) -> EstimatorResponse:
+    async def run(self, name: str) -> str:
         age_estimate = await workflow.execute_activity_method(
             AgeEstimationActivities.retrieve_estimate,
             name,
             start_to_close_timeout=timedelta(seconds=5),
         )
 
-        return age_estimate
+        return f"{name} has an estimated age of {age_estimate.age}"