def LanguageServerConnection_ServerConnectionDies_test():
connection = MockConnection()
return_values = [IOError]
with patch.object(connection, 'ReadData', side_effect=return_values):
# No exception is thrown
connection.run()
def LanguageServerConnection_ServerConnectionDies_test():
connection = MockConnection()
return_values = [
IOError
]
with patch.object( connection, 'ReadData', side_effect = return_values ):
# No exception is thrown
connection.run()
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)
def LanguageServerConnection_ReadPartialMessage_test():
connection = MockConnection()
return_values = [
bytes(b'Content-Length: 10\n\n{"abc":'),
bytes(b'""}'), lsc.LanguageServerConnectionStopped
]
with patch.object(connection, 'ReadData', side_effect=return_values):
with patch.object(connection, '_DispatchMessage') as dispatch_message:
connection.run()
dispatch_message.assert_called_with({'abc': ''})
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_ReadPartialMessage_test():
connection = MockConnection()
return_values = [
bytes( b'Content-Length: 10\n\n{"abc":' ),
bytes( b'""}' ),
lsc.LanguageServerConnectionStopped
]
with patch.object( connection, 'ReadData', side_effect = return_values ):
with patch.object( connection, '_DispatchMessage' ) as dispatch_message:
connection.run()
dispatch_message.assert_called_with( { 'abc': '' } )
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 )
def LanguageServerConnection_RejectUnsupportedRequest_test():
connection = MockConnection()
return_values = [
bytes(b'Content-Length: 26\r\n\r\n{"id":"1","method":"test"}'),
lsc.LanguageServerConnectionStopped
]
expected_response = bytes(b'Content-Length: 79\r\n\r\n'
b'{"error":{'
b'"code":-32601,'
b'"message":"Method not found"'
b'},'
b'"id":"1",'
b'"jsonrpc":"2.0"}')
with patch.object(connection, 'ReadData', side_effect=return_values):
with patch.object(connection, 'WriteData') as write_data:
connection.run()
write_data.assert_called_with(expected_response)
def LanguageServerConnection_RejectUnsupportedRequest_test():
connection = MockConnection()
return_values = [
bytes( b'Content-Length: 26\r\n\r\n{"id":"1","method":"test"}' ),
lsc.LanguageServerConnectionStopped
]
expected_response = bytes( b'Content-Length: 104\r\n\r\n'
b'{"error": {'
b'"code": -32601, '
b'"reason": "Method not found"'
b'}, '
b'"id": "1", '
b'"jsonrpc": "2.0", '
b'"method": "test"}' )
with patch.object( connection, 'ReadData', side_effect = return_values ):
with patch.object( connection, 'WriteData' ) as write_data:
connection.run()
write_data.assert_called_with( expected_response )