61 lines
2 KiB
Python
61 lines
2 KiB
Python
import io
|
|
|
|
import pytest
|
|
|
|
from infrasynth.files.storage import _LocalStorage, get_storage_backend
|
|
|
|
pytestmark = pytest.mark.django_db
|
|
|
|
|
|
class TestLocalStorage:
|
|
def test_save_and_open_roundtrip(self, media_root):
|
|
storage = _LocalStorage({})
|
|
name = storage.save("dir/file.txt", io.BytesIO(b"hello"))
|
|
assert name.endswith("file.txt")
|
|
assert storage.open(name, "rb").read() == b"hello"
|
|
|
|
def test_exists(self, media_root):
|
|
storage = _LocalStorage({})
|
|
name = storage.save("a.txt", io.BytesIO(b"data"))
|
|
assert storage.exists(name)
|
|
assert not storage.exists("nope.txt")
|
|
|
|
def test_delete(self, media_root):
|
|
storage = _LocalStorage({})
|
|
name = storage.save("b.txt", io.BytesIO(b"data"))
|
|
storage.delete(name)
|
|
assert not storage.exists(name)
|
|
|
|
def test_url(self, media_root):
|
|
storage = _LocalStorage({})
|
|
name = storage.save("c.txt", io.BytesIO(b"data"))
|
|
url = storage.url(name)
|
|
assert "media" in url
|
|
assert url.endswith("c.txt")
|
|
|
|
def test_signed_url_equals_url_for_local(self, media_root):
|
|
storage = _LocalStorage({})
|
|
name = storage.save("d.txt", io.BytesIO(b"data"))
|
|
assert storage.generate_signed_url(name, 60) == storage.url(name)
|
|
|
|
|
|
class TestGetStorageBackend:
|
|
def test_default_is_local(self):
|
|
storage = get_storage_backend()
|
|
assert isinstance(storage, _LocalStorage)
|
|
|
|
def test_unknown_backend_falls_back_to_local(self):
|
|
storage = get_storage_backend("nonexistent")
|
|
assert isinstance(storage, _LocalStorage)
|
|
|
|
def test_local_backend_instantiation(self):
|
|
assert isinstance(get_storage_backend("local"), _LocalStorage)
|
|
|
|
|
|
class TestSaveFile:
|
|
def test_save_file_returns_key(self, media_root):
|
|
from infrasynth.files.storage import save_file
|
|
|
|
key = save_file(io.BytesIO(b"content"), "uploads/x.txt", backend="local")
|
|
assert key.endswith("x.txt")
|
|
assert (media_root / key).exists()
|