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

在 Django 項目中，我們偶爾會遇到非托管模型，即元選項中沒有 Managed = True 的模型。這些模型可能會使測試變得棘手，特別是當您的測試設(shè)置涉及托管和非托管模型或多個數(shù)據(jù)庫的混合時（例如，一個包含托管模型，另一個包含非托管模型）。

這篇博文探討了使用 pytest-django 測試非托管模型的方法，重點介紹了優(yōu)點、缺點和解決方法，以幫助您有效管理這些場景。

方法 1：將所有模型標記為托管

在測試期間處理非托管模型的一種直接方法是將它們暫時標記為托管。具體方法如下：

# 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 選項

參考：Stack Overflow

優(yōu)點：

• 易于實施。

缺點：

• 跳過遷移測試，當多個開發(fā)人員處理同一個項目時，這可能會導致問題。

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

或者，您可以在測試設(shè)置期間手動創(chuàng)建非托管模型。這種方法可確保遷移經(jīng)過測試：

@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)點：

• 將遷移作為測試用例的一部分進行測試。

缺點：

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

了解事務(wù)測試

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

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

django_db(transaction=True): 在每個測試用例后提交更改并截斷數(shù)據(jù)庫表。由于每次測試后只有托管模型會被截斷，這就是非托管模型在事務(wù)測試期間需要特殊處理的原因。

示例測試用例

@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ù)測試適用于非托管模型

由于事務(wù)測試僅截斷托管模型，因此我們可以在測試運行期間將非托管模型修改為托管模型。這確保它們包含在截斷中：

# 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 掛鉤的場景中，您可以通過直接捕獲和執(zhí)行 on_commit 回調(diào)來避免使用事務(wù)測試，使用 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：測試非托管模型

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

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