diff options
| author | David Smith <smithdc@gmail.com> | 2020-12-16 19:56:18 +0000 |
|---|---|---|
| committer | Mariusz Felisiak <felisiak.mariusz@gmail.com> | 2020-12-17 08:27:56 +0100 |
| commit | 8f384505eee8ce95667d77cfc2a07d4abe63557c (patch) | |
| tree | cf410102bd900e56a76ca036aea5f126c71b4613 /tests/admin_registration/tests.py | |
| parent | b9ba85a7ce582f4821a21119a6c130dacab496c4 (diff) | |
Refs #32273 -- Completed test coverage for AdminSite.unregister().
Diffstat (limited to 'tests/admin_registration/tests.py')
| -rw-r--r-- | tests/admin_registration/tests.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/tests/admin_registration/tests.py b/tests/admin_registration/tests.py index 93031bfbf3..d165ee93b2 100644 --- a/tests/admin_registration/tests.py +++ b/tests/admin_registration/tests.py @@ -23,10 +23,14 @@ class TestRegistration(SimpleTestCase): def test_bare_registration(self): self.site.register(Person) self.assertIsInstance(self.site._registry[Person], admin.ModelAdmin) + self.site.unregister(Person) + self.assertEqual(self.site._registry, {}) def test_registration_with_model_admin(self): self.site.register(Person, NameAdmin) self.assertIsInstance(self.site._registry[Person], NameAdmin) + self.site.unregister(Person) + self.assertEqual(self.site._registry, {}) def test_prevent_double_registration(self): self.site.register(Person) @@ -43,6 +47,11 @@ class TestRegistration(SimpleTestCase): with self.assertRaisesMessage(admin.sites.AlreadyRegistered, msg): self.site.register(Person, PersonAdmin) + def test_unregister_unregistered_model(self): + msg = 'The model Person is not registered' + with self.assertRaisesMessage(admin.sites.NotRegistered, msg): + self.site.unregister(Person) + def test_registration_with_star_star_options(self): self.site.register(Person, search_fields=['name']) self.assertEqual(self.site._registry[Person].search_fields, ['name']) @@ -59,6 +68,8 @@ class TestRegistration(SimpleTestCase): self.assertEqual(self.site._registry[Person].search_fields, ['name']) self.assertIsInstance(self.site._registry[Place], admin.ModelAdmin) self.assertEqual(self.site._registry[Place].search_fields, ['name']) + self.site.unregister([Person, Place]) + self.assertEqual(self.site._registry, {}) def test_abstract_model(self): """ |
