|
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