diff options
Diffstat (limited to 'tests/multiple_database/tests.py')
| -rw-r--r-- | tests/multiple_database/tests.py | 226 |
1 files changed, 82 insertions, 144 deletions
diff --git a/tests/multiple_database/tests.py b/tests/multiple_database/tests.py index 17a1ef90b1..90e418ba15 100644 --- a/tests/multiple_database/tests.py +++ b/tests/multiple_database/tests.py @@ -883,18 +883,11 @@ class ConnectionRouterTestCase(TestCase): ['TestRouter', 'WriteRouter']) +# Make the 'other' database appear to be a replica of the 'default' +@override_settings(DATABASE_ROUTERS=[TestRouter()]) class RouterTestCase(TestCase): multi_db = True - def setUp(self): - # Make the 'other' database appear to be a replica of the 'default' - self.old_routers = router.routers - router.routers = [TestRouter()] - - def tearDown(self): - # Restore the 'other' database as an independent database - router.routers = self.old_routers - def test_db_selection(self): "Check that querysets obey the router for db suggestions" self.assertEqual(Book.objects.db, 'other') @@ -914,24 +907,22 @@ class RouterTestCase(TestCase): self.assertTrue(router.allow_migrate('other', User)) self.assertTrue(router.allow_migrate('other', Book)) - # Add the auth router to the chain. - # TestRouter is a universal synchronizer, so it should have no effect. - router.routers = [TestRouter(), AuthRouter()] - - self.assertTrue(router.allow_migrate('default', User)) - self.assertTrue(router.allow_migrate('default', Book)) - - self.assertTrue(router.allow_migrate('other', User)) - self.assertTrue(router.allow_migrate('other', Book)) + with override_settings(DATABASE_ROUTERS=[TestRouter(), AuthRouter()]): + # Add the auth router to the chain. TestRouter is a universal + # synchronizer, so it should have no effect. + self.assertTrue(router.allow_migrate('default', User)) + self.assertTrue(router.allow_migrate('default', Book)) - # Now check what happens if the router order is the other way around - router.routers = [AuthRouter(), TestRouter()] + self.assertTrue(router.allow_migrate('other', User)) + self.assertTrue(router.allow_migrate('other', Book)) - self.assertFalse(router.allow_migrate('default', User)) - self.assertTrue(router.allow_migrate('default', Book)) + with override_settings(DATABASE_ROUTERS=[AuthRouter(), TestRouter()]): + # Now check what happens if the router order is reversed. + self.assertFalse(router.allow_migrate('default', User)) + self.assertTrue(router.allow_migrate('default', Book)) - self.assertTrue(router.allow_migrate('other', User)) - self.assertFalse(router.allow_migrate('other', Book)) + self.assertTrue(router.allow_migrate('other', User)) + self.assertFalse(router.allow_migrate('other', Book)) def test_partial_router(self): "A router can choose to implement a subset of methods" @@ -951,18 +942,17 @@ class RouterTestCase(TestCase): self.assertTrue(router.allow_migrate('default', User)) self.assertTrue(router.allow_migrate('default', Book)) - router.routers = [WriteRouter(), AuthRouter(), TestRouter()] - - self.assertEqual(router.db_for_read(User), 'default') - self.assertEqual(router.db_for_read(Book), 'other') + with override_settings(DATABASE_ROUTERS=[WriteRouter(), AuthRouter(), TestRouter()]): + self.assertEqual(router.db_for_read(User), 'default') + self.assertEqual(router.db_for_read(Book), 'other') - self.assertEqual(router.db_for_write(User), 'writer') - self.assertEqual(router.db_for_write(Book), 'writer') + self.assertEqual(router.db_for_write(User), 'writer') + self.assertEqual(router.db_for_write(Book), 'writer') - self.assertTrue(router.allow_relation(dive, dive)) + self.assertTrue(router.allow_relation(dive, dive)) - self.assertFalse(router.allow_migrate('default', User)) - self.assertTrue(router.allow_migrate('default', Book)) + self.assertFalse(router.allow_migrate('default', User)) + self.assertTrue(router.allow_migrate('default', Book)) def test_database_routing(self): marty = Person.objects.using('default').create(name="Marty Alchin") @@ -1447,18 +1437,10 @@ class RouterTestCase(TestCase): datetime.date(2009, 5, 4)) +@override_settings(DATABASE_ROUTERS=[AuthRouter()]) class AuthTestCase(TestCase): multi_db = True - def setUp(self): - # Make the 'other' database appear to be a replica of the 'default' - self.old_routers = router.routers - router.routers = [AuthRouter()] - - def tearDown(self): - # Restore the 'other' database as an independent database - router.routers = self.old_routers - def test_auth_manager(self): "The methods on the auth manager obey database hints" # Create one user using default allocation policy @@ -1522,15 +1504,7 @@ class FixtureTestCase(TestCase): multi_db = True fixtures = ['multidb-common', 'multidb'] - def setUp(self): - # Install the anti-pet router - self.old_routers = router.routers - router.routers = [AntiPetRouter()] - - def tearDown(self): - # Restore the 'other' database as an independent database - router.routers = self.old_routers - + @override_settings(DATABASE_ROUTERS=[AntiPetRouter()]) def test_fixture_loading(self): "Multi-db fixtures are loaded correctly" # Check that "Pro Django" exists on the default database, but not on other database @@ -1571,6 +1545,7 @@ class FixtureTestCase(TestCase): except Book.DoesNotExist: self.fail('"The Definitive Guide to Django" should exist on both databases') + @override_settings(DATABASE_ROUTERS=[AntiPetRouter()]) def test_pseudo_empty_fixtures(self): "A fixture can contain entries, but lead to nothing in the database; this shouldn't raise an error (ref #14068)" new_io = StringIO() @@ -1609,19 +1584,8 @@ class WriteToOtherRouter(object): class SignalTests(TestCase): multi_db = True - def setUp(self): - self.old_routers = router.routers - - def tearDown(self): - router.routers = self.old_routers - - def _write_to_other(self): - "Sends all writes to 'other'." - router.routers = [WriteToOtherRouter()] - - def _write_to_default(self): - "Sends all writes to the default DB" - router.routers = self.old_routers + def override_router(self): + return override_settings(DATABASE_ROUTERS=[WriteToOtherRouter()]) def test_database_arg_save_and_delete(self): """ @@ -1684,33 +1648,29 @@ class SignalTests(TestCase): # Test addition b.authors.add(p) self.assertEqual(receiver._database, DEFAULT_DB_ALIAS) - self._write_to_other() - b.authors.add(p) - self._write_to_default() + with self.override_router(): + b.authors.add(p) self.assertEqual(receiver._database, "other") # Test removal b.authors.remove(p) self.assertEqual(receiver._database, DEFAULT_DB_ALIAS) - self._write_to_other() - b.authors.remove(p) - self._write_to_default() + with self.override_router(): + b.authors.remove(p) self.assertEqual(receiver._database, "other") # Test addition in reverse p.book_set.add(b) self.assertEqual(receiver._database, DEFAULT_DB_ALIAS) - self._write_to_other() - p.book_set.add(b) - self._write_to_default() + with self.override_router(): + p.book_set.add(b) self.assertEqual(receiver._database, "other") # Test clearing b.authors.clear() self.assertEqual(receiver._database, DEFAULT_DB_ALIAS) - self._write_to_other() - b.authors.clear() - self._write_to_default() + with self.override_router(): + b.authors.clear() self.assertEqual(receiver._database, "other") @@ -1726,47 +1686,41 @@ class AttributeErrorRouter(object): class RouterAttributeErrorTestCase(TestCase): multi_db = True - def setUp(self): - self.old_routers = router.routers - router.routers = [AttributeErrorRouter()] - - def tearDown(self): - router.routers = self.old_routers + def override_router(self): + return override_settings(DATABASE_ROUTERS=[AttributeErrorRouter()]) def test_attribute_error_read(self): "Check that the AttributeError from AttributeErrorRouter bubbles up" - router.routers = [] # Reset routers so we can save a Book instance b = Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16)) - router.routers = [AttributeErrorRouter()] # Install our router - self.assertRaises(AttributeError, Book.objects.get, pk=b.pk) + with self.override_router(): + self.assertRaises(AttributeError, Book.objects.get, pk=b.pk) def test_attribute_error_save(self): "Check that the AttributeError from AttributeErrorRouter bubbles up" dive = Book() dive.title = "Dive into Python" dive.published = datetime.date(2009, 5, 4) - self.assertRaises(AttributeError, dive.save) + with self.override_router(): + self.assertRaises(AttributeError, dive.save) def test_attribute_error_delete(self): "Check that the AttributeError from AttributeErrorRouter bubbles up" - router.routers = [] # Reset routers so we can save our Book, Person instances b = Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16)) p = Person.objects.create(name="Marty Alchin") b.authors = [p] b.editor = p - router.routers = [AttributeErrorRouter()] # Install our router - self.assertRaises(AttributeError, b.delete) + with self.override_router(): + self.assertRaises(AttributeError, b.delete) def test_attribute_error_m2m(self): "Check that the AttributeError from AttributeErrorRouter bubbles up" - router.routers = [] # Reset routers so we can save our Book, Person instances b = Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16)) p = Person.objects.create(name="Marty Alchin") - router.routers = [AttributeErrorRouter()] # Install our router - self.assertRaises(AttributeError, setattr, b, 'authors', [p]) + with self.override_router(): + self.assertRaises(AttributeError, setattr, b, 'authors', [p]) class ModelMetaRouter(object): @@ -1776,16 +1730,10 @@ class ModelMetaRouter(object): raise ValueError +@override_settings(DATABASE_ROUTERS=[ModelMetaRouter()]) class RouterModelArgumentTestCase(TestCase): multi_db = True - def setUp(self): - self.old_routers = router.routers - router.routers = [ModelMetaRouter()] - - def tearDown(self): - router.routers = self.old_routers - def test_m2m_collection(self): b = Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16)) @@ -1840,13 +1788,9 @@ class MigrateTestCase(TestCase): cts = ContentType.objects.using('other').filter(app_label='multiple_database') cts.delete() - try: - old_routers = router.routers - router.routers = [SyncOnlyDefaultDatabaseRouter()] + with override_settings(DATABASE_ROUTERS=[SyncOnlyDefaultDatabaseRouter()]): management.call_command('migrate', verbosity=0, interactive=False, load_initial_data=False, database='other') - finally: - router.routers = old_routers self.assertEqual(cts.count(), 0) @@ -1868,21 +1812,15 @@ class RouteForWriteTestCase(TestCase): def db_for_write(self, model, **hints): raise RouterUsed(mode=RouterUsed.WRITE, model=model, hints=hints) - def setUp(self): - self._old_rtrs = router.routers - - def tearDown(self): - router.routers = self._old_rtrs - - def enable_router(self): - router.routers = [RouteForWriteTestCase.WriteCheckRouter()] + def override_router(self): + return override_settings(DATABASE_ROUTERS=[RouteForWriteTestCase.WriteCheckRouter()]) def test_fk_delete(self): owner = Person.objects.create(name='Someone') pet = Pet.objects.create(name='fido', owner=owner) - self.enable_router() try: - pet.owner.delete() + with self.override_router(): + pet.owner.delete() self.fail('db_for_write() not invoked on router') except RouterUsed as e: self.assertEqual(e.mode, RouterUsed.WRITE) @@ -1892,9 +1830,9 @@ class RouteForWriteTestCase(TestCase): def test_reverse_fk_delete(self): owner = Person.objects.create(name='Someone') to_del_qs = owner.pet_set.all() - self.enable_router() try: - to_del_qs.delete() + with self.override_router(): + to_del_qs.delete() self.fail('db_for_write() not invoked on router') except RouterUsed as e: self.assertEqual(e.mode, RouterUsed.WRITE) @@ -1903,9 +1841,9 @@ class RouteForWriteTestCase(TestCase): def test_reverse_fk_get_or_create(self): owner = Person.objects.create(name='Someone') - self.enable_router() try: - owner.pet_set.get_or_create(name='fido') + with self.override_router(): + owner.pet_set.get_or_create(name='fido') self.fail('db_for_write() not invoked on router') except RouterUsed as e: self.assertEqual(e.mode, RouterUsed.WRITE) @@ -1915,9 +1853,9 @@ class RouteForWriteTestCase(TestCase): def test_reverse_fk_update(self): owner = Person.objects.create(name='Someone') Pet.objects.create(name='fido', owner=owner) - self.enable_router() try: - owner.pet_set.update(name='max') + with self.override_router(): + owner.pet_set.update(name='max') self.fail('db_for_write() not invoked on router') except RouterUsed as e: self.assertEqual(e.mode, RouterUsed.WRITE) @@ -1928,9 +1866,9 @@ class RouteForWriteTestCase(TestCase): auth = Person.objects.create(name='Someone') book = Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16)) - self.enable_router() try: - book.authors.add(auth) + with self.override_router(): + book.authors.add(auth) self.fail('db_for_write() not invoked on router') except RouterUsed as e: self.assertEqual(e.mode, RouterUsed.WRITE) @@ -1942,9 +1880,9 @@ class RouteForWriteTestCase(TestCase): book = Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16)) book.authors.add(auth) - self.enable_router() try: - book.authors.clear() + with self.override_router(): + book.authors.clear() self.fail('db_for_write() not invoked on router') except RouterUsed as e: self.assertEqual(e.mode, RouterUsed.WRITE) @@ -1956,9 +1894,9 @@ class RouteForWriteTestCase(TestCase): book = Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16)) book.authors.add(auth) - self.enable_router() try: - book.authors.all().delete() + with self.override_router(): + book.authors.all().delete() self.fail('db_for_write() not invoked on router') except RouterUsed as e: self.assertEqual(e.mode, RouterUsed.WRITE) @@ -1969,9 +1907,9 @@ class RouteForWriteTestCase(TestCase): Person.objects.create(name='Someone') book = Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16)) - self.enable_router() try: - book.authors.get_or_create(name='Someone else') + with self.override_router(): + book.authors.get_or_create(name='Someone else') self.fail('db_for_write() not invoked on router') except RouterUsed as e: self.assertEqual(e.mode, RouterUsed.WRITE) @@ -1983,10 +1921,10 @@ class RouteForWriteTestCase(TestCase): book = Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16)) book.authors.add(auth) - self.enable_router() self.assertRaisesMessage(AttributeError, self.RAISE_MSG, ) try: - book.authors.remove(auth) + with self.override_router(): + book.authors.remove(auth) self.fail('db_for_write() not invoked on router') except RouterUsed as e: self.assertEqual(e.mode, RouterUsed.WRITE) @@ -1998,9 +1936,9 @@ class RouteForWriteTestCase(TestCase): book = Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16)) book.authors.add(auth) - self.enable_router() try: - book.authors.all().update(name='Different') + with self.override_router(): + book.authors.all().update(name='Different') self.fail('db_for_write() not invoked on router') except RouterUsed as e: self.assertEqual(e.mode, RouterUsed.WRITE) @@ -2011,9 +1949,9 @@ class RouteForWriteTestCase(TestCase): auth = Person.objects.create(name='Someone') book = Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16)) - self.enable_router() try: - auth.book_set.add(book) + with self.override_router(): + auth.book_set.add(book) self.fail('db_for_write() not invoked on router') except RouterUsed as e: self.assertEqual(e.mode, RouterUsed.WRITE) @@ -2025,9 +1963,9 @@ class RouteForWriteTestCase(TestCase): book = Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16)) book.authors.add(auth) - self.enable_router() try: - auth.book_set.clear() + with self.override_router(): + auth.book_set.clear() self.fail('db_for_write() not invoked on router') except RouterUsed as e: self.assertEqual(e.mode, RouterUsed.WRITE) @@ -2039,9 +1977,9 @@ class RouteForWriteTestCase(TestCase): book = Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16)) book.authors.add(auth) - self.enable_router() try: - auth.book_set.all().delete() + with self.override_router(): + auth.book_set.all().delete() self.fail('db_for_write() not invoked on router') except RouterUsed as e: self.assertEqual(e.mode, RouterUsed.WRITE) @@ -2052,9 +1990,9 @@ class RouteForWriteTestCase(TestCase): auth = Person.objects.create(name='Someone') Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16)) - self.enable_router() try: - auth.book_set.get_or_create(title="New Book", published=datetime.datetime.now()) + with self.override_router(): + auth.book_set.get_or_create(title="New Book", published=datetime.datetime.now()) self.fail('db_for_write() not invoked on router') except RouterUsed as e: self.assertEqual(e.mode, RouterUsed.WRITE) @@ -2066,9 +2004,9 @@ class RouteForWriteTestCase(TestCase): book = Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16)) book.authors.add(auth) - self.enable_router() try: - auth.book_set.remove(book) + with self.override_router(): + auth.book_set.remove(book) self.fail('db_for_write() not invoked on router') except RouterUsed as e: self.assertEqual(e.mode, RouterUsed.WRITE) @@ -2080,9 +2018,9 @@ class RouteForWriteTestCase(TestCase): book = Book.objects.create(title="Pro Django", published=datetime.date(2008, 12, 16)) book.authors.add(auth) - self.enable_router() try: - auth.book_set.all().update(title='Different') + with self.override_router(): + auth.book_set.all().update(title='Different') self.fail('db_for_write() not invoked on router') except RouterUsed as e: self.assertEqual(e.mode, RouterUsed.WRITE) |