chore: import upstream snapshot with attribution

tests/sandbox/integration_tests/test_model.py

@@ -0,0 +1,59 @@
+from __future__ import annotations
+
+import json
+from collections.abc import Mapping, Sequence
+from typing import Any
+
+from agents.items import TResponseOutputItem
+from tests.fake_model import FakeModel
+from tests.test_responses import get_final_output_message, get_function_tool_call
+
+__test__ = False
+
+
+class TestModel(FakeModel):
+    """Reusable queued model for sandbox integration tests."""
+
+    __test__ = False
+
+    def queue_turn(self, *items: TResponseOutputItem) -> None:
+        self.set_next_output(list(items))
+
+    def queue_function_call(
+        self,
+        name: str,
+        arguments: Mapping[str, Any] | str | None = None,
+        *,
+        call_id: str | None = None,
+        namespace: str | None = None,
+    ) -> None:
+        self.queue_turn(
+            get_function_tool_call(
+                name,
+                _serialize_arguments(arguments),
+                call_id=call_id,
+                namespace=namespace,
+            )
+        )
+
+    def queue_function_calls(
+        self,
+        calls: Sequence[tuple[str, Mapping[str, Any] | str | None, str | None]],
+    ) -> None:
+        self.queue_turn(
+            *[
+                get_function_tool_call(name, _serialize_arguments(arguments), call_id=call_id)
+                for name, arguments, call_id in calls
+            ]
+        )
+
+    def queue_final_output(self, output: str) -> None:
+        self.queue_turn(get_final_output_message(output))
+
+
+def _serialize_arguments(arguments: Mapping[str, Any] | str | None) -> str:
+    if arguments is None:
+        return "{}"
+    if isinstance(arguments, str):
+        return arguments
+    return json.dumps(arguments)