如何在 django pytest 中使用多个数据库进行单元测试用例答案

【问题标题】：How to use multiple databases for unit test cases in django pytest如何在 django pytest 中使用多个数据库进行单元测试用例
【发布时间】：2021-01-29 13:19:17
【问题描述】：

我一直在开发一个 django 应用程序，它通过数据库路由器使用多个数据库，如下所示。我想为使用 Django Rest Framework 开发的 api 编写单元测试用例。

class MultiDbRouter(object):
    """
        Router for handling multiple database connections
        based on the db configs in the request object
    """

    def _multi_db(self):
        if hasattr(request_cfg, 'db'):
            if request_cfg.db in settings.DATABASES:
                return request_cfg.db
            else:
                raise Http404
        else:
            return 'default'

    def db_for_read(self, model, **hints):
        """
            Set database for reading based on request config
        """
        if model._meta.app_label == 'survey':
            return 'survey'
        if model._meta.app_label != 'analytics':
            return 'default'
        return self._multi_db()

    def db_for_write(self, model, **hints):
        """
            Set database for writing based on request config
        """
        if model._meta.app_label == 'survey':
            return 'survey'
        if model._meta.app_label != 'analytics':
            return 'default'
        return self._multi_db()

    def allow_relation(self, obj1, obj2, **hints):
        """
            allow relation between objects
        """
        return True

    def allow_syncdb(self, db, model):
        """
            allow syncing of database
        """
        return True

我已经开始使用 pytest 来编写测试用例，但是当我阅读 pytest-django 库的文档时，它说“目前 pytest-django 不专门支持 Django 的多数据库支持。”

有没有办法解决这个限制？或者我应该使用 pytest-django 以外的其他库来编写单元测试用例？如果有，是哪一个？

理想的前进方式是什么？

提前致谢！

【问题讨论】：

正如 pytest 的文档所说，使用 Django 的普通测试类。 Tests and multiple databases - Django docs
我读到了，但是 pytest 有什么用呢？我将不得不使用 Django 的 Testcase 编写所有测试用例。

标签： django pytest-django

【解决方案1】：

可以使用 pytest 和多个数据库，所有测试都使用通常的 pytest 语法编写。在您的 conftest.py 文件中，将以下内容添加到顶部：

from django.test import TestCase

TestCase.databases = {"default", "survey"}

这将导致所有 pytest 测试都使用您的两个数据库。

pytest-django 上的文档暗示无法在逐个测试的基础上或使用 pytest 固定装置来配置多数据库支持。当然，上述情况可能会使您的测试套件变慢，因为您的测试将在每次测试之间回滚两个数据库之间的更改。

我从 pytest-django 上的 this comment 和 django docs on multi database support 获得了这个解决方案。

【讨论】：