wehub-resource-sync
148 lines
4.4 KiB
Python
148 lines
4.4 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
from pathlib import Path
|
|
|
|
from agents.sandbox.snapshot import LocalSnapshotSpec
|
|
from agents.sandbox.snapshot_defaults import (
|
|
_DEFAULT_LOCAL_SNAPSHOT_TTL_SECONDS,
|
|
cleanup_stale_default_local_snapshots,
|
|
default_local_snapshot_base_dir,
|
|
resolve_default_local_snapshot_spec,
|
|
)
|
|
|
|
|
|
def test_default_local_snapshot_base_dir_uses_xdg_state_home(tmp_path: Path) -> None:
|
|
state_home = tmp_path / "state"
|
|
result = default_local_snapshot_base_dir(
|
|
home=tmp_path / "home",
|
|
env={"XDG_STATE_HOME": str(state_home)},
|
|
platform="linux",
|
|
os_name="posix",
|
|
)
|
|
|
|
assert result == state_home / "openai-agents-python" / "sandbox" / "snapshots"
|
|
|
|
|
|
def test_default_local_snapshot_base_dir_uses_macos_application_support(tmp_path: Path) -> None:
|
|
home = tmp_path / "home"
|
|
result = default_local_snapshot_base_dir(
|
|
home=home,
|
|
env={},
|
|
platform="darwin",
|
|
os_name="posix",
|
|
)
|
|
|
|
assert (
|
|
result
|
|
== home
|
|
/ "Library"
|
|
/ "Application Support"
|
|
/ "openai-agents-python"
|
|
/ "sandbox"
|
|
/ "snapshots"
|
|
)
|
|
|
|
|
|
def test_default_local_snapshot_base_dir_uses_localappdata_on_windows(tmp_path: Path) -> None:
|
|
local_app_data = Path(r"C:\Users\me\AppData\Local")
|
|
result = default_local_snapshot_base_dir(
|
|
home=tmp_path / "home",
|
|
env={"LOCALAPPDATA": str(local_app_data)},
|
|
platform="win32",
|
|
os_name="nt",
|
|
)
|
|
|
|
assert result == local_app_data / "openai-agents-python" / "sandbox" / "snapshots"
|
|
|
|
|
|
def test_default_local_snapshot_base_dir_uses_absolute_appdata_when_localappdata_is_relative(
|
|
tmp_path: Path,
|
|
) -> None:
|
|
app_data = Path(r"C:\Users\me\AppData\Roaming")
|
|
result = default_local_snapshot_base_dir(
|
|
home=tmp_path / "home",
|
|
env={"LOCALAPPDATA": "relative-local", "APPDATA": str(app_data)},
|
|
platform="win32",
|
|
os_name="nt",
|
|
)
|
|
|
|
assert result == app_data / "openai-agents-python" / "sandbox" / "snapshots"
|
|
|
|
|
|
def test_default_local_snapshot_base_dir_ignores_relative_windows_env_paths(
|
|
tmp_path: Path,
|
|
) -> None:
|
|
home = tmp_path / "home"
|
|
result = default_local_snapshot_base_dir(
|
|
home=home,
|
|
env={"LOCALAPPDATA": "relative-local", "APPDATA": "relative-roaming"},
|
|
platform="win32",
|
|
os_name="nt",
|
|
)
|
|
|
|
assert result == home / "AppData" / "Local" / "openai-agents-python" / "sandbox" / "snapshots"
|
|
|
|
|
|
def test_default_local_snapshot_base_dir_ignores_posix_absolute_localappdata_on_windows(
|
|
tmp_path: Path,
|
|
) -> None:
|
|
home = tmp_path / "home"
|
|
result = default_local_snapshot_base_dir(
|
|
home=home,
|
|
env={"LOCALAPPDATA": "/tmp/localappdata"},
|
|
platform="win32",
|
|
os_name="nt",
|
|
)
|
|
|
|
assert result == home / "AppData" / "Local" / "openai-agents-python" / "sandbox" / "snapshots"
|
|
|
|
|
|
def test_cleanup_stale_default_local_snapshots_removes_only_old_tar_files(tmp_path: Path) -> None:
|
|
managed_dir = tmp_path / "snapshots"
|
|
managed_dir.mkdir()
|
|
stale = managed_dir / "stale.tar"
|
|
fresh = managed_dir / "fresh.tar"
|
|
keep = managed_dir / "keep.txt"
|
|
stale.write_bytes(b"stale")
|
|
fresh.write_bytes(b"fresh")
|
|
keep.write_text("keep")
|
|
|
|
now = 2_000_000_000.0
|
|
stale_mtime = now - (_DEFAULT_LOCAL_SNAPSHOT_TTL_SECONDS + 60)
|
|
fresh_mtime = now - 60
|
|
os.utime(stale, (stale_mtime, stale_mtime))
|
|
os.utime(fresh, (fresh_mtime, fresh_mtime))
|
|
|
|
cleanup_stale_default_local_snapshots(managed_dir, now=now)
|
|
|
|
assert not stale.exists()
|
|
assert fresh.exists()
|
|
assert keep.exists()
|
|
|
|
|
|
def test_resolve_default_local_snapshot_spec_keeps_existing_stale_files(
|
|
tmp_path: Path,
|
|
) -> None:
|
|
state_home = tmp_path / "state"
|
|
managed_dir = state_home / "openai-agents-python" / "sandbox" / "snapshots"
|
|
managed_dir.mkdir(parents=True)
|
|
stale = managed_dir / "stale.tar"
|
|
stale.write_bytes(b"stale")
|
|
now = 2_000_000_000.0
|
|
stale_mtime = now - (_DEFAULT_LOCAL_SNAPSHOT_TTL_SECONDS + 60)
|
|
os.utime(stale, (stale_mtime, stale_mtime))
|
|
|
|
spec = resolve_default_local_snapshot_spec(
|
|
home=tmp_path / "home",
|
|
env={"XDG_STATE_HOME": str(state_home)},
|
|
platform="linux",
|
|
os_name="posix",
|
|
now=now,
|
|
)
|
|
|
|
assert isinstance(spec, LocalSnapshotSpec)
|
|
assert spec.base_path == managed_dir
|
|
assert managed_dir.exists()
|
|
assert stale.exists()
|