在pytest为django设置数据库之前安装postgresql扩展

hs1rzwqc 于 2023-04-13 发布在 Go

关注(0)|答案(1)|浏览(151)

我需要为django项目的postgresql数据库安装citext扩展。对于项目本身来说，它通过迁移进行得很顺利，工作得很好，但是我的pytest配置了--no-migrations选项，所以pytest在不运行迁移的情况下创建数据库。我如何让pytest在创建表之前安装citext postgres扩展？目前我正在获取
- django.db.utils.ProgrammingError: type "citext" does not exist
当pytest尝试创建表auth_users时

sql = 'CREATE TABLE "auth_user" ("id" serial NOT NULL PRIMARY KEY, "password" varchar(128) NOT NULL, "last_login" timestamp ...T NULL, "is_active" boolean NOT NULL, "date_joined" timestamp with time zone NOT NULL, "email" citext NOT NULL UNIQUE)', params = None
ignored_wrapper_args = (False, {'connection': <django.contrib.gis.db.backends.postgis.base.DatabaseWrapper object at 0x7fb313bb0100>, 'cursor': <django.db.backends.utils.CursorWrapper object at 0x7fb30d9f8580>})

我尝试使用django_db_setup fixture，但我没有弄清楚如何更改它，因为这样的东西

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

    setup_databases_args = {}

    if not django_db_use_migrations:
        from pytest_django.fixtures import _disable_native_migrations
        _disable_native_migrations()

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

    with django_db_blocker.unblock():
        from django.db import connection
        cursor = connection.cursor()
        cursor.execute("CREATE EXTENSION IF NOT EXISTS citext;")

        db_cfg = setup_databases(
            verbosity=request.config.option.verbose,
            interactive=False,
            **setup_databases_args
        )
    def teardown_database():
        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)

对我没有帮助

django

来源：https://stackoverflow.com/questions/74740640/install-postgresql-extension-before-pytest-set-up-database-for-django

1条答案

按热度按时间

e0bqpujr1#

恼人的是，在设置数据库和加载适当的postgresql扩展之间没有适当的挂钩。你可以通过复制/修改禁止迁移的pytest-django代码并运行你的代码 * 而不是 * 上游代码来解决这个问题。

@pytest.fixture(scope="session")
def django_migration_disabler() -> None:
    """Disable migrations when running django tests.

    This copies/alters the behavior of pytest_django.fixtures._disable_migrations,
    which is called when pytest is invoked with --no-migrations.

    See: https://github.com/pytest-dev/pytest-django/blob/v4.5.2/pytest_django/fixtures.py#L260

    We do this instead of invoking with --no-migrations because constructing the
    database without migrations fails to create necessary postgres extensions like
    citext and uuid-ossp.

    We then override the django_db_setup fixture and ensure that this
    fixture is called before the parent django_db_setup fixture.
    """
    from django.conf import settings
    from django.core.management.commands import migrate
    from django.db import connections

    class DisableMigrations:
        def __contains__(self, item: str) -> bool:
            return True

        def __getitem__(self, item: str) -> None:
            return None

    settings.MIGRATION_MODULES = DisableMigrations()

    class MigrateSilentCommand(migrate.Command):
        def handle(self, *args, **options):
            options["verbosity"] = 0
            database = options["database"]
            connection = connections[database]
            with connection.cursor() as cursor:
                cursor.execute('CREATE EXTENSION IF NOT EXISTS "citext";')
            return super().handle(*args, **options)

    migrate.Command = MigrateSilentCommand  # type: ignore

@pytest.fixture(scope="session")
def django_db_use_migrations() -> bool:
    """Force pytest-django to use migrations.

    This is necessary because we disable migrations in the django_migration_disabler
    and the existing pytest-django mechanisms would override our disabling mechanism
    with their own, which would fail to create the necessary extensions in postgres.
    """
    return True

@pytest.fixture(scope="session")
def django_db_setup(django_migration_disabler, django_db_setup, django_db_blocker):
    """Override django_db_setup ensuring django_migration_disabler is loaded."""
    pass

赞(0）回复(0）举报 2023-04-13

我来回答

在pytest为django设置数据库之前安装postgresql扩展

1条答案

相关问题

热门标签

最新问答