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_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