chore: import upstream snapshot with attribution

2026-07-13 12:39:17 +08:00
commit 4ed4e9ff99
1368 changed files with 334957 additions and 0 deletions
@@ -0,0 +1,147 @@
+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()