mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-10-01 02:43:13 +00:00
Adam Moussa
ae1f883b4c
Applies the plan's C5 step: git mv every test per the domain-reorg
move-map (movemap-m50.txt) into tests/{agent,analyzer,auth,dashboard,
github,middleware,models,reviewer,sandbox,slack,tools,webhooks}/, plus
the 13 fork-only placements from the scoping report §2c (Atlassian
webhook tests -> tests/webhooks/, test_atlassian_connect.py and
test_auth_error_leak.py -> tests/auth/, jira/confluence util tests ->
tests/tools/, test_repo_binding_isolation.py -> tests/sandbox/,
bot-identity/autofix tests -> tests/github/).
Path-only move: the only content edits are parents[1] -> parents[2]
fixes in test_e2b_integration.py and test_daytona_integration.py,
required because their __file__-relative ROOT path gained one more
directory level in the move.
Monkeypatch retargets for these files were already completed in C4;
none remained outstanding here.
99 lines
2.9 KiB
Python
99 lines
2.9 KiB
Python
"""Tests for LangSmith sandbox env-var configuration parsing."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from agent.integrations.langsmith import (
|
|
DEFAULT_SANDBOX_DELETE_AFTER_STOP_SECONDS,
|
|
DEFAULT_SANDBOX_IDLE_TTL_SECONDS,
|
|
DEFAULT_SANDBOX_MEM_BYTES,
|
|
DEFAULT_SANDBOX_VCPUS,
|
|
DEFAULT_SNAPSHOT_FS_CAPACITY_BYTES,
|
|
LangSmithProvider,
|
|
_get_sandbox_snapshot_config,
|
|
)
|
|
|
|
|
|
def test_defaults_when_env_unset() -> None:
|
|
with patch.dict(
|
|
"os.environ",
|
|
{"DEFAULT_SANDBOX_SNAPSHOT_ID": "snap-1"},
|
|
clear=True,
|
|
):
|
|
snapshot_id, fs, vcpus, mem, idle, delete_after = _get_sandbox_snapshot_config()
|
|
assert snapshot_id == "snap-1"
|
|
assert fs == DEFAULT_SNAPSHOT_FS_CAPACITY_BYTES
|
|
assert vcpus == DEFAULT_SANDBOX_VCPUS
|
|
assert mem == DEFAULT_SANDBOX_MEM_BYTES
|
|
assert idle == DEFAULT_SANDBOX_IDLE_TTL_SECONDS
|
|
assert delete_after == DEFAULT_SANDBOX_DELETE_AFTER_STOP_SECONDS
|
|
|
|
|
|
def test_overrides_from_env() -> None:
|
|
with patch.dict(
|
|
"os.environ",
|
|
{
|
|
"DEFAULT_SANDBOX_SNAPSHOT_ID": "snap-2",
|
|
"DEFAULT_SANDBOX_IDLE_TTL_SECONDS": "120",
|
|
"DEFAULT_SANDBOX_DELETE_AFTER_STOP_SECONDS": "3600",
|
|
},
|
|
clear=True,
|
|
):
|
|
_, _, _, _, idle, delete_after = _get_sandbox_snapshot_config()
|
|
assert idle == 120
|
|
assert delete_after == 3600
|
|
|
|
|
|
def test_zero_disables_ttls() -> None:
|
|
with patch.dict(
|
|
"os.environ",
|
|
{
|
|
"DEFAULT_SANDBOX_SNAPSHOT_ID": "snap-3",
|
|
"DEFAULT_SANDBOX_IDLE_TTL_SECONDS": "0",
|
|
"DEFAULT_SANDBOX_DELETE_AFTER_STOP_SECONDS": "0",
|
|
},
|
|
clear=True,
|
|
):
|
|
_, _, _, _, idle, delete_after = _get_sandbox_snapshot_config()
|
|
assert idle == 0
|
|
assert delete_after == 0
|
|
|
|
|
|
def test_validate_startup_rejects_non_integer_ttl() -> None:
|
|
with patch.dict(
|
|
"os.environ",
|
|
{
|
|
"DEFAULT_SANDBOX_SNAPSHOT_ID": "snap-4",
|
|
"DEFAULT_SANDBOX_IDLE_TTL_SECONDS": "not-a-number",
|
|
},
|
|
clear=True,
|
|
):
|
|
with pytest.raises(ValueError, match="DEFAULT_SANDBOX_IDLE_TTL_SECONDS"):
|
|
LangSmithProvider.validate_startup_config()
|
|
|
|
|
|
def test_validate_startup_rejects_negative_ttl() -> None:
|
|
with patch.dict(
|
|
"os.environ",
|
|
{
|
|
"DEFAULT_SANDBOX_SNAPSHOT_ID": "snap-5",
|
|
"DEFAULT_SANDBOX_DELETE_AFTER_STOP_SECONDS": "-1",
|
|
},
|
|
clear=True,
|
|
):
|
|
with pytest.raises(ValueError, match=">= 0"):
|
|
LangSmithProvider.validate_startup_config()
|
|
|
|
|
|
def test_validate_startup_accepts_valid_config() -> None:
|
|
with patch.dict(
|
|
"os.environ",
|
|
{
|
|
"DEFAULT_SANDBOX_SNAPSHOT_ID": "snap-6",
|
|
"DEFAULT_SANDBOX_IDLE_TTL_SECONDS": "1800",
|
|
"DEFAULT_SANDBOX_DELETE_AFTER_STOP_SECONDS": "86400",
|
|
},
|
|
clear=True,
|
|
):
|
|
LangSmithProvider.validate_startup_config()
|