Figure out mocks for one test

Author: donald-pinckney

pyproject.toml

@@ -26,6 +26,7 @@ dev = [
     "types-pyyaml>=6.0.12.20241230,<7",
     "pytest-pretty>=1.3.0",
     "poethepoet>=0.36.0",
+    "pytest-mock>=3.15.1",
 ]
 bedrock = ["boto3>=1.34.92,<2"]
 dsl = [

tests/conftest.py

@@ -37,14 +37,14 @@ def event_loop():
     loop.close()
 
 
-@pytest.fixture(scope="session")
+@pytest.fixture
 def plugins():
     # By default, no plugins.
     # Other tests can override this fixture, such as in tests/openai_agents/conftest.py
     return []
 
 
-@pytest_asyncio.fixture(scope="session")
+@pytest_asyncio.fixture
 async def env(request, plugins) -> AsyncGenerator[WorkflowEnvironment, None]:
     env_type = request.config.getoption("--workflow-environment")
     if env_type == "local":

tests/openai_agents/basic/test_hello_world_workflow.py

@@ -1,12 +1,43 @@
 import uuid
 from concurrent.futures import ThreadPoolExecutor
 
+import pytest
+from agents import ModelResponse, Usage
+from openai.types.responses import ResponseOutputMessage, ResponseOutputText
 from temporalio.client import Client
 from temporalio.worker import Worker
 
 from openai_agents.basic.workflows.hello_world_workflow import HelloWorldAgent
 
 
+@pytest.fixture
+def mocked_model(mocker):
+    mock = mocker.AsyncMock()
+    mock.get_response.side_effect = [
+        ModelResponse(
+            output=[
+                ResponseOutputMessage(
+                    id="1",
+                    content=[
+                        ResponseOutputText(
+                            annotations=[],
+                            text="This is a haiku (not really)",
+                            type="output_text",
+                        )
+                    ],
+                    role="assistant",
+                    status="completed",
+                    type="message",
+                )
+            ],
+            usage=Usage(requests=1, input_tokens=1, output_tokens=1, total_tokens=1),
+            response_id="1",
+        )
+    ]
+
+    return mock
+
+
 async def test_execute_workflow(client: Client):
     task_queue_name = str(uuid.uuid4())
 

tests/openai_agents/conftest.py

@@ -1,15 +1,31 @@
 from datetime import timedelta
+from typing import Optional
 
 import pytest
+from agents import Model, ModelProvider
 from temporalio.contrib.openai_agents import ModelActivityParameters, OpenAIAgentsPlugin
 
 
-@pytest.fixture(scope="session")
-def plugins():
+class MockedModelProvider(ModelProvider):
+    def __init__(self, mocked_model):
+        self.mocked_model = mocked_model
+
+    def get_model(self, model_name: str | None) -> Model:
+        return self.mocked_model
+
+
+@pytest.fixture
+def model_provider(mocked_model):
+    return MockedModelProvider(mocked_model)
+
+
+@pytest.fixture
+def plugins(model_provider: Optional[ModelProvider]):
     return [
         OpenAIAgentsPlugin(
             model_params=ModelActivityParameters(
                 start_to_close_timeout=timedelta(seconds=30)
-            )
+            ),
+            model_provider=model_provider,
         )
     ]

tests/openai_agents/e2e/conftest.py

@@ -0,0 +1,9 @@
+from datetime import timedelta
+
+import pytest
+from temporalio.contrib.openai_agents import ModelActivityParameters, OpenAIAgentsPlugin
+
+
+@pytest.fixture
+def model_provider():
+    return None

tests/openai_agents/e2e/test_hello_world_workflow_e2e.py

@@ -0,0 +1,26 @@
+import uuid
+from concurrent.futures import ThreadPoolExecutor
+
+from temporalio.client import Client
+from temporalio.worker import Worker
+
+from openai_agents.basic.workflows.hello_world_workflow import HelloWorldAgent
+
+
+async def test_execute_workflow(client: Client):
+    task_queue_name = str(uuid.uuid4())
+
+    async with Worker(
+        client,
+        task_queue=task_queue_name,
+        workflows=[HelloWorldAgent],
+        activity_executor=ThreadPoolExecutor(5),
+    ):
+        result = await client.execute_workflow(
+            HelloWorldAgent.run,
+            "Write a recursive haiku about recursive haikus.",
+            id=str(uuid.uuid4()),
+            task_queue=task_queue_name,
+        )
+        assert isinstance(result, str)
+        assert len(result) > 0