1 files changed, 236 insertions, 0 deletions
diff --git a/testing/web-platform/tests/tools/third_party/websockets/tests/test_datastructures.py b/testing/web-platform/tests/tools/third_party/websockets/tests/test_datastructures.py
new file mode 100644
index 0000000000..32b79817ae
--- /dev/null
+++ b/testing/web-platform/tests/tools/third_party/websockets/tests/test_datastructures.py
@@ -0,0 +1,236 @@
+import unittest
+
+from websockets.datastructures import *
+
+
+class MultipleValuesErrorTests(unittest.TestCase):
+    def test_multiple_values_error_str(self):
+        self.assertEqual(str(MultipleValuesError("Connection")), "'Connection'")
+        self.assertEqual(str(MultipleValuesError()), "")
+
+
+class HeadersTests(unittest.TestCase):
+    def setUp(self):
+        self.headers = Headers([("Connection", "Upgrade"), ("Server", "websockets")])
+
+    def test_init(self):
+        self.assertEqual(
+            Headers(),
+            Headers(),
+        )
+
+    def test_init_from_kwargs(self):
+        self.assertEqual(
+            Headers(connection="Upgrade", server="websockets"),
+            self.headers,
+        )
+
+    def test_init_from_headers(self):
+        self.assertEqual(
+            Headers(self.headers),
+            self.headers,
+        )
+
+    def test_init_from_headers_and_kwargs(self):
+        self.assertEqual(
+            Headers(Headers(connection="Upgrade"), server="websockets"),
+            self.headers,
+        )
+
+    def test_init_from_mapping(self):
+        self.assertEqual(
+            Headers({"Connection": "Upgrade", "Server": "websockets"}),
+            self.headers,
+        )
+
+    def test_init_from_mapping_and_kwargs(self):
+        self.assertEqual(
+            Headers({"Connection": "Upgrade"}, server="websockets"),
+            self.headers,
+        )
+
+    def test_init_from_iterable(self):
+        self.assertEqual(
+            Headers([("Connection", "Upgrade"), ("Server", "websockets")]),
+            self.headers,
+        )
+
+    def test_init_from_iterable_and_kwargs(self):
+        self.assertEqual(
+            Headers([("Connection", "Upgrade")], server="websockets"),
+            self.headers,
+        )
+
+    def test_init_multiple_positional_arguments(self):
+        with self.assertRaises(TypeError):
+            Headers(Headers(connection="Upgrade"), Headers(server="websockets"))
+
+    def test_str(self):
+        self.assertEqual(
+            str(self.headers), "Connection: Upgrade\r\nServer: websockets\r\n\r\n"
+        )
+
+    def test_repr(self):
+        self.assertEqual(
+            repr(self.headers),
+            "Headers([('Connection', 'Upgrade'), ('Server', 'websockets')])",
+        )
+
+    def test_copy(self):
+        self.assertEqual(repr(self.headers.copy()), repr(self.headers))
+
+    def test_serialize(self):
+        self.assertEqual(
+            self.headers.serialize(),
+            b"Connection: Upgrade\r\nServer: websockets\r\n\r\n",
+        )
+
+    def test_contains(self):
+        self.assertIn("Server", self.headers)
+
+    def test_contains_case_insensitive(self):
+        self.assertIn("server", self.headers)
+
+    def test_contains_not_found(self):
+        self.assertNotIn("Date", self.headers)
+
+    def test_contains_non_string_key(self):
+        self.assertNotIn(42, self.headers)
+
+    def test_iter(self):
+        self.assertEqual(set(iter(self.headers)), {"connection", "server"})
+
+    def test_len(self):
+        self.assertEqual(len(self.headers), 2)
+
+    def test_getitem(self):
+        self.assertEqual(self.headers["Server"], "websockets")
+
+    def test_getitem_case_insensitive(self):
+        self.assertEqual(self.headers["server"], "websockets")
+
+    def test_getitem_key_error(self):
+        with self.assertRaises(KeyError):
+            self.headers["Upgrade"]
+
+    def test_setitem(self):
+        self.headers["Upgrade"] = "websocket"
+        self.assertEqual(self.headers["Upgrade"], "websocket")
+
+    def test_setitem_case_insensitive(self):
+        self.headers["upgrade"] = "websocket"
+        self.assertEqual(self.headers["Upgrade"], "websocket")
+
+    def test_delitem(self):
+        del self.headers["Connection"]
+        with self.assertRaises(KeyError):
+            self.headers["Connection"]
+
+    def test_delitem_case_insensitive(self):
+        del self.headers["connection"]
+        with self.assertRaises(KeyError):
+            self.headers["Connection"]
+
+    def test_eq(self):
+        other_headers = Headers([("Connection", "Upgrade"), ("Server", "websockets")])
+        self.assertEqual(self.headers, other_headers)
+
+    def test_eq_case_insensitive(self):
+        other_headers = Headers(connection="Upgrade", server="websockets")
+        self.assertEqual(self.headers, other_headers)
+
+    def test_eq_not_equal(self):
+        other_headers = Headers([("Connection", "close"), ("Server", "websockets")])
+        self.assertNotEqual(self.headers, other_headers)
+
+    def test_eq_other_type(self):
+        self.assertNotEqual(
+            self.headers, "Connection: Upgrade\r\nServer: websockets\r\n\r\n"
+        )
+
+    def test_clear(self):
+        self.headers.clear()
+        self.assertFalse(self.headers)
+        self.assertEqual(self.headers, Headers())
+
+    def test_get_all(self):
+        self.assertEqual(self.headers.get_all("Connection"), ["Upgrade"])
+
+    def test_get_all_case_insensitive(self):
+        self.assertEqual(self.headers.get_all("connection"), ["Upgrade"])
+
+    def test_get_all_no_values(self):
+        self.assertEqual(self.headers.get_all("Upgrade"), [])
+
+    def test_raw_items(self):
+        self.assertEqual(
+            list(self.headers.raw_items()),
+            [("Connection", "Upgrade"), ("Server", "websockets")],
+        )
+
+
+class MultiValueHeadersTests(unittest.TestCase):
+    def setUp(self):
+        self.headers = Headers([("Server", "Python"), ("Server", "websockets")])
+
+    def test_init_from_headers(self):
+        self.assertEqual(
+            Headers(self.headers),
+            self.headers,
+        )
+
+    def test_init_from_headers_and_kwargs(self):
+        self.assertEqual(
+            Headers(Headers(server="Python"), server="websockets"),
+            self.headers,
+        )
+
+    def test_str(self):
+        self.assertEqual(
+            str(self.headers), "Server: Python\r\nServer: websockets\r\n\r\n"
+        )
+
+    def test_repr(self):
+        self.assertEqual(
+            repr(self.headers),
+            "Headers([('Server', 'Python'), ('Server', 'websockets')])",
+        )
+
+    def test_copy(self):
+        self.assertEqual(repr(self.headers.copy()), repr(self.headers))
+
+    def test_serialize(self):
+        self.assertEqual(
+            self.headers.serialize(),
+            b"Server: Python\r\nServer: websockets\r\n\r\n",
+        )
+
+    def test_iter(self):
+        self.assertEqual(set(iter(self.headers)), {"server"})
+
+    def test_len(self):
+        self.assertEqual(len(self.headers), 1)
+
+    def test_getitem_multiple_values_error(self):
+        with self.assertRaises(MultipleValuesError):
+            self.headers["Server"]
+
+    def test_setitem(self):
+        self.headers["Server"] = "redux"
+        self.assertEqual(
+            self.headers.get_all("Server"), ["Python", "websockets", "redux"]
+        )
+
+    def test_delitem(self):
+        del self.headers["Server"]
+        with self.assertRaises(KeyError):
+            self.headers["Server"]
+
+    def test_get_all(self):
+        self.assertEqual(self.headers.get_all("Server"), ["Python", "websockets"])
+
+    def test_raw_items(self):
+        self.assertEqual(
+            list(self.headers.raw_items()),
+            [("Server", "Python"), ("Server", "websockets")],
+        )