Add a test for numeric status code responses

python-websockets · Oct 7, 2023 · 9aa60ec · 9aa60ec
1 parent 60a172e
commit 9aa60ec
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/legacy/test_client_server.py b/tests/legacy/test_client_server.py
@@ -543,6 +543,17 @@ def test_process_request_override(self):
         with contextlib.closing(response):
             self.assertEqual(response.code, 200)
 
+    class ProcessRequestOKServerProtocolNumeric(WebSocketServerProtocol):
+        async def process_request(self, path, request_headers):
+            return 200, [], b"OK\n"
+
+    @with_server(create_protocol=ProcessRequestOKServerProtocolNumeric)
+    def test_numeric_response_code(self):
+        response = self.loop.run_until_complete(self.make_http_request("/"))
+
+        with contextlib.closing(response):
+            self.assertEqual(response.code, 200)
+
     class LegacyProcessRequestOKServerProtocol(WebSocketServerProtocol):
         def process_request(self, path, request_headers):
             return http.HTTPStatus.OK, [], b"OK\n"