chore: import upstream snapshot with attribution

2026-07-13 12:39:17 +08:00
commit 4ed4e9ff99
1368 changed files with 334957 additions and 0 deletions
@@ -0,0 +1,81 @@
+from __future__ import annotations
+
+import pytest
+
+from agents.sandbox.errors import ExposedPortUnavailableError
+from agents.sandbox.sandboxes import UnixLocalSandboxClient, UnixLocalSandboxClientOptions
+from agents.sandbox.types import ExposedPortEndpoint
+
+
+def test_exposed_port_endpoint_formats_urls() -> None:
+    insecure = ExposedPortEndpoint(host="127.0.0.1", port=8765, tls=False)
+    secure = ExposedPortEndpoint(host="sandbox.example.test", port=443, tls=True)
+
+    assert insecure.url_for("http") == "http://127.0.0.1:8765/"
+    assert insecure.url_for("ws") == "ws://127.0.0.1:8765/"
+    assert secure.url_for("http") == "https://sandbox.example.test/"
+    assert secure.url_for("ws") == "wss://sandbox.example.test/"
+
+
+def test_exposed_port_endpoint_with_query() -> None:
+    endpoint = ExposedPortEndpoint(
+        host="preview.example.com",
+        port=443,
+        tls=True,
+        query="bl_preview_token=abc123",
+    )
+    assert endpoint.url_for("http") == "https://preview.example.com/?bl_preview_token=abc123"
+    assert endpoint.url_for("ws") == "wss://preview.example.com/?bl_preview_token=abc123"
+
+
+def test_exposed_port_endpoint_accepts_leading_question_mark_query() -> None:
+    endpoint = ExposedPortEndpoint(
+        host="preview.example.com",
+        port=443,
+        tls=True,
+        query="?bl_preview_token=abc123",
+    )
+
+    assert endpoint.url_for("http") == "https://preview.example.com/?bl_preview_token=abc123"
+    assert endpoint.url_for("ws") == "wss://preview.example.com/?bl_preview_token=abc123"
+
+
+def test_exposed_port_endpoint_empty_query() -> None:
+    endpoint = ExposedPortEndpoint(host="127.0.0.1", port=8080, tls=False, query="")
+    assert endpoint.url_for("http") == "http://127.0.0.1:8080/"
+
+
+@pytest.mark.asyncio
+async def test_unix_local_resolve_exposed_port_uses_wrapper_and_normalizes_state() -> None:
+    client = UnixLocalSandboxClient()
+    session = await client.create(
+        options=UnixLocalSandboxClientOptions(exposed_ports=(8765, 8765)),
+    )
+
+    try:
+        endpoint = await session.resolve_exposed_port(8765)
+    finally:
+        await session.aclose()
+        await client.delete(session)
+
+    assert session.state.exposed_ports == (8765,)
+    assert endpoint == ExposedPortEndpoint(host="127.0.0.1", port=8765, tls=False)
+    assert endpoint.url_for("ws") == "ws://127.0.0.1:8765/"
+
+
+@pytest.mark.asyncio
+async def test_unix_local_resolve_exposed_port_rejects_undeclared_ports() -> None:
+    client = UnixLocalSandboxClient()
+    session = await client.create(
+        options=UnixLocalSandboxClientOptions(exposed_ports=(8765,)),
+    )
+
+    try:
+        with pytest.raises(ExposedPortUnavailableError) as exc_info:
+            await session.resolve_exposed_port(9000)
+    finally:
+        await session.aclose()
+        await client.delete(session)
+
+    assert exc_info.value.context["reason"] == "not_configured"
+    assert exc_info.value.context["exposed_ports"] == [8765]