Revert "Fix pending SSL data handling in stream client"

Author: yliaog

kubernetes/base/stream/ws_client.py

@@ -229,9 +229,7 @@ def update(self, timeout=0):
         #                   efficient as epoll. Will work for fd numbers above 1024.
         # select.epoll()  - newest and most efficient way of polling.
         #                   However, only works on linux.
-        if self.sock.is_ssl() and self.sock.sock.pending() > 0:
-            r = [self.sock.sock]
-        elif hasattr(select, "poll"):
+        if hasattr(select, "poll"):
             poll = select.poll()
             poll.register(self.sock.sock, select.POLLIN)
             if timeout is not None and timeout != float("inf"):

kubernetes/base/stream/ws_client_test.py

@@ -18,7 +18,6 @@
 from . import ws_client as ws_client_module
 from .ws_client import get_websocket_url, WSClient, V5_CHANNEL_PROTOCOL, V4_CHANNEL_PROTOCOL, CLOSE_CHANNEL, STDIN_CHANNEL, STDOUT_CHANNEL
 from .ws_client import websocket_proxycare
-from .ws_client import STDOUT_CHANNEL
 from kubernetes.client.configuration import Configuration
 import os
 import select
@@ -327,7 +326,6 @@ def test_update_receives_close_v5(self):
             mock_ws = MagicMock()
             mock_ws.subprotocol = V5_CHANNEL_PROTOCOL
             mock_ws.connected = True
-            mock_ws.is_ssl.return_value = False
             mock_ws.sock.fileno.return_value = 10
 
             # Setup frame with close signal for channel 0
@@ -353,7 +351,6 @@ def test_update_ignores_close_signal_v4(self):
             mock_ws = MagicMock()
             mock_ws.subprotocol = V4_CHANNEL_PROTOCOL
             mock_ws.connected = True
-            mock_ws.is_ssl.return_value = False
             mock_ws.sock.fileno.return_value = 10
 
             # Setup frame that looks like close signal but should be treated as data
@@ -457,7 +454,6 @@ def test_peek_channel_closed_with_leftover_data(self):
             mock_ws = MagicMock()
             mock_ws.subprotocol = V5_CHANNEL_PROTOCOL
             mock_ws.connected = True
-            mock_ws.is_ssl.return_value = False
             mock_ws.sock.fileno.return_value = 10
             mock_create.return_value = mock_ws
 
@@ -493,7 +489,6 @@ def test_update_infinite_timeout_polls_without_overflow(self):
             mock_ws = MagicMock()
             mock_ws.subprotocol = V5_CHANNEL_PROTOCOL
             mock_ws.connected = True
-            mock_ws.is_ssl.return_value = False
             mock_ws.sock.fileno.return_value = 10
             mock_create.return_value = mock_ws
 
@@ -502,46 +497,6 @@ def test_update_infinite_timeout_polls_without_overflow(self):
 
             mock_poll.return_value.poll.assert_called_once_with(None)
 
-    def test_update_reads_pending_ssl_data_without_polling(self):
-        with (
-            patch.object(
-                ws_client_module,
-                'create_websocket',
-            ) as mock_create,
-            patch('select.poll') as mock_poll,
-            patch('select.select') as mock_select,
-        ):
-            mock_ws = MagicMock()
-            mock_ws.subprotocol = V5_CHANNEL_PROTOCOL
-            mock_ws.connected = True
-            mock_ws.is_ssl.return_value = True
-            mock_ws.sock.pending.return_value = 1
-
-            frame = MagicMock()
-            frame.data = bytes([STDOUT_CHANNEL]) + b"pending"
-            mock_ws.recv_data_frame.return_value = (
-                websocket.ABNF.OPCODE_BINARY,
-                frame,
-            )
-            mock_create.return_value = mock_ws
-
-            client = WSClient(
-                self.config_mock,
-                "wss://test",
-                headers=None,
-                capture_all=True,
-                binary=True,
-            )
-            client.update(timeout=None)
-
-            mock_poll.assert_not_called()
-            mock_select.assert_not_called()
-            mock_ws.recv_data_frame.assert_called_once_with(True)
-            self.assertEqual(
-                client.read_channel(STDOUT_CHANNEL),
-                b"pending",
-            )
-
     def test_readline_channel_returns_empty_string_on_expired_timeout(self):
         """Verify readline_channel returns '' (not None) when a finite timeout expires"""
         with patch.object(ws_client_module, 'create_websocket') as mock_create: