1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42
|
from pathlib import Path
import pytest
from _pytest.monkeypatch import MonkeyPatch
from litestar.config.compression import CompressionConfig
from litestar.utils.module_loader import import_string, module_to_os_path
def test_import_string() -> None:
cls = import_string("litestar.config.compression.CompressionConfig")
assert type(cls) == type(CompressionConfig)
with pytest.raises(ImportError):
_ = import_string("CompressionConfigNew")
_ = import_string("litestar.config.compression.CompressionConfigNew")
_ = import_string("imaginary_module_that_doesnt_exist.Config") # a random nonexistent class
def test_module_path(tmp_path: Path, monkeypatch: MonkeyPatch) -> None:
the_path = module_to_os_path("litestar.config.compression")
assert the_path.exists()
tmp_path.joinpath("simple_module.py").write_text("x = 'foo'")
monkeypatch.syspath_prepend(tmp_path)
os_path = module_to_os_path("simple_module")
assert os_path == Path(tmp_path)
with pytest.raises(TypeError):
_ = module_to_os_path("litestar.config.compression.Config")
_ = module_to_os_path("litestar.config.compression.extra.module")
def test_import_non_existing_attribute_raises() -> None:
with pytest.raises(ImportError):
import_string("litestar.app.some_random_string")
def test_import_string_cached(tmp_path: Path, monkeypatch: MonkeyPatch) -> None:
tmp_path.joinpath("testmodule.py").write_text("x = 'foo'")
monkeypatch.chdir(tmp_path)
monkeypatch.syspath_prepend(tmp_path)
assert import_string("testmodule.x") == "foo"
|