From 3b713bcc90045aaea645285a81da191adf70ba62 Mon Sep 17 00:00:00 2001 From: Quantum Date: Sat, 15 Jul 2017 02:51:46 -0400 Subject: [PATCH] Add a test for Django 1.10+ middleware interface. --- csp_advanced/tests.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/csp_advanced/tests.py b/csp_advanced/tests.py index 7c02db3..810eb0c 100644 --- a/csp_advanced/tests.py +++ b/csp_advanced/tests.py @@ -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'")