Test robustness for pypy build #nolog

miguelgrinberg · Jun 4, 2023 · f5f2aa7 · f5f2aa7
1 parent 5a5325b
commit f5f2aa7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/test_simple_websocket_client.py b/tests/test_simple_websocket_client.py
@@ -66,6 +66,11 @@ def test_make_client_headers(self, mock_wsconn, mock_socket):
         client.sock.send.assert_called_with(
             b"Request(host='example.com', target='/ws?a=1', extensions=[], "
             b"extra_headers=[('Foo', 'Bar')], subprotocols=[])")
+
+    @mock.patch('simple_websocket.ws.socket.socket')
+    @mock.patch('simple_websocket.ws.WSConnection')
+    def test_make_client_headers2(self, mock_wsconn, mock_socket):
+        mock_socket.return_value.recv.return_value = b'x'
         client = self.get_client(mock_wsconn, 'ws://example.com/ws?a=1',
                                  headers=[('Foo', 'Bar'), ('Foo', 'Baz')])
         client.sock.send.assert_called_with(