diff options
author | django-bot <ops@djangoproject.com> | 2022-02-03 20:24:19 +0100 |
---|---|---|
committer | Mariusz Felisiak <felisiak.mariusz@gmail.com> | 2022-02-07 20:37:05 +0100 |
commit | 9c19aff7c7561e3a82978a272ecdaad40dda5c00 (patch) | |
tree | f0506b668a013d0063e5fba3dbf4863b466713ba /tests/swappable_models | |
parent | f68fa8b45dfac545cfc4111d4e52804c86db68d3 (diff) | |
download | django-9c19aff7c7561e3a82978a272ecdaad40dda5c00.tar.gz |
Refs #33476 -- Reformatted code with Black.
Diffstat (limited to 'tests/swappable_models')
-rw-r--r-- | tests/swappable_models/models.py | 2 | ||||
-rw-r--r-- | tests/swappable_models/tests.py | 37 |
2 files changed, 20 insertions, 19 deletions
diff --git a/tests/swappable_models/models.py b/tests/swappable_models/models.py index c5b4425489..02a1eb99c0 100644 --- a/tests/swappable_models/models.py +++ b/tests/swappable_models/models.py @@ -6,7 +6,7 @@ class Article(models.Model): publication_date = models.DateField() class Meta: - swappable = 'TEST_ARTICLE_MODEL' + swappable = "TEST_ARTICLE_MODEL" class AlternateArticle(models.Model): diff --git a/tests/swappable_models/tests.py b/tests/swappable_models/tests.py index b1bcaa9f70..a2d043cb56 100644 --- a/tests/swappable_models/tests.py +++ b/tests/swappable_models/tests.py @@ -10,35 +10,36 @@ class SwappableModelTests(TestCase): # Limit memory usage when calling 'migrate'. available_apps = [ - 'swappable_models', - 'django.contrib.auth', - 'django.contrib.contenttypes', + "swappable_models", + "django.contrib.auth", + "django.contrib.contenttypes", ] - @override_settings(TEST_ARTICLE_MODEL='swappable_models.AlternateArticle') + @override_settings(TEST_ARTICLE_MODEL="swappable_models.AlternateArticle") def test_generated_data(self): "Permissions and content types are not created for a swapped model" # Delete all permissions and content_types - Permission.objects.filter(content_type__app_label='swappable_models').delete() - ContentType.objects.filter(app_label='swappable_models').delete() + Permission.objects.filter(content_type__app_label="swappable_models").delete() + ContentType.objects.filter(app_label="swappable_models").delete() # Re-run migrate. This will re-build the permissions and content types. - management.call_command('migrate', interactive=False, verbosity=0) + management.call_command("migrate", interactive=False, verbosity=0) # Content types and permissions exist for the swapped model, # but not for the swappable model. - apps_models = [(p.content_type.app_label, p.content_type.model) - for p in Permission.objects.all()] - self.assertIn(('swappable_models', 'alternatearticle'), apps_models) - self.assertNotIn(('swappable_models', 'article'), apps_models) - - apps_models = [(ct.app_label, ct.model) - for ct in ContentType.objects.all()] - self.assertIn(('swappable_models', 'alternatearticle'), apps_models) - self.assertNotIn(('swappable_models', 'article'), apps_models) - - @override_settings(TEST_ARTICLE_MODEL='swappable_models.article') + apps_models = [ + (p.content_type.app_label, p.content_type.model) + for p in Permission.objects.all() + ] + self.assertIn(("swappable_models", "alternatearticle"), apps_models) + self.assertNotIn(("swappable_models", "article"), apps_models) + + apps_models = [(ct.app_label, ct.model) for ct in ContentType.objects.all()] + self.assertIn(("swappable_models", "alternatearticle"), apps_models) + self.assertNotIn(("swappable_models", "article"), apps_models) + + @override_settings(TEST_ARTICLE_MODEL="swappable_models.article") def test_case_insensitive(self): "Model names are case insensitive. Model swapping honors this." Article.objects.all() |