generated from coulomb/repo-seed
tegwick
15f031fd65
Add credentialed E2B and Modal extensions, burst routing fallback, fin-hub meter export hook, BYOK docs, and 77 tests.
75 lines
2.5 KiB
Python
75 lines
2.5 KiB
Python
"""Extension routing tests."""
|
|
|
|
|
|
import pytest
|
|
|
|
from sandboxer.extensions.registry import load_extension
|
|
from sandboxer.models import Profile, RouteSpec, RouteStrategy
|
|
from sandboxer.routing.resolver import resolve_extension
|
|
|
|
|
|
def _burst_profile() -> Profile:
|
|
return Profile.model_validate(
|
|
{
|
|
"id": "profile.burst-sandbox",
|
|
"version": "1.0.0",
|
|
"extension": "ext.compose-ssh",
|
|
"route": {
|
|
"strategy": "prefer-self-hosted",
|
|
"extensions": [
|
|
"ext.compose-ssh",
|
|
"ext.e2b",
|
|
"ext.modal",
|
|
"ext.saas-stub",
|
|
],
|
|
},
|
|
}
|
|
)
|
|
|
|
|
|
def test_explicit_uses_profile_extension() -> None:
|
|
profile = Profile.model_validate(
|
|
{
|
|
"id": "profile.compose-e2e",
|
|
"version": "1.0.0",
|
|
"extension": "ext.compose-ssh",
|
|
}
|
|
)
|
|
ext = resolve_extension(profile, {}, host_override="coulombcore")
|
|
assert ext.id == "ext.compose-ssh"
|
|
|
|
|
|
def test_prefer_self_hosted_when_host_set(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setenv("SANDBOXER_HOST", "coulombcore")
|
|
ext = resolve_extension(_burst_profile(), {"repo": "/tmp/x"}, host_override=None)
|
|
assert ext.id == "ext.compose-ssh"
|
|
|
|
|
|
def test_prefer_self_hosted_falls_back_to_saas(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.delenv("SANDBOXER_HOST", raising=False)
|
|
monkeypatch.delenv("E2B_API_KEY", raising=False)
|
|
monkeypatch.delenv("MODAL_TOKEN_ID", raising=False)
|
|
monkeypatch.setenv("SANDBOXER_FORCE_SAAS", "1")
|
|
ext = resolve_extension(_burst_profile(), {}, host_override=None)
|
|
assert ext.id == "ext.saas-stub"
|
|
assert ext.capabilities.pricing_model == "metered"
|
|
|
|
|
|
def test_prefer_self_hosted_falls_back_to_e2b_when_credentialed(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
monkeypatch.setenv("SANDBOXER_FORCE_SAAS", "1")
|
|
monkeypatch.setenv("E2B_API_KEY", "test-key")
|
|
ext = resolve_extension(_burst_profile(), {}, host_override=None)
|
|
assert ext.id == "ext.e2b"
|
|
|
|
|
|
def test_lowest_cost_picks_metered_when_forced(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
profile = _burst_profile()
|
|
profile.route = RouteSpec(
|
|
strategy=RouteStrategy.LOWEST_COST,
|
|
extensions=["ext.compose-ssh", "ext.saas-stub"],
|
|
)
|
|
monkeypatch.setenv("SANDBOXER_FORCE_SAAS", "1")
|
|
ext = resolve_extension(profile, {}, host_override=None)
|
|
assert load_extension(ext.id).capabilities.pricing_model == "metered" |