diff options
Diffstat (limited to 'tests/template_tests')
| -rw-r--r-- | tests/template_tests/test_response.py | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/tests/template_tests/test_response.py b/tests/template_tests/test_response.py index 0a51d68f01..8ac629f3bd 100644 --- a/tests/template_tests/test_response.py +++ b/tests/template_tests/test_response.py @@ -216,6 +216,14 @@ class SimpleTemplateResponseTest(SimpleTestCase): self.assertEqual(unpickled_response.cookies['key'].value, 'value') + def test_headers(self): + response = SimpleTemplateResponse( + 'first/test.html', + {'value': 123, 'fn': datetime.now}, + headers={'X-Foo': 'foo'}, + ) + self.assertEqual(response.headers['X-Foo'], 'foo') + @override_settings(TEMPLATES=[{ 'BACKEND': 'django.template.backends.django.DjangoTemplates', @@ -319,6 +327,15 @@ class TemplateResponseTest(SimpleTestCase): unpickled_response = pickle.loads(pickled_response) pickle.dumps(unpickled_response) + def test_headers(self): + response = TemplateResponse( + self.factory.get('/'), + 'first/test.html', + {'value': 123, 'fn': datetime.now}, + headers={'X-Foo': 'foo'}, + ) + self.assertEqual(response.headers['X-Foo'], 'foo') + @modify_settings(MIDDLEWARE={'append': ['template_tests.test_response.custom_urlconf_middleware']}) @override_settings(ROOT_URLCONF='template_tests.urls') |