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)