django/daphne
1
1
Fork 0
mirror of https://github.com/django/daphne.git synced 2024-11-25 01:03:45 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix header matching in websocket tests

Browse Source
This commit is contained in:
Andrew Godwin 2018-02-01 21:02:27 -08:00
parent 228142cab5
commit 51c2de3f8c
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_websocket.py
View File

@ -52,12 +52,20 @@ class TestWebsocket(DaphneTestCase):
# {name: [value1, value2, ...]} and check if they're equal. # {name: [value1, value2, ...]} and check if they're equal.
transformed_scope_headers = collections.defaultdict(list) transformed_scope_headers = collections.defaultdict(list)
for name, value in scope["headers"]: for name, value in scope["headers"]:
transformed_scope_headers[name].append(value) transformed_scope_headers[name] = []
# Make sure to split out any headers collapsed with commas
for bit in value.split(b","):
if bit.strip():
transformed_scope_headers[name].append(bit.strip())
transformed_request_headers = collections.defaultdict(list) transformed_request_headers = collections.defaultdict(list)
for name, value in (headers or []): for name, value in (headers or []):
expected_name = name.lower().strip().encode("ascii") expected_name = name.lower().strip().encode("ascii")
expected_value = value.strip().encode("ascii") expected_value = value.strip().encode("ascii")
transformed_request_headers[expected_name].append(expected_value) transformed_request_headers[expected_name] = []
# Make sure to split out any headers collapsed with commas
for bit in expected_value.split(b","):
if bit.strip():
transformed_request_headers[expected_name].append(bit.strip())
for name, value in transformed_request_headers.items(): for name, value in transformed_request_headers.items():
self.assertIn(name, transformed_scope_headers) self.assertIn(name, transformed_scope_headers)
self.assertEqual(value, transformed_scope_headers[name]) self.assertEqual(value, transformed_scope_headers[name])