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,68 @@
+import httpx
+import pytest
+
+from e2b import Volume
+from e2b.exceptions import NotFoundException
+import e2b.volume.volume_sync as volume_sync_mod
+
+_files = {
+    "hello.txt": b"hello world",
+    "empty.txt": b"",
+}
+
+
+def _handler(request: httpx.Request) -> httpx.Response:
+    path = request.url.params.get("path")
+    content = _files.get(path)
+    if content is None:
+        return httpx.Response(404)
+    return httpx.Response(200, content=content)
+
+
+@pytest.fixture
+def volume(monkeypatch) -> Volume:
+    real_get_api_client = volume_sync_mod.get_volume_api_client
+
+    def mock_get_api_client(config, **kwargs):
+        client = real_get_api_client(config, **kwargs)
+        client.set_httpx_client(
+            httpx.Client(
+                base_url=config.api_url,
+                transport=httpx.MockTransport(_handler),
+            )
+        )
+        return client
+
+    monkeypatch.setattr(volume_sync_mod, "get_volume_api_client", mock_get_api_client)
+    return Volume(volume_id="vol-1", name="test-volume", token="vol-token")
+
+
+def test_read_file_stream_yields_content(volume: Volume):
+    stream = volume.read_file("hello.txt", format="stream")
+    assert b"".join(stream) == b"hello world"
+
+
+def test_read_file_stream_raises_for_missing_path(volume: Volume):
+    with pytest.raises(NotFoundException):
+        for _ in volume.read_file("missing.txt", format="stream"):
+            pass
+
+
+def test_read_file_stream_of_empty_file(volume: Volume):
+    stream = volume.read_file("empty.txt", format="stream")
+    assert b"".join(stream) == b""
+
+
+def test_read_file_text_and_bytes(volume: Volume):
+    assert volume.read_file("hello.txt") == "hello world"
+    assert volume.read_file("hello.txt", format="bytes") == b"hello world"
+
+
+def test_read_file_text_and_bytes_of_empty_file(volume: Volume):
+    assert volume.read_file("empty.txt") == ""
+    assert volume.read_file("empty.txt", format="bytes") == b""
+
+
+def test_read_file_missing_path_raises(volume: Volume):
+    with pytest.raises(NotFoundException):
+        volume.read_file("missing.txt")