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)