Tests: improved response receiving while upgrade.

The patch required to process non-101 response.
2020-09-22 12:40:35 +03:00
parent 767c4cb508
commit dc49c561e2
1 changed files with 1 additions and 4 deletions
--- a/test/unit/applications/websockets.py
+++ b/test/unit/applications/websockets.py
@@ -58,10 +58,7 @@ class TestApplicationWebsocket(TestApplicationProto):

            resp += sock.recv(4096).decode()

-            if (
-                re.search('101 Switching Protocols', resp)
-                and resp[-4:] == '\r\n\r\n'
-            ):
+            if (resp.startswith('HTTP/') and '\r\n\r\n' in resp):
                resp = self._resp_to_dict(resp)
                break