def test_stream_handle_read_event_processes_recv_buffer_before_closing(self):
# to ensure we don't reintroduce issue #41
stream = Stream()
stream._socket_recv = MagicMock(return_value=None)
manager = MagicMock()
stream._process_recv_buffer = manager._process_recv_buffer
stream.close = manager.close
stream._handle_read_event()
expected_calls = [call._process_recv_buffer(), call.close(flush=False)]
self.assertTrue(manager.mock_calls == expected_calls)
def test_stream_handle_read_event_processes_recv_buffer_before_closing(
self):
# to ensure we don't reintroduce issue #41
stream = Stream()
stream._socket_recv = MagicMock(return_value=None)
manager = MagicMock()
stream._process_recv_buffer = manager._process_recv_buffer
stream.close = manager.close
stream._handle_read_event()
expected_calls = [call._process_recv_buffer(), call.close(flush=False)]
self.assertTrue(manager.mock_calls == expected_calls)
