wehub-resource-sync
69 lines
2.0 KiB
Python
69 lines
2.0 KiB
Python
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")
|