mirrors/synapse
1
0
Fork 0
mirror of https://github.com/element-hq/synapse.git synced 2024-12-20 10:55:09 +03:00
Code Issues Projects Releases Packages Wiki Activity

Fix tests relying on headers not being Headers

Browse source
This commit is contained in:
Olivier 'reivilibre 2024-07-30 12:40:16 +01:00
parent 3b8ed30163
commit 98876d342f
2 changed files with 5 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/rest/client/test_login.py
View file

@ -969,9 +969,8 @@ class CASTestCase(unittest.HomeserverTestCase):
# Test that the response is HTML. # Test that the response is HTML.
self.assertEqual(channel.code, 200, channel.result) self.assertEqual(channel.code, 200, channel.result)
content_type_header_value = "" content_type_header_value = ""
for header in channel.result.get("headers", []): for header in channel.headers.getRawHeaders("Content-Type"):
if header[0] == b"Content-Type": content_type_header_value = header
content_type_header_value = header[1].decode("utf8")
self.assertTrue(content_type_header_value.startswith("text/html")) self.assertTrue(content_type_header_value.startswith("text/html"))

6
tests/test_server.py
View file

@ -393,7 +393,7 @@ class WrapHtmlRequestHandlerTests(unittest.TestCase):
self.assertEqual(channel.code, 301) self.assertEqual(channel.code, 301)
headers = channel.result["headers"] headers = channel.result["headers"]
location_headers = [v for k, v in headers if k == b"Location"] location_headers = headers.getRawHeaders(b"Location", [])
self.assertEqual(location_headers, [b"/look/an/eagle"]) self.assertEqual(location_headers, [b"/look/an/eagle"])
def test_redirect_exception_with_cookie(self) -> None: def test_redirect_exception_with_cookie(self) -> None:
@ -416,9 +416,9 @@ class WrapHtmlRequestHandlerTests(unittest.TestCase):
self.assertEqual(channel.code, 304) self.assertEqual(channel.code, 304)
headers = channel.result["headers"] headers = channel.result["headers"]
location_headers = [v for k, v in headers if k == b"Location"] location_headers = headers.getRawHeaders(b"Location", [])
self.assertEqual(location_headers, [b"/no/over/there"]) self.assertEqual(location_headers, [b"/no/over/there"])
cookies_headers = [v for k, v in headers if k == b"Set-Cookie"] cookies_headers = headers.getRawHeaders(b"Set-Cookie", [])
self.assertEqual(cookies_headers, [b"session=yespls"]) self.assertEqual(cookies_headers, [b"session=yespls"])
def test_head_request(self) -> None: def test_head_request(self) -> None: