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

在 Django 項(xiàng)目中，我們偶爾會(huì)遇到非托管模型，即元選項(xiàng)中沒有 Managed = True 的模型。這些模型可能會(huì)使測(cè)試變得棘手，特別是當(dāng)您的測(cè)試設(shè)置涉及托管和非托管模型或多個(gè)數(shù)據(jù)庫的混合時(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è)項(xiàng)目時(shí)，這可能會(huì)導(dǎo)致問題。

方法 2：手動(dòng)創(chuàng)建非托管模型

或者，您可以在測(cè)試設(shè)置期間手動(dòng)創(chuà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)：

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

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

Pytest-django 提供了數(shù)據(jù)庫固定裝置：django_db 和 django_db(transaction=True)。它們的區(qū)別如下：

django_db: 在測(cè)試用例結(jié)束時(shí)回滾更改，這意味著不會(huì)對(duì)數(shù)據(jù)庫進(jìn)行實(shí)際提交。

django_db(transaction=True): 在每個(gè)測(cè)試用例后提交更改并截?cái)鄶?shù)據(jù)庫表。由于每次測(cè)試后只有托管模型會(huì)被截?cái)啵@就是非托管模型在事務(wù)測(cè)試期間需要特殊處理的原因。

示例測(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è)試適用于非托管模型

由于事務(wù)測(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 回調(diào)來避免使用事務(wù)測(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)文章！