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ę:
<code>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! </code>
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:
<code>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 </code>