diff options
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() |