diff options
Diffstat (limited to 'tests/check_framework/test_multi_db.py')
| -rw-r--r-- | tests/check_framework/test_multi_db.py | 20 |
1 files changed, 10 insertions, 10 deletions
diff --git a/tests/check_framework/test_multi_db.py b/tests/check_framework/test_multi_db.py index 0697b3b94a..aca0a1f472 100644 --- a/tests/check_framework/test_multi_db.py +++ b/tests/check_framework/test_multi_db.py @@ -9,25 +9,25 @@ class TestRouter: """ Routes to the 'other' database if the model name starts with 'Other'. """ + def allow_migrate(self, db, app_label, model_name=None, **hints): - return db == ('other' if model_name.startswith('other') else 'default') + return db == ("other" if model_name.startswith("other") else "default") @override_settings(DATABASE_ROUTERS=[TestRouter()]) -@isolate_apps('check_framework') +@isolate_apps("check_framework") class TestMultiDBChecks(SimpleTestCase): - def _patch_check_field_on(self, db): - return mock.patch.object(connections[db].validation, 'check_field') + return mock.patch.object(connections[db].validation, "check_field") def test_checks_called_on_the_default_database(self): class Model(models.Model): field = models.CharField(max_length=100) model = Model() - with self._patch_check_field_on('default') as mock_check_field_default: - with self._patch_check_field_on('other') as mock_check_field_other: - model.check(databases={'default', 'other'}) + with self._patch_check_field_on("default") as mock_check_field_default: + with self._patch_check_field_on("other") as mock_check_field_other: + model.check(databases={"default", "other"}) self.assertTrue(mock_check_field_default.called) self.assertFalse(mock_check_field_other.called) @@ -36,8 +36,8 @@ class TestMultiDBChecks(SimpleTestCase): field = models.CharField(max_length=100) model = OtherModel() - with self._patch_check_field_on('other') as mock_check_field_other: - with self._patch_check_field_on('default') as mock_check_field_default: - model.check(databases={'default', 'other'}) + with self._patch_check_field_on("other") as mock_check_field_other: + with self._patch_check_field_on("default") as mock_check_field_default: + model.check(databases={"default", "other"}) self.assertTrue(mock_check_field_other.called) self.assertFalse(mock_check_field_default.called) |