chore: import upstream snapshot with attribution

2026-07-13 12:47:58 +08:00
commit b16403ea71
789 changed files with 115226 additions and 0 deletions
@@ -0,0 +1,84 @@
+import httpx
+
+from e2b.envd.api import (
+    format_envd_api_exception,
+    handle_envd_api_transport_exception,
+)
+from e2b.exceptions import (
+    AuthenticationException,
+    InvalidArgumentException,
+    NotEnoughSpaceException,
+    NotFoundException,
+    RateLimitException,
+    SandboxException,
+    TimeoutException,
+)
+
+
+def test_maps_400_to_invalid_argument():
+    err = format_envd_api_exception(400, "Bad request")
+    assert isinstance(err, InvalidArgumentException)
+
+
+def test_maps_401_to_authentication():
+    err = format_envd_api_exception(401, "Invalid token")
+    assert isinstance(err, AuthenticationException)
+
+
+def test_maps_404_to_not_found():
+    err = format_envd_api_exception(404, "Not found")
+    assert isinstance(err, NotFoundException)
+
+
+def test_maps_429_to_rate_limit():
+    err = format_envd_api_exception(429, "Too many requests")
+    assert isinstance(err, RateLimitException)
+    assert "rate limited" in str(err)
+
+
+def test_maps_502_to_timeout():
+    err = format_envd_api_exception(502, "Bad gateway")
+    assert isinstance(err, TimeoutException)
+
+
+def test_maps_507_to_not_enough_space():
+    err = format_envd_api_exception(507, "No space left")
+    assert isinstance(err, NotEnoughSpaceException)
+
+
+def test_falls_back_to_sandbox_exception():
+    err = format_envd_api_exception(500, "Internal error")
+    assert isinstance(err, SandboxException)
+    assert "500" in str(err)
+
+
+def test_returns_raw_remote_protocol_error_without_health_result():
+    original = httpx.RemoteProtocolError("peer closed connection")
+    err = handle_envd_api_transport_exception(original)
+    assert err is original
+
+
+def test_returns_raw_network_errors():
+    original = httpx.ReadError("read failed")
+    err = handle_envd_api_transport_exception(original)
+    assert err is original
+
+
+def test_returns_original_when_not_transport_error():
+    original = ValueError("not transport")
+    err = handle_envd_api_transport_exception(original)
+    assert err is original
+
+
+def test_health_result_confirms_sandbox_killed():
+    err = handle_envd_api_transport_exception(
+        httpx.RemoteProtocolError("peer closed connection"), sandbox_running=False
+    )
+    assert isinstance(err, TimeoutException)
+    assert "sandbox was killed or reached its end of life" in str(err)
+
+
+def test_health_result_running_returns_raw_error():
+    original = httpx.RemoteProtocolError("peer closed connection")
+    err = handle_envd_api_transport_exception(original, sandbox_running=True)
+    assert err is original