diff --git a/tests/test_tutorial/test_security/test_tutorial005.py b/tests/test_tutorial/test_security/test_tutorial005.py index ad644d61b..1abfdeb52 100644 --- a/tests/test_tutorial/test_security/test_tutorial005.py +++ b/tests/test_tutorial/test_security/test_tutorial005.py @@ -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 ): diff --git a/tests/test_tutorial/test_sql_databases/test_tutorial001.py b/tests/test_tutorial/test_sql_databases/test_tutorial001.py index cc7e590df..6fcf09357 100644 --- a/tests/test_tutorial/test_sql_databases/test_tutorial001.py +++ b/tests/test_tutorial/test_sql_databases/test_tutorial001.py @@ -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() @@ -46,6 +47,8 @@ def get_client(request: pytest.FixtureRequest): with TestClient(mod.app) as c: yield c + mod.engine.dispose() + def test_crud_app(client: TestClient): # TODO: this warns that SQLModel.from_orm is deprecated in Pydantic v1, refactor diff --git a/tests/test_tutorial/test_sql_databases/test_tutorial002.py b/tests/test_tutorial/test_sql_databases/test_tutorial002.py index 8a98f9a2d..0cf94cce1 100644 --- a/tests/test_tutorial/test_sql_databases/test_tutorial002.py +++ b/tests/test_tutorial/test_sql_databases/test_tutorial002.py @@ -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() @@ -46,6 +47,8 @@ def get_client(request: pytest.FixtureRequest): with TestClient(mod.app) as c: yield c + mod.engine.dispose() + def test_crud_app(client: TestClient): # TODO: this warns that SQLModel.from_orm is deprecated in Pydantic v1, refactor