測(cè)試非託管模型的挑戰(zhàn)

在 Django 專案中，我們偶爾會(huì)遇到非託管模型，也就是元選項(xiàng)中沒有 Managed = True 的模型。這些模型可能會(huì)使測(cè)試變得棘手，特別是當(dāng)您的測(cè)試設(shè)定涉及託管和非託管模型或多個(gè)資料庫的混合時(shí)（例如，一個(gè)包含託管模型，另一個(gè)包含非託管模型）。

這篇部落格文章探討了使用 pytest-django 測(cè)試非託管模型的方法，重點(diǎn)介紹了優(yōu)點(diǎn)、缺點(diǎn)和解決方法，以幫助您有效管理這些場(chǎng)景。

方法 1：將所有模型標(biāo)記為託管

在測(cè)試期間處理非託管模型的一種直接方法是將它們暫時(shí)標(biāo)記為託管。具體方法如下：

# Add this to conftest.py
@pytest.hookimpl(tryfirst=True)
def pytest_runtestloop():
    from django.apps import apps
    unmanaged_models = []
    for app in apps.get_app_configs():
        unmanaged_models += [m for m in app.get_models()
                             if not m._meta.managed]
    for m in unmanaged_models:
        m._meta.managed = True

注意：要使此方法發(fā)揮作用，您需要在 pytest 設(shè)定（或 pytest.ini）中新增 --no-migrations 選項(xiàng)

參考：Stack Overflow

優(yōu)點(diǎn)：

• 易於實(shí)作。

缺點(diǎn)：

• 跳過遷移測(cè)試，當(dāng)多個(gè)開發(fā)人員處理同一個(gè)專案時(shí)，這可能會(huì)導(dǎo)致問題。

方法 2：手動(dòng)建立非託管模型

或者，您可以在測(cè)試設(shè)定期間手動(dòng)建立非託管模型。此方法可確保遷移經(jīng)過測(cè)試：

@pytest.fixture(scope="session", autouse=True)
def django_db_setup(django_db_blocker, django_db_setup):
    with django_db_blocker.unblock():
        for _connection in connections.all():
            with _connection.schema_editor() as schema_editor:
                setup_unmanaged_models(_connection, schema_editor)
        yield

def setup_unmanaged_models(connection, schema_editor):
    from django.apps import apps

    unmanaged_models = [
        model for model in apps.get_models() if model._meta.managed is False
    ]
    for model in unmanaged_models:
        if model._meta.db_table in connection.introspection.table_names():
            schema_editor.delete_model(model)
        schema_editor.create_model(model)

優(yōu)點(diǎn)：

• 將遷移作為測(cè)試案例的一部分進(jìn)行測(cè)試。

缺點(diǎn)：

• 稍微複雜一點(diǎn)。
• transaction=True 不適用於此方法（在下一節(jié)中討論）。

了解事務(wù)測(cè)試

Pytest-django 提供了資料庫固定裝置：django_db 和 django_db(transaction=True)。它們的差異如下：

django_db: 在測(cè)試案例結(jié)束時(shí)回滾更改，這表示不會(huì)對(duì)資料庫進(jìn)行實(shí)際提交。

django_db(transaction=True): 在每個(gè)測(cè)試案例後提交更改並截?cái)噘Y料庫表。由於每次測(cè)試後只有託管模型會(huì)被截?cái)?，這就是為什麼非託管模型在交易測(cè)試期間需要特殊處理的原因。

範(fàn)例測(cè)試案例

@pytest.mark.django_db
def test_example():
    # Test case logic here
    pass

@pytest.mark.django_db(transaction=True)
def test_transactional_example():
    # Test case logic here
    pass

使事務(wù)測(cè)試適用於非託管模型

由於交易測(cè)試僅截?cái)嘤毠苣Ｐ?，因此我們可以在測(cè)試運(yùn)行期間將非託管模型修改為託管模型。這確保它們包含在截?cái)嘀校?br>

# Add this to conftest.py
@pytest.hookimpl(tryfirst=True)
def pytest_runtestloop():
    from django.apps import apps
    unmanaged_models = []
    for app in apps.get_app_configs():
        unmanaged_models += [m for m in app.get_models()
                             if not m._meta.managed]
    for m in unmanaged_models:
        m._meta.managed = True

使用 on_commit Hooks 避免 transaction=True （如果可能的話）

在涉及 on_commit 掛鉤的場(chǎng)景中，您可以透過直接擷取和執(zhí)行 on_commit 回呼來避免使用交易測(cè)試，使用 pytest-django(>= v.4.4) 中的固定裝置 django_capture_on_commit_callbacks:

@pytest.fixture(scope="session", autouse=True)
def django_db_setup(django_db_blocker, django_db_setup):
    with django_db_blocker.unblock():
        for _connection in connections.all():
            with _connection.schema_editor() as schema_editor:
                setup_unmanaged_models(_connection, schema_editor)
        yield

def setup_unmanaged_models(connection, schema_editor):
    from django.apps import apps

    unmanaged_models = [
        model for model in apps.get_models() if model._meta.managed is False
    ]
    for model in unmanaged_models:
        if model._meta.db_table in connection.introspection.table_names():
            schema_editor.delete_model(model)
        schema_editor.create_model(model)

參考

• pytest-django 文件
• Stack Overflow：測(cè)試非託管模型

您還有其他處理非託管模型的方法或技巧嗎？在下面的評(píng)論中分享它們！

以上是在 Pytest-Django 中處理非託管模型的詳細(xì)內(nèi)容。更多資訊請(qǐng)關(guān)注PHP中文網(wǎng)其他相關(guān)文章！