Ticket #13706: 13706_tests.diff

tests/regressiontests/multiple_database/models.py

@@ -74 +74 @@
 
     class Meta:
         ordering = ('flavor',)
+
+class Pupil(models.Model):
+    pass
+
+class Teacher(models.Model):
+    pupils = models.ManyToManyField(Pupil)

tests/regressiontests/multiple_database/tests.py

@@ -11 +11 @@
 from django.db.utils import ConnectionRouter
 from django.test import TestCase
 
-from models import Book, Person, Pet, Review, UserProfile
+from models import Book, Person, Pet, Review, UserProfile, Pupil, Teacher
 
 try:
     # we only have these models if the user is using multi-db, it's safe the
@@ -1785 +1785 @@
         b.authors.clear()
         self._write_to_default()
         self.assertEqual(receiver._database, "other")
+
+class Ticket13706DefaultDbRouter(object):
+    """
+    Read and write only to the 'default' database.
+    """
+    def db_for_read(self, model, **hints):
+        return 'default'
+
+    def db_for_write(self, model, **hints):
+        return 'default'
+
+class Ticket13706OtherDbRouter(object):
+    """
+    Read and write only to the 'other' database.
+    """
+    def db_for_read(self, model, **hints):
+        return 'other'
+
+    def db_for_write(self, model, **hints):
+        return 'other'
+
+class Ticket13706TestCase(TestCase):
+    multi_db = True
+
+    def setUp(self):
+        self.old_routers = router.routers
+        router.routers = [Ticket13706OtherDbRouter(), Ticket13706DefaultDbRouter()]
+
+    def tearDown(self):
+        router.routers = self.old_routers
+
+    def test_ticket13706(self):
+        t = Teacher.objects.create()
+        p = Pupil.objects.create()
+        t.pupils.add(p)
+        self.assertEquals(t._state.db, 'other')
+        self.assertEquals(p._state.db, 'other')
+        self.assertEquals(Teacher.objects.using('default').count(), 0)
+        self.assertEquals(Pupil.objects.using('default').count(), 0)
+
+        t1 = Teacher()
+        t1.save(using='other')
+        p1 = Pupil()
+        p1.save(using='other')
+        t1.pupils.add(p1)
+        self.assertEquals(t1._state.db, 'other')
+        self.assertEquals(p1._state.db, 'other')
+        self.assertEquals(Teacher.objects.using('default').count(), 0)
+        self.assertEquals(Pupil.objects.using('default').count(), 0)