mirror of https://github.com/tiangolo/fastapi.git
Merge 8609193960 into 272204c0c7
This commit is contained in:
commit
2fff077169
|
|
@ -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()
|
||||
|
|
@ -48,6 +49,8 @@ def get_client(request: pytest.FixtureRequest):
|
|||
# Clean up connection explicitly to avoid resource warning
|
||||
mod.engine.dispose()
|
||||
|
||||
mod.engine.dispose()
|
||||
|
||||
|
||||
def test_crud_app(client: TestClient):
|
||||
# TODO: this warns that SQLModel.from_orm is deprecated in Pydantic v1, refactor
|
||||
|
|
|
|||
|
|
@ -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()
|
||||
|
|
@ -48,6 +49,8 @@ def get_client(request: pytest.FixtureRequest):
|
|||
# Clean up connection explicitly to avoid resource warning
|
||||
mod.engine.dispose()
|
||||
|
||||
mod.engine.dispose()
|
||||
|
||||
|
||||
def test_crud_app(client: TestClient):
|
||||
# TODO: this warns that SQLModel.from_orm is deprecated in Pydantic v1, refactor
|
||||
|
|
|
|||
Loading…
Reference in New Issue