|
11 | 11 |
|
12 | 12 | from csp.constants import HEADER, HEADER_REPORT_ONLY, SELF |
13 | 13 | from csp.exceptions import CSPNonceError |
14 | | -from csp.middleware import ( |
15 | | - CheckableLazyObject, |
16 | | - CSPMiddleware, |
17 | | - CSPMiddlewareAlwaysGenerateNonce, |
18 | | -) |
| 14 | +from csp.middleware import CheckableLazyObject, CSPMiddleware |
19 | 15 | from csp.tests.utils import response |
20 | 16 |
|
21 | 17 | mw = CSPMiddleware(response()) |
@@ -291,44 +287,3 @@ def test_set_nonce_access_after_middleware_is_ok() -> None: |
291 | 287 | mw.process_response(request, HttpResponse()) |
292 | 288 | assert bool(getattr(request, "csp_nonce", False)) is True |
293 | 289 | assert str(getattr(request, "csp_nonce")) == nonce |
294 | | - |
295 | | - |
296 | | -def test_csp_always_nonce_middleware_has_nonce() -> None: |
297 | | - request = rf.get("/") |
298 | | - mw_agn = CSPMiddlewareAlwaysGenerateNonce(response()) |
299 | | - mw_agn.process_request(request) |
300 | | - nonce = getattr(request, "csp_nonce") |
301 | | - assert bool(nonce) is True |
302 | | - resp = HttpResponse() |
303 | | - mw_agn.process_response(request, resp) |
304 | | - assert str(nonce) in resp[HEADER] |
305 | | - |
306 | | - |
307 | | -def test_csp_always_nonce_middleware_nonce_regenerated_on_new_request() -> None: |
308 | | - mw_agn = CSPMiddlewareAlwaysGenerateNonce(response()) |
309 | | - request1 = rf.get("/") |
310 | | - request2 = rf.get("/") |
311 | | - mw_agn.process_request(request1) |
312 | | - mw_agn.process_request(request2) |
313 | | - nonce1 = str(getattr(request1, "csp_nonce")) |
314 | | - nonce2 = str(getattr(request2, "csp_nonce")) |
315 | | - assert nonce1 != nonce2 |
316 | | - |
317 | | - response1 = HttpResponse() |
318 | | - response2 = HttpResponse() |
319 | | - mw_agn.process_response(request1, response1) |
320 | | - mw_agn.process_response(request2, response2) |
321 | | - assert nonce1 not in response2[HEADER] |
322 | | - assert nonce2 not in response1[HEADER] |
323 | | - |
324 | | - |
325 | | -def test_csp_always_nonce_middleware_access_after_middleware_is_ok() -> None: |
326 | | - # Test accessing a set nonce after the response has been processed is OK. |
327 | | - request = rf.get("/") |
328 | | - mw_agn = CSPMiddlewareAlwaysGenerateNonce(response()) |
329 | | - mw_agn.process_request(request) |
330 | | - nonce = getattr(request, "csp_nonce") |
331 | | - assert bool(nonce) is True |
332 | | - mw_agn.process_response(request, HttpResponse()) |
333 | | - assert bool(nonce) is True |
334 | | - assert str(getattr(request, "csp_nonce")) == nonce |
0 commit comments