def test_core_servicer_get_search_soltuion_results(solutions_results_mock):
instance = CoreServicer('/input', '/output', '/static', 0.5)
instance._get_progress = MagicMock()
solutions = {
1: {
'id': 1,
'score': 1,
'rank': 1
},
2: {
'id': 2,
'score': 2,
'rank': 2
}
}
# case 1: len(solutions) < returned
searcher = MagicMock(solutions=solutions)
session = {'searcher': searcher}
result = instance._get_search_soltuion_results(session, 10)
instance._get_progress.assert_not_called()
solutions_results_mock.assert_not_called()
assert result is None
# case 2: len(solutions) > returned
result = instance._get_search_soltuion_results(session, 1)
instance._get_progress.assert_called_once()
solutions_results_mock.assert_called_once()
def test_core_servicer_start_session(thread_mock, logger_mock):
session_id = 'test-id'
session_type = 'test-type'
method = MagicMock()
# debug mode
instance = CoreServicer('/input', '/output', '/static', 0.5, debug=True)
instance._start_session(session_id,
session_type,
method,
'first-argument',
second='argument')
args, kwargs = method.call_args
assert len(instance.DB) == 1
assert len(instance.DB[session_type + '_sessions']) == 1
session = instance.DB[session_type + '_sessions'][session_id]
assert method.call_count == 1
assert logger_mock.call_count == 2
assert args == ('first-argument', )
assert kwargs == {} # TODO: should kwargs be used?
assert 'id' in session
assert 'type' in session
assert 'start' in session
assert 'end' in session
assert 'done' in session
assert 'error' not in session
# without debugging
logger_mock.reset_mock()
instance = CoreServicer('/input', '/output', '/static', 0.5)
instance._start_session(session_id,
session_type,
method,
'first-argument',
second='argument')
assert len(instance.DB) == 1
assert len(instance.DB[session_type + '_sessions']) == 1
session = instance.DB[session_type + '_sessions'][session_id]
expected_args = [session, method] + list(args)
thread_mock.assert_called_once_with(target=instance._run_session,
args=expected_args)
assert logger_mock.call_count == 1
assert 'id' in session
assert 'type' in session
assert 'start' in session
def test_core_servicer():
input_dir = '/input-dir'
output_dir = '/output-dir'
timeout = 0.5
instance = CoreServicer(input_dir, output_dir, timeout)
assert isinstance(instance, core_pb2_grpc.CoreServicer)
assert instance.input_dir == input_dir
assert instance.output_dir == output_dir
assert instance.timeout == timeout
assert not instance.debug
def test_core_servicer_stopsearchsolutions(stop_search_mock):
instance = CoreServicer('/input-dir', '/output-dir', 0.5)
expected_result = 'result'
stop_search_mock.return_value = expected_result
search_id = 'test-id'
request = MagicMock(search_id=search_id)
# empty session
instance.DB['search_sessions'] = {search_id: {}}
result = instance.StopSearchSolutions(request, None)
stop_search_mock.assert_called_once()
assert result == expected_result
# session with searcher
searcher = MagicMock(done=True)
searcher.stop = MagicMock()
instance.DB['search_sessions'] = {search_id: {'searcher': searcher}}
instance.StopSearchSolutions(request, None)
searcher.stop.assert_called_once()
assert result == expected_result
assert stop_search_mock.call_count == 2
def test_core_servicer_searchsolutions(searcher_mock, pipeline_searcher_mock, decode_mock):
instance = CoreServicer('/input-dir', '/output-dir', 0.5)
instance._start_session = MagicMock()
expected_result = 'result'
searcher_mock.return_value = expected_result
# wrong version
request = MagicMock(version='fake-version')
with pytest.raises(AssertionError):
instance.SearchSolutions(request, None) # context (None) is not used
# wrong problem inputs
request = MagicMock(version='2019.6.11')
with pytest.raises(AssertionError):
instance.SearchSolutions(request, None) # context (None) is not used
# correct parameters
inputs = [1]
problem = MagicMock(inputs=inputs)
request = MagicMock(version='2019.6.11', inputs=inputs, problem=problem)
result = instance.SearchSolutions(request, None) # context (None) is not used
decode_mock.assert_called_once_with(problem)
pipeline_searcher_mock.assert_called_once_with(instance.input_dir, instance.output_dir)
assert instance._start_session.call_count == 1
assert result == expected_result
def test_core_servicer_run_session(logger_info_mock, logger_exception_mock):
session = {'type': 'test-type', 'id': 'test-id'}
# no arguments
method = MagicMock()
instance = CoreServicer('/input-dir', '/output-dir', 0.5)
instance._run_session(session, method)
args, kwargs = method.call_args
assert method.call_count == 1
assert logger_info_mock.call_count == 1
assert logger_exception_mock.call_count == 0
assert args == ()
assert kwargs == {}
assert 'end' in session
assert 'done' in session
assert 'error' not in session
# with arguments
instance._run_session(session, method, 'first-argument', second='argument')
args, kwargs = method.call_args
assert method.call_count == 2
assert logger_info_mock.call_count == 2
assert logger_exception_mock.call_count == 0
assert args == ('first-argument',)
assert kwargs == {'second': 'argument'}
assert 'end' in session
assert 'done' in session
assert 'error' not in session
# with exception
method = MagicMock(side_effect=IndexError)
instance._run_session(session, method)
assert method.call_count == 1
assert logger_info_mock.call_count == 3
assert logger_exception_mock.call_count == 1
assert 'end' in session
assert 'done' in session
assert 'error' in session
def test_core_servicer_get_progress(progress_state_mock, progress_mock):
instance = CoreServicer('/input-dir', '/output-dir', 0.5)
# ERRORED
session = {'error': 'test-value'}
instance._get_progress(session)
progress_state_mock.assert_called_with('ERRORED')
assert progress_mock.call_count == 1
# COMPLETED
session = {'done': 'test-value'}
instance._get_progress(session)
progress_state_mock.assert_called_with('COMPLETED')
assert progress_mock.call_count == 2
# RUNNING
session = {'other': 'test-value'}
instance._get_progress(session)
progress_state_mock.assert_called_with('RUNNING')
assert progress_mock.call_count == 3
def test_core_servicer_getsearchsolutionsresults():
instance = CoreServicer('/input-dir', '/output-dir', 0.5)
expected_result = 'result'
instance._stream = MagicMock(return_value=expected_result)
# invalid search_id
request = MagicMock()
with pytest.raises(ValueError):
instance.GetSearchSolutionsResults(request, None)
instance._stream.assert_not_called()
# valid search_id
search_id = 'test-id'
instance.DB['search_sessions'] = {search_id: 'test-value'}
request = MagicMock(search_id=search_id)
result = instance.GetSearchSolutionsResults(request, None)
instance._stream.assert_called_with('test-value', instance._get_search_soltuion_results)
assert result == expected_result