Add a test for Django 1.10+ middleware interface.

This commit is contained in:
Quantum 2017-07-15 02:51:46 -04:00
parent 591ea6f2fd
commit 3b713bcc90

View file

@ -164,6 +164,11 @@ class TestMiddleware(SimpleTestCase):
def test_no_csp(self):
self.assertRaises(MiddlewareNotUsed, self.decorator_factory)
@override_settings(ADVANCED_CSP={'script-src': ['self']})
def test_new_style(self):
middleware = AdvancedCSPMiddleware(lambda request: HttpResponse('ok'))
self.assertEqual(middleware(self.get_request())['Content-Security-Policy'], "script-src 'self'")
@override_settings(ADVANCED_CSP={'script-src': ['self']})
def test_setting_csp(self):
self.assertEqual(self.make_ok_view()(self.get_request())['Content-Security-Policy'], "script-src 'self'")