Pytest-django: Регистрация типа базы данных после создания тестовых баз данных

У меня есть поле модели django, которое регистрирует тип базы данных:

class FieldMeta(type):
    def __init__(cls, name, bases, clsdict):
        super().__init__(name, bases, clsdict)
        cls.register_type()

    def register_type(cls):
        db_type = cls().db_type(connection)
        cursor = connection.cursor()
        try:
            cursor.execute(
                "CREATE TYPE foo AS (first integer, second integer);"
            )
            _logger.warn("creating type")
        except ProgrammingError:
            pass

        cls.python_type = register_composite(
            db_type, connection.cursor().cursor, globally=True
        ).type
        _logger.warn("registering composite")

        def adapt_composite(composite):
            return AsIs(...get sql...)

        register_adapter(Foo, adapt_composite)

class FooField(models.Field, metaclass=FieldMeta):
    def db_type(self, connection):
        return "foo"

Это отлично работает в таких ситуациях:

• База данных уже знает о типе foo
• База данных не знает о типе foo

Проблема в том, что pytest-django, похоже, запускает эти методы init перед созданием тестовой базы данных, что означает, что тестовая база данных не имеет требуемого типа. Я могу исправить это на отдельных тестах, вызывая FooField.register_type() в методе setUpClass теста, но это не идеально, так как он должен присутствовать всегда.

Я попробовал переопределить приспособление django_db_setup, чтобы оно выглядело следующим образом:

@pytest.fixture(scope="session")
def django_db_setup(
    request,
    django_test_environment: None,
    django_db_blocker,
    django_db_use_migrations: bool,
    django_db_keepdb: bool,
    django_db_createdb: bool,
    django_db_modify_db_settings: None,
) -> None:
    """Top level fixture to ensure test databases are available"""
    from django.test.utils import setup_databases, teardown_databases

    setup_databases_args = {}

    if django_db_keepdb and not django_db_createdb:
        setup_databases_args["keepdb"] = True

    with django_db_blocker.unblock():
        db_cfg = setup_databases(
            verbosity=request.config.option.verbose,
            interactive=False,
            **setup_databases_args
        )
        FooField.register_type() # <----------------------------

    def teardown_database() -> None:
        with django_db_blocker.unblock():
            try:
                teardown_databases(db_cfg, verbosity=request.config.option.verbose)
            except Exception as exc:
                request.node.warn(
                    pytest.PytestWarning(
                        "Error when trying to teardown test databases: %r" % exc
                    )
                )

    if not django_db_keepdb:
        request.addfinalizer(teardown_database)

Но это не дало никакого эффекта. Я также попробовал поставить точку останова вместо этой строки (она никогда не вызывалась) и поставить точку останова перед with django_db_blocker.unblock(): (которая была вызвана, так что conftest.py определенно вызывается).

В общем, вопрос в том, как заставить pytest-django запустить этот метод послесоздания тестовой базы данных?

Вернуться на верх