feat: Speed up tests

This commit is contained in:
Postnov Roman 2025-04-05 10:02:10 +03:00
parent 8032e21418
commit 8e62206a78
3 changed files with 18 additions and 0 deletions

View File

@ -1,4 +1,5 @@
import importlib
from functools import lru_cache
from types import ModuleType
import pytest
@ -18,6 +19,7 @@ from ...utils import needs_py39, needs_py310
pytest.param("tutorial005_an_py39", marks=needs_py39),
pytest.param("tutorial005_an_py310", marks=needs_py310),
],
scope="module",
)
def get_mod(request: pytest.FixtureRequest):
mod = importlib.import_module(f"docs_src.security.{request.param}")
@ -25,6 +27,20 @@ def get_mod(request: pytest.FixtureRequest):
return mod
@pytest.fixture(scope="module", autouse=True)
def cache_verify_password(mod: ModuleType):
assert hasattr(mod, "verify_password"), (
f"Module {mod.__name__} does not have attribute 'verify_password'"
)
original_func = mod.verify_password
cached_func = lru_cache()(original_func)
mod.verify_password = cached_func
yield
mod.verify_password = original_func
def get_access_token(
*, username="johndoe", password="secret", scope=None, client: TestClient
):

View File

@ -29,6 +29,7 @@ def clear_sqlmodel():
pytest.param("tutorial001_an_py39", marks=needs_py39),
pytest.param("tutorial001_an_py310", marks=needs_py310),
],
scope="module",
)
def get_client(request: pytest.FixtureRequest):
clear_sqlmodel()

View File

@ -29,6 +29,7 @@ def clear_sqlmodel():
pytest.param("tutorial002_an_py39", marks=needs_py39),
pytest.param("tutorial002_an_py310", marks=needs_py310),
],
scope="module",
)
def get_client(request: pytest.FixtureRequest):
clear_sqlmodel()