def LanguageServerConnection_RequestAbortAwait_test(): connection = MockConnection() return_values = [lsc.LanguageServerConnectionStopped] with patch.object(connection, 'ReadData', side_effect=return_values): response = connection.GetResponseAsync(1, bytes(b'{"test":"test"}')) connection.run() assert_that( calling(response.AwaitResponse).with_args(10), raises(lsc.ResponseAbortedException))
def LanguageServerConnection_RequestAbortCallback_test(): connection = MockConnection() return_values = [lsc.LanguageServerConnectionStopped] with patch.object(connection, 'ReadData', side_effect=return_values): callback = MagicMock() response = connection.GetResponseAsync(1, bytes(b'{"test":"test"}'), response_callback=callback) connection.run() callback.assert_called_with(response, None)