Replaced router.routers usage with override_settings(DATABASE_ROUTERS); refs #23933.
This commit is contained in:
parent
e6f19ec322
commit
9136ceb6fb
|
@ -8,9 +8,9 @@ import unittest
|
|||
from unittest import skipUnless
|
||||
|
||||
from django.contrib.gis.gdal import HAS_GDAL
|
||||
from django.db import connection, router
|
||||
from django.db import connection
|
||||
from django.conf import settings
|
||||
from django.test import TestCase, skipUnlessDBFeature
|
||||
from django.test import TestCase, override_settings, skipUnlessDBFeature
|
||||
from django.utils._os import upath
|
||||
|
||||
if HAS_GDAL:
|
||||
|
@ -328,15 +328,9 @@ class OtherRouter(object):
|
|||
|
||||
@skipUnless(HAS_GDAL, "LayerMapRouterTest needs GDAL support")
|
||||
@skipUnlessDBFeature("gis_enabled")
|
||||
@override_settings(DATABASE_ROUTERS=[OtherRouter()])
|
||||
class LayerMapRouterTest(TestCase):
|
||||
|
||||
def setUp(self):
|
||||
self.old_routers = router.routers
|
||||
router.routers = [OtherRouter()]
|
||||
|
||||
def tearDown(self):
|
||||
router.routers = self.old_routers
|
||||
|
||||
@unittest.skipUnless(len(settings.DATABASES) > 1, 'multiple databases required')
|
||||
def test_layermapping_default_db(self):
|
||||
lm = LayerMapping(City, city_shp, city_mapping)
|
||||
|
|
|
@ -19,7 +19,7 @@ from django.core import management
|
|||
from django.core.cache import (cache, caches, CacheKeyWarning,
|
||||
InvalidCacheBackendError, DEFAULT_CACHE_ALIAS)
|
||||
from django.core.context_processors import csrf
|
||||
from django.db import connection, connections, router, transaction
|
||||
from django.db import connection, connections, transaction
|
||||
from django.core.cache.utils import make_template_fragment_key
|
||||
from django.http import HttpResponse, StreamingHttpResponse
|
||||
from django.middleware.cache import (FetchFromCacheMiddleware,
|
||||
|
@ -974,10 +974,8 @@ class DBCacheRouter(object):
|
|||
class CreateCacheTableForDBCacheTests(TestCase):
|
||||
multi_db = True
|
||||
|
||||
@override_settings(DATABASE_ROUTERS=[DBCacheRouter()])
|
||||
def test_createcachetable_observes_database_router(self):
|
||||
old_routers = router.routers
|
||||
try:
|
||||
router.routers = [DBCacheRouter()]
|
||||
# cache table should not be created on 'default'
|
||||
with self.assertNumQueries(0, using='default'):
|
||||
management.call_command('createcachetable',
|
||||
|
@ -995,8 +993,6 @@ class CreateCacheTableForDBCacheTests(TestCase):
|
|||
management.call_command('createcachetable',
|
||||
database='other',
|
||||
verbosity=0, interactive=False)
|
||||
finally:
|
||||
router.routers = old_routers
|
||||
|
||||
|
||||
class PicklingSideEffect(object):
|
||||
|
|
|
@ -7,8 +7,8 @@ from django.apps import apps
|
|||
from django.core.management.color import no_style
|
||||
from django.core.management.sql import (sql_create, sql_delete, sql_indexes,
|
||||
sql_destroy_indexes, sql_all)
|
||||
from django.db import connections, DEFAULT_DB_ALIAS, router
|
||||
from django.test import TestCase
|
||||
from django.db import connections, DEFAULT_DB_ALIAS
|
||||
from django.test import TestCase, override_settings
|
||||
from django.utils import six
|
||||
|
||||
# See also initial_sql_regress for 'custom_sql_for_model' tests
|
||||
|
@ -91,13 +91,8 @@ class TestRouter(object):
|
|||
return False
|
||||
|
||||
|
||||
@override_settings(DATABASE_ROUTERS=[TestRouter()])
|
||||
class SQLCommandsRouterTestCase(TestCase):
|
||||
def setUp(self):
|
||||
self._old_routers = router.routers
|
||||
router.routers = [TestRouter()]
|
||||
|
||||
def tearDown(self):
|
||||
router.routers = self._old_routers
|
||||
|
||||
def test_router_honored(self):
|
||||
app_config = apps.get_app_config('commands_sql')
|
||||
|
|
|
@ -8,9 +8,9 @@ from django.contrib.contenttypes.fields import (
|
|||
from django.contrib.contenttypes import management
|
||||
from django.contrib.contenttypes.models import ContentType
|
||||
from django.core import checks
|
||||
from django.db import connections, models, router
|
||||
from django.test import TestCase
|
||||
from django.test.utils import captured_stdout, override_settings
|
||||
from django.db import connections, models
|
||||
from django.test import TestCase, override_settings
|
||||
from django.test.utils import captured_stdout
|
||||
from django.utils.encoding import force_str
|
||||
|
||||
from .models import Author, Article, SchemeIncludedURL
|
||||
|
@ -397,12 +397,10 @@ class TestRouter(object):
|
|||
return 'default'
|
||||
|
||||
|
||||
@override_settings(DATABASE_ROUTERS=[TestRouter()])
|
||||
class ContentTypesMultidbTestCase(TestCase):
|
||||
|
||||
def setUp(self):
|
||||
self.old_routers = router.routers
|
||||
router.routers = [TestRouter()]
|
||||
|
||||
# Whenever a test starts executing, only the "default" database is
|
||||
# connected. We explicitly connect to the "other" database here. If we
|
||||
# don't do it, then it will be implicitly connected later when we query
|
||||
|
@ -411,9 +409,6 @@ class ContentTypesMultidbTestCase(TestCase):
|
|||
# "SET SQL_AUTO_IS_NULL = 0"), which will affect assertNumQueries().
|
||||
connections['other'].ensure_connection()
|
||||
|
||||
def tearDown(self):
|
||||
router.routers = self.old_routers
|
||||
|
||||
def test_multidb(self):
|
||||
"""
|
||||
Test that, when using multiple databases, we use the db_for_read (see
|
||||
|
|
|
@ -8,13 +8,13 @@ except ImportError:
|
|||
sqlparse = None
|
||||
|
||||
from django import test
|
||||
from django.test import override_settings
|
||||
from django.db import connection, migrations, models, router
|
||||
from django.db import connection, migrations, models
|
||||
from django.db.migrations.migration import Migration
|
||||
from django.db.migrations.state import ProjectState
|
||||
from django.db.models.fields import NOT_PROVIDED
|
||||
from django.db.transaction import atomic
|
||||
from django.db.utils import IntegrityError, DatabaseError
|
||||
from django.test import override_settings
|
||||
from django.utils import six
|
||||
|
||||
from .test_base import MigrationTestBase
|
||||
|
@ -1566,18 +1566,10 @@ class MigrateNothingRouter(object):
|
|||
return False
|
||||
|
||||
|
||||
@override_settings(DATABASE_ROUTERS=[MigrateNothingRouter()])
|
||||
class MultiDBOperationTests(MigrationTestBase):
|
||||
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 = [MigrateNothingRouter()]
|
||||
|
||||
def tearDown(self):
|
||||
# Restore the 'other' database as an independent database
|
||||
router.routers = self.old_routers
|
||||
|
||||
def test_create_model(self):
|
||||
"""
|
||||
Tests that CreateModel honours multi-db settings.
|
||||
|
|
|
@ -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,19 +907,17 @@ 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()]
|
||||
|
||||
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))
|
||||
|
||||
self.assertTrue(router.allow_migrate('other', User))
|
||||
self.assertTrue(router.allow_migrate('other', Book))
|
||||
|
||||
# Now check what happens if the router order is the other way around
|
||||
router.routers = [AuthRouter(), TestRouter()]
|
||||
|
||||
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))
|
||||
|
||||
|
@ -951,8 +942,7 @@ class RouterTestCase(TestCase):
|
|||
self.assertTrue(router.allow_migrate('default', User))
|
||||
self.assertTrue(router.allow_migrate('default', Book))
|
||||
|
||||
router.routers = [WriteRouter(), AuthRouter(), TestRouter()]
|
||||
|
||||
with override_settings(DATABASE_ROUTERS=[WriteRouter(), AuthRouter(), TestRouter()]):
|
||||
self.assertEqual(router.db_for_read(User), 'default')
|
||||
self.assertEqual(router.db_for_read(Book), 'other')
|
||||
|
||||
|
@ -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()
|
||||
with self.override_router():
|
||||
b.authors.add(p)
|
||||
self._write_to_default()
|
||||
self.assertEqual(receiver._database, "other")
|
||||
|
||||
# Test removal
|
||||
b.authors.remove(p)
|
||||
self.assertEqual(receiver._database, DEFAULT_DB_ALIAS)
|
||||
self._write_to_other()
|
||||
with self.override_router():
|
||||
b.authors.remove(p)
|
||||
self._write_to_default()
|
||||
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()
|
||||
with self.override_router():
|
||||
p.book_set.add(b)
|
||||
self._write_to_default()
|
||||
self.assertEqual(receiver._database, "other")
|
||||
|
||||
# Test clearing
|
||||
b.authors.clear()
|
||||
self.assertEqual(receiver._database, DEFAULT_DB_ALIAS)
|
||||
self._write_to_other()
|
||||
with self.override_router():
|
||||
b.authors.clear()
|
||||
self._write_to_default()
|
||||
self.assertEqual(receiver._database, "other")
|
||||
|
||||
|
||||
|
@ -1726,19 +1686,14 @@ 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
|
||||
with self.override_router():
|
||||
self.assertRaises(AttributeError, Book.objects.get, pk=b.pk)
|
||||
|
||||
def test_attribute_error_save(self):
|
||||
|
@ -1746,26 +1701,25 @@ class RouterAttributeErrorTestCase(TestCase):
|
|||
dive = Book()
|
||||
dive.title = "Dive into Python"
|
||||
dive.published = datetime.date(2009, 5, 4)
|
||||
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
|
||||
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
|
||||
with self.override_router():
|
||||
self.assertRaises(AttributeError, setattr, b, 'authors', [p])
|
||||
|
||||
|
||||
|
@ -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,20 +1812,14 @@ 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:
|
||||
with self.override_router():
|
||||
pet.owner.delete()
|
||||
self.fail('db_for_write() not invoked on router')
|
||||
except RouterUsed as e:
|
||||
|
@ -1892,8 +1830,8 @@ 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:
|
||||
with self.override_router():
|
||||
to_del_qs.delete()
|
||||
self.fail('db_for_write() not invoked on router')
|
||||
except RouterUsed as e:
|
||||
|
@ -1903,8 +1841,8 @@ class RouteForWriteTestCase(TestCase):
|
|||
|
||||
def test_reverse_fk_get_or_create(self):
|
||||
owner = Person.objects.create(name='Someone')
|
||||
self.enable_router()
|
||||
try:
|
||||
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:
|
||||
|
@ -1915,8 +1853,8 @@ 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:
|
||||
with self.override_router():
|
||||
owner.pet_set.update(name='max')
|
||||
self.fail('db_for_write() not invoked on router')
|
||||
except RouterUsed as e:
|
||||
|
@ -1928,8 +1866,8 @@ 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:
|
||||
with self.override_router():
|
||||
book.authors.add(auth)
|
||||
self.fail('db_for_write() not invoked on router')
|
||||
except RouterUsed as e:
|
||||
|
@ -1942,8 +1880,8 @@ class RouteForWriteTestCase(TestCase):
|
|||
book = Book.objects.create(title="Pro Django",
|
||||
published=datetime.date(2008, 12, 16))
|
||||
book.authors.add(auth)
|
||||
self.enable_router()
|
||||
try:
|
||||
with self.override_router():
|
||||
book.authors.clear()
|
||||
self.fail('db_for_write() not invoked on router')
|
||||
except RouterUsed as e:
|
||||
|
@ -1956,8 +1894,8 @@ class RouteForWriteTestCase(TestCase):
|
|||
book = Book.objects.create(title="Pro Django",
|
||||
published=datetime.date(2008, 12, 16))
|
||||
book.authors.add(auth)
|
||||
self.enable_router()
|
||||
try:
|
||||
with self.override_router():
|
||||
book.authors.all().delete()
|
||||
self.fail('db_for_write() not invoked on router')
|
||||
except RouterUsed as e:
|
||||
|
@ -1969,8 +1907,8 @@ 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:
|
||||
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:
|
||||
|
@ -1983,9 +1921,9 @@ 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:
|
||||
with self.override_router():
|
||||
book.authors.remove(auth)
|
||||
self.fail('db_for_write() not invoked on router')
|
||||
except RouterUsed as e:
|
||||
|
@ -1998,8 +1936,8 @@ class RouteForWriteTestCase(TestCase):
|
|||
book = Book.objects.create(title="Pro Django",
|
||||
published=datetime.date(2008, 12, 16))
|
||||
book.authors.add(auth)
|
||||
self.enable_router()
|
||||
try:
|
||||
with self.override_router():
|
||||
book.authors.all().update(name='Different')
|
||||
self.fail('db_for_write() not invoked on router')
|
||||
except RouterUsed as e:
|
||||
|
@ -2011,8 +1949,8 @@ 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:
|
||||
with self.override_router():
|
||||
auth.book_set.add(book)
|
||||
self.fail('db_for_write() not invoked on router')
|
||||
except RouterUsed as e:
|
||||
|
@ -2025,8 +1963,8 @@ class RouteForWriteTestCase(TestCase):
|
|||
book = Book.objects.create(title="Pro Django",
|
||||
published=datetime.date(2008, 12, 16))
|
||||
book.authors.add(auth)
|
||||
self.enable_router()
|
||||
try:
|
||||
with self.override_router():
|
||||
auth.book_set.clear()
|
||||
self.fail('db_for_write() not invoked on router')
|
||||
except RouterUsed as e:
|
||||
|
@ -2039,8 +1977,8 @@ class RouteForWriteTestCase(TestCase):
|
|||
book = Book.objects.create(title="Pro Django",
|
||||
published=datetime.date(2008, 12, 16))
|
||||
book.authors.add(auth)
|
||||
self.enable_router()
|
||||
try:
|
||||
with self.override_router():
|
||||
auth.book_set.all().delete()
|
||||
self.fail('db_for_write() not invoked on router')
|
||||
except RouterUsed as e:
|
||||
|
@ -2052,8 +1990,8 @@ 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:
|
||||
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:
|
||||
|
@ -2066,8 +2004,8 @@ class RouteForWriteTestCase(TestCase):
|
|||
book = Book.objects.create(title="Pro Django",
|
||||
published=datetime.date(2008, 12, 16))
|
||||
book.authors.add(auth)
|
||||
self.enable_router()
|
||||
try:
|
||||
with self.override_router():
|
||||
auth.book_set.remove(book)
|
||||
self.fail('db_for_write() not invoked on router')
|
||||
except RouterUsed as e:
|
||||
|
@ -2080,8 +2018,8 @@ class RouteForWriteTestCase(TestCase):
|
|||
book = Book.objects.create(title="Pro Django",
|
||||
published=datetime.date(2008, 12, 16))
|
||||
book.authors.add(auth)
|
||||
self.enable_router()
|
||||
try:
|
||||
with self.override_router():
|
||||
auth.book_set.all().update(title='Different')
|
||||
self.fail('db_for_write() not invoked on router')
|
||||
except RouterUsed as e:
|
||||
|
|
|
@ -6,9 +6,10 @@ import unittest
|
|||
from django.conf import settings
|
||||
from django.db import transaction, connection, router
|
||||
from django.db.utils import ConnectionHandler, DEFAULT_DB_ALIAS, DatabaseError
|
||||
from django.test import (TransactionTestCase, skipIfDBFeature,
|
||||
skipUnlessDBFeature)
|
||||
from django.test import override_settings
|
||||
from django.test import (
|
||||
TransactionTestCase, override_settings, skipIfDBFeature,
|
||||
skipUnlessDBFeature,
|
||||
)
|
||||
|
||||
from multiple_database.routers import TestRouter
|
||||
|
||||
|
@ -257,14 +258,10 @@ class SelectForUpdateTests(TransactionTestCase):
|
|||
self.assertIsInstance(status[-1], DatabaseError)
|
||||
|
||||
@skipUnlessDBFeature('has_select_for_update')
|
||||
@override_settings(DATABASE_ROUTERS=[TestRouter()])
|
||||
def test_select_for_update_on_multidb(self):
|
||||
old_routers = router.routers
|
||||
try:
|
||||
router.routers = [TestRouter()]
|
||||
query = Person.objects.select_for_update()
|
||||
self.assertEqual(router.db_for_write(Person), query.db)
|
||||
finally:
|
||||
router.routers = old_routers
|
||||
|
||||
@skipUnlessDBFeature('has_select_for_update')
|
||||
def test_select_for_update_with_get(self):
|
||||
|
|
Loading…
Reference in New Issue