mirror of https://github.com/tiangolo/fastapi.git
feat: Speed up tests
This commit is contained in:
parent
8032e21418
commit
8e62206a78
|
|
@ -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
|
||||
):
|
||||
|
|
|
|||
|
|
@ -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()
|
||||
|
|
|
|||
|
|
@ -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()
|
||||
|
|
|
|||
Loading…
Reference in New Issue