How to mock ormar_config in unit tests for 0.20.x

[schnerd]

I have a project that uses fastapi + ormar + pytest. To keep tests isolated in unit tests, I implement a pytest fixture that patches the ormar BaseMeta object to have a new SQLite database for each test. However this approach no longer works reliably with ormar 0.20.0 because of the way each model has it's own separate OrmarConfig (due to OrmarConfig.copy behavior)

Here's the code for the current approach:

import databases
import sqlalchemy
import alembic
from project.database import db as db_module

@pytest.fixture(autouse=True, scope="function")
def db_isolation(mocker):
    """Create a new database per unit test and run all migrations."""
    test_db_file = f"/tmp/test_{uuid.uuid4().hex}.db"
    try:
        url = f"sqlite:///{test_db_file}"
        test_db = databases.Database(url)
        test_metadata = sqlalchemy.MetaData()
        mocker.patch.object(db_module.BaseMeta, "metadata", test_metadata)
        mocker.patch.object(db_module.BaseMeta, "database", test_db)

        migrations_dir = os.path.join(PROJECT_DIR, "database/migrations")
        alembic_config_file = os.path.join(migrations_dir, "alembic.ini")
        cfg = alembic.config.Config(alembic_config_file)
        cfg.set_main_option("sqlalchemy.url", url)
        cfg.set_main_option("script_location", migrations_dir)
        alembic.command.upgrade(cfg, "head")
        yield
    finally:
        if os.path.exists(test_db_file):
            os.remove(test_db_file)

In 0.20.0 there is no longer a centralized BaseMeta I can patch, instead there is a OrmarConfig and each model gets a separate instance due to OrmarConfig.copy. Does anyone have recommendations on the most reliable way for each test to get it's own database/metadata instances across all models in the newer 0.20.0 paradigm?

[schnerd]

Seems like the following solution works, but still curious if anyone has something better!

        for Model in project.db_models.__dict__.values():
            if hasattr(Model, "ormar_config"):
                mocker.patch.object(Model.ormar_config, "metadata", test_metadata)
                mocker.patch.object(Model.ormar_config, "database", test_db)