python - Django 中的多个数据库破坏了测试隔离。如何解决？

Question

Django’s TestCase class wraps each test in a transaction and rolls back that transaction after each test, in order to provide test isolation.

然而，显然只有 default 数据库中的操作在事务范围内。我有一个带有路由器的多数据库设置，该路由器将某些模型上的 ORM 调用定向到第二个数据库。这意味着在下面的示例中，test2 失败了:

class MyTestCase(TestCase):

    def test1(self):
        # Let's say Foo model ORM calls are routed to a different DB
        foo = Foo.objects.create(...)
        assert foo.is_ok()

    def test2(self):
        assert not Foo.objects.exists()

解决此问题的最直接方法是覆盖 MyTestCase 中的 tearDown 方法，并手动确保删除所有 Foo 对象。但这有点烦人，因为它有点像 hack 和数据库序列(例如自动增量列)不会被重置，例如，只有在测试套件完成并且数据库被销毁之后。

有没有办法正确解决这个问题，确保默认情况下所有数据库操作都在事务内进行并在每次测试结束时回滚？

[更新]

这是我的路由器:

class FooRouter(object):

    def db_for_read(self, model, **hints):

        if model._meta.app_label == 'foo':
            return 'foo_db'

        return None

    def db_for_write(self, model, **hints):

        if model._meta.app_label == 'foo':
            return 'foo_db'

        return None

    def allow_migrate(self, db, app_label, model_name=None, **hints):

        if app_label == 'foo':
            return db == 'foo_db'

        return None

Answer 1

由于 multi_db 已弃用，您现在可以使用新属性 databases 实现对多个数据库的测试

class YourTestCase(TestCase):
   databases = '__all__'

   def test_something(self):
       pass