summaryrefslogtreecommitdiff
path: root/tests/check_framework/test_database.py
diff options
context:
space:
mode:
authordjango-bot <ops@djangoproject.com>2022-02-03 20:24:19 +0100
committerMariusz Felisiak <felisiak.mariusz@gmail.com>2022-02-07 20:37:05 +0100
commit9c19aff7c7561e3a82978a272ecdaad40dda5c00 (patch)
treef0506b668a013d0063e5fba3dbf4863b466713ba /tests/check_framework/test_database.py
parentf68fa8b45dfac545cfc4111d4e52804c86db68d3 (diff)
Refs #33476 -- Reformatted code with Black.
Diffstat (limited to 'tests/check_framework/test_database.py')
-rw-r--r--tests/check_framework/test_database.py30
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()