diff options
Diffstat (limited to 'tests/template_tests/test_response.py')
-rw-r--r-- | tests/template_tests/test_response.py | 22 |
1 files changed, 12 insertions, 10 deletions
diff --git a/tests/template_tests/test_response.py b/tests/template_tests/test_response.py index e7f71be4f1..3e99134afb 100644 --- a/tests/template_tests/test_response.py +++ b/tests/template_tests/test_response.py @@ -77,7 +77,8 @@ class SimpleTemplateResponseTest(SimpleTestCase): def iteration(): for x in response: pass - self.assertRaises(ContentNotRenderedError, iteration) + with self.assertRaises(ContentNotRenderedError): + iteration() self.assertFalse(response.is_rendered) def test_iteration_rendered(self): @@ -90,7 +91,8 @@ class SimpleTemplateResponseTest(SimpleTestCase): # unrendered response raises an exception when content is accessed response = self._response() self.assertFalse(response.is_rendered) - self.assertRaises(ContentNotRenderedError, lambda: response.content) + with self.assertRaises(ContentNotRenderedError): + response.content self.assertFalse(response.is_rendered) def test_content_access_rendered(self): @@ -160,8 +162,8 @@ class SimpleTemplateResponseTest(SimpleTestCase): 'value': 123, 'fn': datetime.now, }) - self.assertRaises(ContentNotRenderedError, - pickle.dumps, response) + with self.assertRaises(ContentNotRenderedError): + pickle.dumps(response) # But if we render the response, we can pickle it. response.render() @@ -188,8 +190,8 @@ class SimpleTemplateResponseTest(SimpleTestCase): 'value': 123, 'fn': datetime.now, }) - self.assertRaises(ContentNotRenderedError, - pickle.dumps, response) + with self.assertRaises(ContentNotRenderedError): + pickle.dumps(response) response.render() pickled_response = pickle.dumps(response) @@ -274,8 +276,8 @@ class TemplateResponseTest(SimpleTestCase): 'fn': datetime.now, } ) - self.assertRaises(ContentNotRenderedError, - pickle.dumps, response) + with self.assertRaises(ContentNotRenderedError): + pickle.dumps(response) # But if we render the response, we can pickle it. response.render() @@ -307,8 +309,8 @@ class TemplateResponseTest(SimpleTestCase): 'value': 123, 'fn': datetime.now, }) - self.assertRaises(ContentNotRenderedError, - pickle.dumps, response) + with self.assertRaises(ContentNotRenderedError): + pickle.dumps(response) response.render() pickled_response = pickle.dumps(response) |