Django的TestCase的不使用辅助数据库事务(Django TestCase not usi

2019-06-23 14:19发布

我使用Django 1.3.1。 我有两个数据库,我的一些车型生活在一个数据库中,一些在其他。 这两个数据库都contrib.gis.db.backends.postgis数据库。

令我惊讶的是,Django的TestCase的不回滚我在测试中的辅助数据库所做的更改。

在下面的代码,myproject.models.WellOwner是,基本上只有一个字段“名”一个非常简单的模型。 路由器说,它应该是在辅助数据库。 在第一次测试的断言成功,第二次测试失败:

from django.test import TestCase
from myproject.models import WellOwner

class SimpleTest(TestCase):
    def test1(self):
        WellOwner.objects.create(name="Remco")
        self.assertEquals(1, WellOwner.objects.count())  # Succeeds

class SimpleTest2(TestCase):
    def test2(self):
        # I would expect to have an empty database at this point
        self.assertEquals(0, WellOwner.objects.count())  # Fails!

我认为Django的包装此默认数据库的事务,而不是辅助数据库。 这是一个已知的问题? 有没有解决? 在1.4吧? 我的谷歌福失败。

(如果我更改了设置DATABASE_ROUTERS为[]让一切都进入同一个数据库,问题就会消失)

我会添加路由器的整个代码,在情况下,它可以帮助:

SECONDARY_MODELS = ('WellOwner', ...)

import logging
logger = logging.getLogger(__name__)


class GmdbRouter(object):
    """Keep some models in a secondary database."""

    def db_for_read(self, model, **hints):
        if model._meta.app_label == 'gmdb':
            if model._meta.object_name in SECONDARY_MODELS:
                return 'secondary'

        return None

    def db_for_write(self, model, **hints):
        # Same criteria as for reading
        return self.db_for_read(model, **hints)

    def allow_syncdb(self, db, model):
        if db == 'secondary':
            if model._meta.app_label in ('sites', 'south'):
                # Hack for bug https://code.djangoproject.com/ticket/16353
                # When testing, create django_site and south in both databases
                return True

            return self.db_for_read(model) == 'secondary'
        else:
            # Some other db
            if model._meta.app_label == 'gmdb':
                # Our models go in the other db if they don't go into secondary
                return self.db_for_read(model) != 'secondary'

            # Some other model in some other db, no opinion
            return None

Answer 1:

试试这个:

class MyTestCase(TestCase):
    multi_db = True

https://docs.djangoproject.com/en/1.2/topics/testing/#django.test.TestCase.multi_db



文章来源: Django TestCase not using transactions on secondary database