chore: import upstream snapshot with attribution

libs/python/cua-sandbox/tests/test_integration_agent.py

@@ -0,0 +1,77 @@
+"""Integration tests for agent handler adapters."""
+
+from __future__ import annotations
+
+import pytest
+from cua_sandbox.agent import LocalhostHandler, SandboxHandler, is_sandbox
+
+pytestmark = pytest.mark.asyncio
+
+
+class TestSandboxHandler:
+    async def test_screenshot(self, local_sandbox):
+        handler = SandboxHandler(local_sandbox)
+        b64 = await handler.screenshot()
+        assert isinstance(b64, str)
+        assert len(b64) > 100
+
+    async def test_get_environment(self, local_sandbox):
+        handler = SandboxHandler(local_sandbox)
+        env = await handler.get_environment()
+        assert env in ("windows", "mac", "linux", "browser")
+
+    async def test_get_dimensions(self, local_sandbox):
+        handler = SandboxHandler(local_sandbox)
+        w, h = await handler.get_dimensions()
+        assert w > 0 and h > 0
+
+    async def test_click(self, local_sandbox):
+        handler = SandboxHandler(local_sandbox)
+        await handler.click(100, 100)
+
+    async def test_type(self, local_sandbox):
+        handler = SandboxHandler(local_sandbox)
+        await handler.type("test")
+
+    async def test_keypress(self, local_sandbox):
+        handler = SandboxHandler(local_sandbox)
+        await handler.keypress(["ctrl", "a"])
+
+    async def test_scroll(self, local_sandbox):
+        handler = SandboxHandler(local_sandbox)
+        await handler.scroll(100, 100, 0, 3)
+
+    async def test_move(self, local_sandbox):
+        handler = SandboxHandler(local_sandbox)
+        await handler.move(200, 200)
+
+    async def test_drag(self, local_sandbox):
+        handler = SandboxHandler(local_sandbox)
+        await handler.drag([{"x": 100, "y": 100}, {"x": 200, "y": 200}])
+
+    async def test_wait(self, local_sandbox):
+        handler = SandboxHandler(local_sandbox)
+        await handler.wait(10)  # 10ms
+
+
+class TestLocalhostHandler:
+    async def test_screenshot(self, localhost_instance):
+        handler = LocalhostHandler(localhost_instance)
+        b64 = await handler.screenshot()
+        assert isinstance(b64, str)
+
+    async def test_click(self, localhost_instance):
+        handler = LocalhostHandler(localhost_instance)
+        await handler.click(50, 50)
+
+
+class TestIsSandbox:
+    async def test_sandbox_detected(self, local_sandbox):
+        assert is_sandbox(local_sandbox)
+
+    async def test_localhost_detected(self, localhost_instance):
+        assert is_sandbox(localhost_instance)
+
+    def test_other_not_detected(self):
+        assert not is_sandbox("not a sandbox")
+        assert not is_sandbox(42)