mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-09-30 09:13:14 +00:00
71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
|
import importlib.util
|
||
|
|
import sys
|
||
|
|
import types
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
ROOT = Path(__file__).resolve().parents[1]
|
||
|
|
|
||
|
|
|
||
|
|
class _FakeCreateSandboxFromSnapshotParams:
|
||
|
|
def __init__(self, *, snapshot: str):
|
||
|
|
self.snapshot = snapshot
|
||
|
|
|
||
|
|
|
||
|
|
class _FakeDaytonaConfig:
|
||
|
|
def __init__(self, *, api_key: str):
|
||
|
|
self.api_key = api_key
|
||
|
|
|
||
|
|
|
||
|
|
class _FakeDaytonaSandbox:
|
||
|
|
def __init__(self, *, sandbox):
|
||
|
|
self.sandbox = sandbox
|
||
|
|
|
||
|
|
|
||
|
|
def _load_daytona_module(monkeypatch):
|
||
|
|
fake_daytona = types.ModuleType("daytona")
|
||
|
|
fake_daytona.CreateSandboxFromSnapshotParams = _FakeCreateSandboxFromSnapshotParams
|
||
|
|
fake_daytona.DaytonaConfig = _FakeDaytonaConfig
|
||
|
|
fake_daytona.Daytona = object
|
||
|
|
|
||
|
|
fake_langchain_daytona = types.ModuleType("langchain_daytona")
|
||
|
|
fake_langchain_daytona.DaytonaSandbox = _FakeDaytonaSandbox
|
||
|
|
|
||
|
|
monkeypatch.setitem(sys.modules, "daytona", fake_daytona)
|
||
|
|
monkeypatch.setitem(sys.modules, "langchain_daytona", fake_langchain_daytona)
|
||
|
|
module_path = ROOT / "agent" / "integrations" / "daytona.py"
|
||
|
|
spec = importlib.util.spec_from_file_location("daytona_under_test", module_path)
|
||
|
|
assert spec is not None and spec.loader is not None
|
||
|
|
module = importlib.util.module_from_spec(spec)
|
||
|
|
spec.loader.exec_module(module)
|
||
|
|
return module
|
||
|
|
|
||
|
|
|
||
|
|
def test_daytona_params_default_to_existing_snapshot(monkeypatch):
|
||
|
|
monkeypatch.delenv("DAYTONA_SANDBOX_SNAPSHOT", raising=False)
|
||
|
|
module = _load_daytona_module(monkeypatch)
|
||
|
|
|
||
|
|
params = module._get_daytona_sandbox_params()
|
||
|
|
|
||
|
|
assert params.snapshot == "daytonaio/sandbox:0.6.0"
|
||
|
|
|
||
|
|
|
||
|
|
def test_daytona_params_use_env_snapshot(monkeypatch):
|
||
|
|
monkeypatch.setenv("DAYTONA_SANDBOX_SNAPSHOT", "custom/snapshot:1.0")
|
||
|
|
module = _load_daytona_module(monkeypatch)
|
||
|
|
|
||
|
|
params = module._get_daytona_sandbox_params()
|
||
|
|
|
||
|
|
assert params.snapshot == "custom/snapshot:1.0"
|
||
|
|
|
||
|
|
|
||
|
|
def test_daytona_params_reject_empty_snapshot(monkeypatch):
|
||
|
|
monkeypatch.setenv("DAYTONA_SANDBOX_SNAPSHOT", " ")
|
||
|
|
module = _load_daytona_module(monkeypatch)
|
||
|
|
|
||
|
|
try:
|
||
|
|
module._get_daytona_sandbox_params()
|
||
|
|
except ValueError as exc:
|
||
|
|
assert "DAYTONA_SANDBOX_SNAPSHOT must not be empty" in str(exc)
|
||
|
|
else:
|
||
|
|
raise AssertionError("expected empty Daytona snapshot to fail")
|