diff options
Diffstat (limited to 'tests/get_object_or_404/tests.py')
| -rw-r--r-- | tests/get_object_or_404/tests.py | 27 |
1 files changed, 9 insertions, 18 deletions
diff --git a/tests/get_object_or_404/tests.py b/tests/get_object_or_404/tests.py index 55d1de7c2c..ac4b2d4738 100644 --- a/tests/get_object_or_404/tests.py +++ b/tests/get_object_or_404/tests.py @@ -81,27 +81,18 @@ class GetObjectOr404Tests(TestCase): def test_bad_class(self): # Given an argument klass that is not a Model, Manager, or Queryset # raises a helpful ValueError message - self.assertRaisesMessage( - ValueError, - "Object is of type 'str', but must be a Django Model, Manager, " - "or QuerySet", - get_object_or_404, str("Article"), title__icontains="Run" - ) + msg = "Object is of type 'str', but must be a Django Model, Manager, or QuerySet" + with self.assertRaisesMessage(ValueError, msg): + get_object_or_404(str("Article"), title__icontains="Run") class CustomClass(object): pass - self.assertRaisesMessage( - ValueError, - "Object is of type 'CustomClass', but must be a Django Model, " - "Manager, or QuerySet", - get_object_or_404, CustomClass, title__icontains="Run" - ) + msg = "Object is of type 'CustomClass', but must be a Django Model, Manager, or QuerySet" + with self.assertRaisesMessage(ValueError, msg): + get_object_or_404(CustomClass, title__icontains="Run") # Works for lists too - self.assertRaisesMessage( - ValueError, - "Object is of type 'list', but must be a Django Model, Manager, " - "or QuerySet", - get_list_or_404, [Article], title__icontains="Run" - ) + msg = "Object is of type 'list', but must be a Django Model, Manager, or QuerySet" + with self.assertRaisesMessage(ValueError, msg): + get_list_or_404([Article], title__icontains="Run") |