summaryrefslogtreecommitdiff
path: root/tests/admin_views/test_multidb.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/admin_views/test_multidb.py')
-rw-r--r--tests/admin_views/test_multidb.py38
1 files changed, 23 insertions, 15 deletions
diff --git a/tests/admin_views/test_multidb.py b/tests/admin_views/test_multidb.py
index 7ff1f7f663..da5b6b9a59 100644
--- a/tests/admin_views/test_multidb.py
+++ b/tests/admin_views/test_multidb.py
@@ -17,17 +17,17 @@ class Router:
db_for_write = db_for_read
-site = admin.AdminSite(name='test_adminsite')
+site = admin.AdminSite(name="test_adminsite")
site.register(Book)
urlpatterns = [
- path('admin/', site.urls),
+ path("admin/", site.urls),
]
-@override_settings(ROOT_URLCONF=__name__, DATABASE_ROUTERS=['%s.Router' % __name__])
+@override_settings(ROOT_URLCONF=__name__, DATABASE_ROUTERS=["%s.Router" % __name__])
class MultiDatabaseTests(TestCase):
- databases = {'default', 'other'}
+ databases = {"default", "other"}
@classmethod
def setUpTestData(cls):
@@ -36,44 +36,52 @@ class MultiDatabaseTests(TestCase):
for db in cls.databases:
Router.target_db = db
cls.superusers[db] = User.objects.create_superuser(
- username='admin', password='something', email='test@test.org',
+ username="admin",
+ password="something",
+ email="test@test.org",
)
- b = Book(name='Test Book')
+ b = Book(name="Test Book")
b.save(using=db)
cls.test_book_ids[db] = b.id
- @mock.patch('django.contrib.admin.options.transaction')
+ @mock.patch("django.contrib.admin.options.transaction")
def test_add_view(self, mock):
for db in self.databases:
with self.subTest(db=db):
Router.target_db = db
self.client.force_login(self.superusers[db])
self.client.post(
- reverse('test_adminsite:admin_views_book_add'),
- {'name': 'Foobar: 5th edition'},
+ reverse("test_adminsite:admin_views_book_add"),
+ {"name": "Foobar: 5th edition"},
)
mock.atomic.assert_called_with(using=db)
- @mock.patch('django.contrib.admin.options.transaction')
+ @mock.patch("django.contrib.admin.options.transaction")
def test_change_view(self, mock):
for db in self.databases:
with self.subTest(db=db):
Router.target_db = db
self.client.force_login(self.superusers[db])
self.client.post(
- reverse('test_adminsite:admin_views_book_change', args=[self.test_book_ids[db]]),
- {'name': 'Test Book 2: Test more'},
+ reverse(
+ "test_adminsite:admin_views_book_change",
+ args=[self.test_book_ids[db]],
+ ),
+ {"name": "Test Book 2: Test more"},
)
mock.atomic.assert_called_with(using=db)
- @mock.patch('django.contrib.admin.options.transaction')
+ @mock.patch("django.contrib.admin.options.transaction")
def test_delete_view(self, mock):
for db in self.databases:
with self.subTest(db=db):
Router.target_db = db
self.client.force_login(self.superusers[db])
self.client.post(
- reverse('test_adminsite:admin_views_book_delete', args=[self.test_book_ids[db]]),
- {'post': 'yes'},
+ reverse(
+ "test_adminsite:admin_views_book_delete",
+ args=[self.test_book_ids[db]],
+ ),
+ {"post": "yes"},
)
mock.atomic.assert_called_with(using=db)