test

Author: pakrym-oai

src/agents/run.py

@@ -49,7 +49,7 @@
 DEFAULT_RUNNER: Runner | None = None
 
 
-def set_default_runner(runner: Runner) -> None:
+def set_default_runner(runner: Runner | None) -> None:
     """
     Set the default runner to use for the agent run.
     """
@@ -146,7 +146,7 @@ def run_sync_impl(
         pass
 
     @abc.abstractmethod
-    def run_streaming_impl(
+    def run_streamed_impl(
         self,
         starting_agent: Agent[TContext],
         input: str | list[TResponseInputItem],
@@ -297,7 +297,7 @@ def run_streamed(
             A result object that contains data about the run, as well as a method to stream events.
         """
         runner = DEFAULT_RUNNER or DefaultRunner()
-        return runner.run_streaming_impl(
+        return runner.run_streamed_impl(
             starting_agent,
             input,
             context=context,
@@ -511,7 +511,7 @@ def run_sync_impl(
             )
         )
 
-    def run_streaming_impl(
+    def run_streamed_impl(
         self,
         starting_agent: Agent[TContext],
         input: str | list[TResponseInputItem],

tests/conftest.py

@@ -5,6 +5,7 @@
 from agents.models import _openai_shared
 from agents.models.openai_chatcompletions import OpenAIChatCompletionsModel
 from agents.models.openai_responses import OpenAIResponsesModel
+from agents.run import set_default_runner
 from agents.tracing import set_trace_processors
 from agents.tracing.setup import GLOBAL_TRACE_PROVIDER
 
@@ -33,6 +34,11 @@ def clear_openai_settings():
     _openai_shared._use_responses_by_default = True
 
 
+@pytest.fixture(autouse=True)
+def clear_default_runner():
+    set_default_runner(None)
+
+
 # This fixture will run after all tests end
 @pytest.fixture(autouse=True, scope="session")
 def shutdown_trace_provider():

tests/test_run.py

@@ -0,0 +1,26 @@
+from __future__ import annotations
+
+from unittest import mock
+
+import pytest
+
+from agents import Agent, Runner
+from agents.run import set_default_runner
+
+from .fake_model import FakeModel
+
+
+@pytest.mark.asyncio
+async def test_static_run_methods_call_into_default_runner() -> None:
+    runner = mock.Mock(spec=Runner)
+    set_default_runner(runner)
+
+    agent = Agent(name="test", model=FakeModel())
+    await Runner.run(agent, input="test")
+    runner.run_impl.assert_called_once()
+
+    Runner.run_streamed(agent, input="test")
+    runner.run_streamed_impl.assert_called_once()
+
+    Runner.run_sync(agent, input="test")
+    runner.run_sync_impl.assert_called_once()