wehub-resource-sync
65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
from opensquilla.tools.builtin.code_exec import _append_code_exec_sandbox_network_hint
|
|
from opensquilla.tools.builtin.shell import (
|
|
_SANDBOX_NETWORK_DISABLED_HINT,
|
|
_SANDBOX_NETWORK_HINT,
|
|
_append_sandbox_network_hint,
|
|
)
|
|
|
|
|
|
def test_sandbox_network_hint_is_appended_to_dns_failures() -> None:
|
|
output = "curl: (6) Could not resolve host: export.arxiv.org\n"
|
|
|
|
hinted = _append_sandbox_network_hint(output)
|
|
|
|
assert _SANDBOX_NETWORK_HINT in hinted
|
|
assert "sandbox_network" in hinted
|
|
assert "managed proxy" in hinted
|
|
assert "http_request" not in hinted
|
|
assert "web_fetch" not in hinted
|
|
|
|
|
|
def test_sandbox_network_hint_is_not_duplicated() -> None:
|
|
output = f"getaddrinfo failed\n{_SANDBOX_NETWORK_HINT}\n"
|
|
|
|
hinted = _append_sandbox_network_hint(output)
|
|
|
|
assert hinted.count(_SANDBOX_NETWORK_HINT) == 1
|
|
|
|
|
|
def test_sandbox_network_hint_names_disabled_network_default(tmp_path) -> None:
|
|
from opensquilla.sandbox.config import SandboxSettings
|
|
from opensquilla.sandbox.integration import configure_runtime, reset_runtime
|
|
|
|
configure_runtime(
|
|
SandboxSettings(
|
|
run_mode="trusted",
|
|
backend="noop",
|
|
allow_legacy_mode=True,
|
|
network_default="none",
|
|
),
|
|
workspace=tmp_path,
|
|
)
|
|
try:
|
|
hinted = _append_sandbox_network_hint("curl: (6) Could not resolve host: example.com\n")
|
|
finally:
|
|
reset_runtime()
|
|
|
|
assert _SANDBOX_NETWORK_DISABLED_HINT in hinted
|
|
assert 'network_default = "proxy_allowlist"' in hinted
|
|
assert "restart the gateway" in hinted
|
|
|
|
|
|
def test_code_exec_hint_uses_combined_stdout_and_stderr() -> None:
|
|
stdout = (
|
|
"urllib.error.URLError: "
|
|
"<urlopen error [Errno -3] Temporary failure in name resolution>\n"
|
|
)
|
|
stderr = "cleanup warning\n"
|
|
|
|
hinted = _append_code_exec_sandbox_network_hint(stdout=stdout, stderr=stderr)
|
|
|
|
assert "cleanup warning" in hinted
|
|
assert _SANDBOX_NETWORK_HINT in hinted
|