diff --git a/tests/test_tutorial/test_security/test_tutorial005.py b/tests/test_tutorial/test_security/test_tutorial005.py index 76b08860f5..563dbfc16d 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 @@ -16,6 +17,7 @@ from ...utils import needs_py310 pytest.param("tutorial005_an_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}") @@ -23,6 +25,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 aec20e42e1..8a41f2a0d4 100644 --- a/tests/test_tutorial/test_sql_databases/test_tutorial001.py +++ b/tests/test_tutorial/test_sql_databases/test_tutorial001.py @@ -27,6 +27,7 @@ def clear_sqlmodel(): pytest.param("tutorial001_an_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): # 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 diff --git a/tests/test_tutorial/test_sql_databases/test_tutorial002.py b/tests/test_tutorial/test_sql_databases/test_tutorial002.py index 4ea7d5f647..7ada7eb161 100644 --- a/tests/test_tutorial/test_sql_databases/test_tutorial002.py +++ b/tests/test_tutorial/test_sql_databases/test_tutorial002.py @@ -27,6 +27,7 @@ def clear_sqlmodel(): pytest.param("tutorial002_an_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): # 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