158 lines
5.4 KiB
Python
158 lines
5.4 KiB
Python
# coding: utf-8
|
|
#
|
|
# Copyright (C) 2016 YouCompleteMe contributors
|
|
#
|
|
# This file is part of YouCompleteMe.
|
|
#
|
|
# YouCompleteMe is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# YouCompleteMe is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with YouCompleteMe. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
from __future__ import unicode_literals
|
|
from __future__ import print_function
|
|
from __future__ import division
|
|
from __future__ import absolute_import
|
|
# Not installing aliases from python-future; it's unreliable and slow.
|
|
from builtins import * # noqa
|
|
|
|
from ycm.tests.test_utils import ( CurrentWorkingDirectory, ExtendedMock,
|
|
MockVimModule, MockVimBuffers, VimBuffer )
|
|
MockVimModule()
|
|
|
|
import contextlib
|
|
from hamcrest import assert_that, contains, empty, has_entries
|
|
from mock import call, MagicMock, patch
|
|
from nose.tools import ok_
|
|
|
|
from ycm.tests import PathToTestFile, YouCompleteMeInstance
|
|
from ycmd.responses import ServerError
|
|
|
|
|
|
@contextlib.contextmanager
|
|
def MockCompletionRequest( response_method ):
|
|
"""Mock out the CompletionRequest, replacing the response handler
|
|
JsonFromFuture with the |response_method| parameter."""
|
|
|
|
# We don't want the requests to actually be sent to the server, just have it
|
|
# return success.
|
|
with patch( 'ycm.client.completer_available_request.'
|
|
'CompleterAvailableRequest.PostDataToHandler',
|
|
return_value = True ):
|
|
with patch( 'ycm.client.completion_request.CompletionRequest.'
|
|
'PostDataToHandlerAsync',
|
|
return_value = MagicMock( return_value=True ) ):
|
|
|
|
# We set up a fake response.
|
|
with patch( 'ycm.client.base_request._JsonFromFuture',
|
|
side_effect = response_method ):
|
|
yield
|
|
|
|
|
|
@YouCompleteMeInstance()
|
|
def SendCompletionRequest_UnicodeWorkingDirectory_test( ycm ):
|
|
unicode_dir = PathToTestFile( 'uni¢𐍈d€' )
|
|
current_buffer = VimBuffer( PathToTestFile( 'uni¢𐍈d€', 'current_buffer' ) )
|
|
|
|
def ServerResponse( *args ):
|
|
return { 'completions': [], 'completion_start_column': 1 }
|
|
|
|
with CurrentWorkingDirectory( unicode_dir ):
|
|
with MockVimBuffers( [ current_buffer ], [ current_buffer ] ):
|
|
with MockCompletionRequest( ServerResponse ):
|
|
ycm.SendCompletionRequest()
|
|
ok_( ycm.CompletionRequestReady() )
|
|
assert_that(
|
|
ycm.GetCompletionResponse(),
|
|
has_entries( {
|
|
'completions': empty(),
|
|
'completion_start_column': 1
|
|
} )
|
|
)
|
|
|
|
|
|
@YouCompleteMeInstance()
|
|
@patch( 'ycm.vimsupport.PostVimMessage', new_callable = ExtendedMock )
|
|
def SendCompletionRequest_ResponseContainingError_test( ycm, post_vim_message ):
|
|
current_buffer = VimBuffer( 'buffer' )
|
|
|
|
def ServerResponse( *args ):
|
|
return {
|
|
'completions': [ {
|
|
'insertion_text': 'insertion_text',
|
|
'menu_text': 'menu_text',
|
|
'extra_menu_info': 'extra_menu_info',
|
|
'detailed_info': 'detailed_info',
|
|
'kind': 'kind',
|
|
'extra_data': {
|
|
'doc_string': 'doc_string'
|
|
}
|
|
} ],
|
|
'completion_start_column': 3,
|
|
'errors': [ {
|
|
'exception': {
|
|
'TYPE': 'Exception'
|
|
},
|
|
'message': 'message',
|
|
'traceback': 'traceback'
|
|
} ]
|
|
}
|
|
|
|
with MockVimBuffers( [ current_buffer ], [ current_buffer ] ):
|
|
with MockCompletionRequest( ServerResponse ):
|
|
ycm.SendCompletionRequest()
|
|
ok_( ycm.CompletionRequestReady() )
|
|
response = ycm.GetCompletionResponse()
|
|
post_vim_message.assert_has_exact_calls( [
|
|
call( 'Exception: message', truncate = True )
|
|
] )
|
|
assert_that(
|
|
response,
|
|
has_entries( {
|
|
'completions': contains( has_entries( {
|
|
'word': 'insertion_text',
|
|
'abbr': 'menu_text',
|
|
'menu': 'extra_menu_info',
|
|
'info': 'detailed_info\ndoc_string',
|
|
'kind': 'k',
|
|
'dup': 1,
|
|
'empty': 1
|
|
} ) ),
|
|
'completion_start_column': 3
|
|
} )
|
|
)
|
|
|
|
|
|
@YouCompleteMeInstance()
|
|
@patch( 'ycm.client.base_request._logger', autospec = True )
|
|
@patch( 'ycm.vimsupport.PostVimMessage', new_callable = ExtendedMock )
|
|
def SendCompletionRequest_ErrorFromServer_test( ycm,
|
|
post_vim_message,
|
|
logger ):
|
|
current_buffer = VimBuffer( 'buffer' )
|
|
with MockVimBuffers( [ current_buffer ], [ current_buffer ] ):
|
|
with MockCompletionRequest( ServerError( 'Server error' ) ):
|
|
ycm.SendCompletionRequest()
|
|
ok_( ycm.CompletionRequestReady() )
|
|
response = ycm.GetCompletionResponse()
|
|
logger.exception.assert_called_with( 'Error while handling server '
|
|
'response' )
|
|
post_vim_message.assert_has_exact_calls( [
|
|
call( 'Server error', truncate = True )
|
|
] )
|
|
assert_that(
|
|
response,
|
|
has_entries( {
|
|
'completions': empty(),
|
|
'completion_start_column': -1
|
|
} )
|
|
)
|