Django TestCase nie używa transakcji w dodatkowej bazie danych
Używam Django 1.3.1. Mam dwie bazy danych, niektóre z moich modeli mieszkają w jednej bazie danych, niektóre w drugiej. Obie bazy danych są bazami danych contrib.gis.db.backends.postgis.
Ku mojemu zdziwieniu, TestCase Django nie wycofuje zmian wprowadzonych w drugiej bazie danych między testami.
W poniższym kodzie myproject.models.WellOwner jest bardzo prostym modelem, który zasadniczo ma tylko pole „nazwa”. Router mówi, że powinien znajdować się w dodatkowej bazie danych. Twierdzenie w pierwszym teście powiedzie się, drugi test nie powiedzie się:
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!
Zakładam, że Django zawija to w transakcji w domyślnej bazie danych, ale nie w dodatkowej bazie danych. Czy to znany problem? Czy istnieje poprawka? Być może w wersji 1.4? Moje Google-fu zawodzi.
(jeśli zmienię DATABASE_ROUTERS na [] w ustawieniach, aby wszystko trafiło do tej samej bazy danych, problem zniknie)
Dodam cały kod routera, na wypadek gdyby to pomogło:
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