wehub-resource-sync
85 lines
3.0 KiB
Python
85 lines
3.0 KiB
Python
from __future__ import annotations
|
|
|
|
import importlib
|
|
import sys
|
|
from types import ModuleType
|
|
from typing import Any
|
|
|
|
import pytest
|
|
|
|
|
|
def _restore_module(name: str, original: ModuleType | None) -> None:
|
|
sys.modules.pop(name, None)
|
|
if original is not None:
|
|
sys.modules[name] = original
|
|
|
|
|
|
def _restore_attr(obj: Any, name: str, original: object, existed: bool) -> None:
|
|
if existed:
|
|
setattr(obj, name, original)
|
|
else:
|
|
try:
|
|
delattr(obj, name)
|
|
except AttributeError:
|
|
pass
|
|
|
|
|
|
def test_sandboxes_package_import_skips_unix_local_on_windows(monkeypatch) -> None:
|
|
sandbox_package = importlib.import_module("agents.sandbox")
|
|
original_sandboxes_module = sys.modules.pop("agents.sandbox.sandboxes", None)
|
|
original_unix_local_module = sys.modules.pop("agents.sandbox.sandboxes.unix_local", None)
|
|
original_sandboxes_attr = getattr(sandbox_package, "sandboxes", None)
|
|
had_sandboxes_attr = hasattr(sandbox_package, "sandboxes")
|
|
|
|
if had_sandboxes_attr:
|
|
delattr(sandbox_package, "sandboxes")
|
|
monkeypatch.setattr(sys, "platform", "win32")
|
|
|
|
try:
|
|
sandboxes = importlib.import_module("agents.sandbox.sandboxes")
|
|
|
|
assert sandboxes.__name__ == "agents.sandbox.sandboxes"
|
|
assert "UnixLocalSandboxClient" not in sandboxes.__all__
|
|
assert "UnixLocalSandboxClient" not in sandboxes.__dict__
|
|
assert "agents.sandbox.sandboxes.unix_local" not in sys.modules
|
|
finally:
|
|
_restore_module("agents.sandbox.sandboxes", original_sandboxes_module)
|
|
_restore_module("agents.sandbox.sandboxes.unix_local", original_unix_local_module)
|
|
_restore_attr(
|
|
sandbox_package,
|
|
"sandboxes",
|
|
original_sandboxes_attr,
|
|
had_sandboxes_attr,
|
|
)
|
|
|
|
|
|
def test_unix_local_backend_import_raises_clear_error_on_windows(monkeypatch) -> None:
|
|
parent = importlib.import_module("agents.sandbox.sandboxes")
|
|
original_unix_local_module = sys.modules.pop("agents.sandbox.sandboxes.unix_local", None)
|
|
original_unix_local_attr = getattr(parent, "unix_local", None)
|
|
had_unix_local_attr = hasattr(parent, "unix_local")
|
|
|
|
if had_unix_local_attr:
|
|
delattr(parent, "unix_local")
|
|
monkeypatch.setattr(sys, "platform", "win32")
|
|
|
|
try:
|
|
with pytest.raises(ImportError, match="not supported on Windows"):
|
|
importlib.import_module("agents.sandbox.sandboxes.unix_local")
|
|
finally:
|
|
_restore_module("agents.sandbox.sandboxes.unix_local", original_unix_local_module)
|
|
_restore_attr(
|
|
parent,
|
|
"unix_local",
|
|
original_unix_local_attr,
|
|
had_unix_local_attr,
|
|
)
|
|
|
|
|
|
@pytest.mark.skipif(sys.platform == "win32", reason="Unix local sandbox is unavailable on Windows")
|
|
def test_sandboxes_package_exports_unix_local_on_supported_platforms() -> None:
|
|
sandboxes = importlib.import_module("agents.sandbox.sandboxes")
|
|
|
|
assert "UnixLocalSandboxClient" in sandboxes.__all__
|
|
assert sandboxes.UnixLocalSandboxClient.__name__ == "UnixLocalSandboxClient"
|