diff options
| author | django-bot <ops@djangoproject.com> | 2022-02-08 12:09:55 +0100 |
|---|---|---|
| committer | Mariusz Felisiak <felisiak.mariusz@gmail.com> | 2022-02-08 12:15:38 +0100 |
| commit | 6a682b38e75d4c975b4c4493565a59f1bc14397c (patch) | |
| tree | 0bd9cda550bea26238656d9f120d769e8b41bb9e /tests/check_framework/test_database.py | |
| parent | e73ce08888e6f34d3f050377cfd2fbb733be94a9 (diff) | |
[4.0.x] Refs #33476 -- Reformatted code with Black.
Backport of 9c19aff7c7561e3a82978a272ecdaad40dda5c00 from main.
Diffstat (limited to 'tests/check_framework/test_database.py')
| -rw-r--r-- | tests/check_framework/test_database.py | 30 |
1 files changed, 18 insertions, 12 deletions
diff --git a/tests/check_framework/test_database.py b/tests/check_framework/test_database.py index c9bc8866e7..8f786474f3 100644 --- a/tests/check_framework/test_database.py +++ b/tests/check_framework/test_database.py @@ -7,46 +7,52 @@ from django.test import TestCase class DatabaseCheckTests(TestCase): - databases = {'default', 'other'} + databases = {"default", "other"} - @mock.patch('django.db.backends.base.validation.BaseDatabaseValidation.check') + @mock.patch("django.db.backends.base.validation.BaseDatabaseValidation.check") def test_database_checks_called(self, mocked_check): check_database_backends() self.assertFalse(mocked_check.called) check_database_backends(databases=self.databases) self.assertTrue(mocked_check.called) - @unittest.skipUnless(connection.vendor == 'mysql', 'Test only for MySQL') + @unittest.skipUnless(connection.vendor == "mysql", "Test only for MySQL") def test_mysql_strict_mode(self): def _clean_sql_mode(): for alias in self.databases: - if hasattr(connections[alias], 'sql_mode'): + if hasattr(connections[alias], "sql_mode"): del connections[alias].sql_mode _clean_sql_mode() good_sql_modes = [ - 'STRICT_TRANS_TABLES,STRICT_ALL_TABLES', - 'STRICT_TRANS_TABLES', - 'STRICT_ALL_TABLES', + "STRICT_TRANS_TABLES,STRICT_ALL_TABLES", + "STRICT_TRANS_TABLES", + "STRICT_ALL_TABLES", ] for sql_mode in good_sql_modes: with mock.patch.object( - connection, 'mysql_server_data', {'sql_mode': sql_mode}, + connection, + "mysql_server_data", + {"sql_mode": sql_mode}, ): self.assertEqual(check_database_backends(databases=self.databases), []) _clean_sql_mode() - bad_sql_modes = ['', 'WHATEVER'] + bad_sql_modes = ["", "WHATEVER"] for sql_mode in bad_sql_modes: mocker_default = mock.patch.object( - connection, 'mysql_server_data', {'sql_mode': sql_mode}, + connection, + "mysql_server_data", + {"sql_mode": sql_mode}, ) mocker_other = mock.patch.object( - connections['other'], 'mysql_server_data', {'sql_mode': sql_mode}, + connections["other"], + "mysql_server_data", + {"sql_mode": sql_mode}, ) with mocker_default, mocker_other: # One warning for each database alias result = check_database_backends(databases=self.databases) self.assertEqual(len(result), 2) - self.assertEqual([r.id for r in result], ['mysql.W002', 'mysql.W002']) + self.assertEqual([r.id for r in result], ["mysql.W002", "mysql.W002"]) _clean_sql_mode() |
