diff options
| author | Hasan <hasan.r67@gmail.com> | 2016-01-04 12:20:08 +0330 |
|---|---|---|
| committer | Tim Graham <timograham@gmail.com> | 2016-01-29 13:03:39 -0500 |
| commit | 253adc2b8a52982139d40c4f55b3fd446e1cb8f3 (patch) | |
| tree | c508d48636f5b37e97c8078737d398d7475ff8cc /tests/admin_registration | |
| parent | 3d0dcd7f5af378d3ab6adb303b913e6c7b2e0ee5 (diff) | |
| download | django-253adc2b8a52982139d40c4f55b3fd446e1cb8f3.tar.gz | |
Refs #26022 -- Used context manager version of assertRaisesMessage in tests.
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.'): |