diff options
Diffstat (limited to 'tests/admin_registration')
| -rw-r--r-- | tests/admin_registration/tests.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/admin_registration/tests.py b/tests/admin_registration/tests.py index ffca76886e..c1efae76c7 100644 --- a/tests/admin_registration/tests.py +++ b/tests/admin_registration/tests.py @@ -127,12 +127,12 @@ class TestRegistrationDecorator(SimpleTestCase): self.default_site.unregister(Place) def test_wrapped_class_not_a_model_admin(self): - self.assertRaisesMessage(ValueError, 'Wrapped class must subclass ModelAdmin.', - register(Person), CustomSite) + with self.assertRaisesMessage(ValueError, 'Wrapped class must subclass ModelAdmin.'): + register(Person)(CustomSite) def test_custom_site_not_an_admin_site(self): - self.assertRaisesMessage(ValueError, 'site must subclass AdminSite', - register(Person, site=Traveler), NameAdmin) + with self.assertRaisesMessage(ValueError, 'site must subclass AdminSite'): + register(Person, site=Traveler)(NameAdmin) def test_empty_models_list_registration_fails(self): with self.assertRaisesMessage(ValueError, 'At least one model must be passed to register.'): |