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'")