97 lines
3.7 KiB
Python
97 lines
3.7 KiB
Python
from unittest.mock import MagicMock, patch
|
|
|
|
|
|
def test_resolve_isolation_owner_prefers_user_id():
|
|
from api.services.browser_runtime_manager import resolve_isolation_owner
|
|
|
|
owner = resolve_isolation_owner(
|
|
"docker-per-principal",
|
|
task_id="task-1",
|
|
metadata={"user_id": "user-7", "session_id": "session-9"},
|
|
thread_id="thread-1",
|
|
)
|
|
|
|
assert owner == "user-7"
|
|
|
|
|
|
def test_resolve_isolation_owner_uses_task_for_per_task_mode():
|
|
from api.services.browser_runtime_manager import resolve_isolation_owner
|
|
|
|
owner = resolve_isolation_owner(
|
|
"docker-per-task",
|
|
task_id="task-42",
|
|
metadata={"user_id": "user-7"},
|
|
thread_id="thread-1",
|
|
)
|
|
|
|
assert owner == "task-42"
|
|
|
|
|
|
def test_hash_runtime_owner_is_stable():
|
|
from api.services.browser_runtime_manager import hash_runtime_owner
|
|
|
|
assert hash_runtime_owner("owner-1") == hash_runtime_owner("owner-1")
|
|
assert hash_runtime_owner("owner-1") != hash_runtime_owner("owner-2")
|
|
|
|
|
|
def test_shared_mode_returns_shared_runtime(monkeypatch):
|
|
from api.services import browser_runtime_manager
|
|
|
|
monkeypatch.setenv("BROWSER_USE_ISOLATION_MODE", "shared")
|
|
monkeypatch.setenv("BROWSER_URL", "http://shared-browser:9333")
|
|
monkeypatch.setenv("BROWSER_USE_RPC_URL", "http://shared-browser:8787/run")
|
|
monkeypatch.setenv("BROWSER_VIEW_BASE_URL", "https://viewer.example.com")
|
|
|
|
runtime = browser_runtime_manager.ensure_browser_runtime(
|
|
task_id="task-1",
|
|
metadata={"user_id": "user-7"},
|
|
thread_id="thread-1",
|
|
)
|
|
|
|
assert runtime["cdp_url"] == "http://shared-browser:9333"
|
|
assert runtime["rpc_url"] == "http://shared-browser:8787/run"
|
|
assert runtime["browser_view"] == "https://viewer.example.com/vnc.html?path=websockify"
|
|
assert runtime["isolation_mode"] == "shared"
|
|
|
|
|
|
def test_isolated_mode_starts_container(monkeypatch):
|
|
from api.services import browser_runtime_manager
|
|
|
|
monkeypatch.setenv("BROWSER_USE_ISOLATION_MODE", "docker-per-principal")
|
|
monkeypatch.setenv("BROWSER_RUNTIME_IMAGE", "browser-use-browser-runtime:test")
|
|
monkeypatch.setenv("BROWSER_RUNTIME_NETWORK", "browser-net")
|
|
monkeypatch.setenv("BROWSER_VIEW_BASE_URL", "https://viewer.example.com")
|
|
|
|
saved_registry = {}
|
|
docker_calls = []
|
|
|
|
def fake_run_docker(args, check=True):
|
|
docker_calls.append(args)
|
|
if args[:2] == ["inspect", "-f"]:
|
|
return MagicMock(returncode=1, stdout="", stderr="")
|
|
if args[:1] == ["inspect"]:
|
|
return MagicMock(returncode=1, stdout="", stderr="")
|
|
return MagicMock(returncode=0, stdout="ok", stderr="")
|
|
|
|
with (
|
|
patch.object(browser_runtime_manager, "_load_registry", return_value={"runtimes": {}}),
|
|
patch.object(browser_runtime_manager, "_save_registry", side_effect=lambda payload: saved_registry.update(payload)),
|
|
patch.object(browser_runtime_manager, "_run_docker", side_effect=fake_run_docker),
|
|
patch.object(browser_runtime_manager, "_wait_for_runtime") as mock_wait,
|
|
):
|
|
runtime = browser_runtime_manager.ensure_browser_runtime(
|
|
task_id="task-1",
|
|
metadata={"user_id": "user-7"},
|
|
thread_id="thread-1",
|
|
)
|
|
|
|
assert runtime["isolation_mode"] == "docker-per-principal"
|
|
assert runtime["cdp_url"].startswith("http://browser-use-browser-")
|
|
assert runtime["rpc_url"].startswith("http://browser-use-browser-")
|
|
assert runtime["rpc_url"].endswith(":8787/run")
|
|
assert "/view/" in runtime["browser_view"]
|
|
assert saved_registry["runtimes"]
|
|
run_commands = [call for call in docker_calls if call[:2] == ["run", "-d"]]
|
|
assert run_commands
|
|
assert "browser-use-browser-runtime:test" in run_commands[0]
|
|
mock_wait.assert_called_once()
|