diff options
Diffstat (limited to 'tests/auth_tests/test_admin_multidb.py')
| -rw-r--r-- | tests/auth_tests/test_admin_multidb.py | 27 |
1 files changed, 16 insertions, 11 deletions
diff --git a/tests/auth_tests/test_admin_multidb.py b/tests/auth_tests/test_admin_multidb.py index fac2c0fc7e..20faab25d8 100644 --- a/tests/auth_tests/test_admin_multidb.py +++ b/tests/auth_tests/test_admin_multidb.py @@ -16,17 +16,17 @@ class Router: db_for_write = db_for_read -site = admin.AdminSite(name='test_adminsite') +site = admin.AdminSite(name="test_adminsite") site.register(User, admin_class=UserAdmin) 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): @@ -34,18 +34,23 @@ 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", ) - @mock.patch('django.contrib.auth.admin.transaction') + @mock.patch("django.contrib.auth.admin.transaction") def test_add_view(self, mock): for db in self.databases: with self.subTest(db_connection=db): Router.target_db = db self.client.force_login(self.superusers[db]) - self.client.post(reverse('test_adminsite:auth_user_add'), { - 'username': 'some_user', - 'password1': 'helloworld', - 'password2': 'helloworld', - }) + self.client.post( + reverse("test_adminsite:auth_user_add"), + { + "username": "some_user", + "password1": "helloworld", + "password2": "helloworld", + }, + ) mock.atomic.assert_called_with(using=db) |