def test_query_and_push_task(mocked_post_le_search): with patch.object(Search, 'handle_response') as mocked_handle_response: search = Search(DummyService, conf_ex.DUMMY_SEARCH_CONFIG, conf_ex.DUMMY_CONFIG) search.query_and_push_task() assert mocked_post_le_search.called assert mocked_handle_response.called
def test_handle_cont_response(mocked_cont_final_resp): with patch.object(Search, 'process_final_response') as mocked_process_final_resp: search = Search(DummyService, conf_ex.DUMMY_SEARCH_CONFIG, conf_ex.DUMMY_CONFIG) httpretty.register_uri(httpretty.GET, req_ex.DEST_URL, body={}, status=202) continuity_response = requests.get(req_ex.DEST_URL) search.handle_response(continuity_response) assert mocked_process_final_resp.called
def test_process_final_response(): with patch.object(DummyService, 'process') as mocked_process_method: search = Search(DummyService, conf_ex.DUMMY_SEARCH_CONFIG, conf_ex.DUMMY_CONFIG) httpretty.register_uri(httpretty.GET, req_ex.DEST_URL, body=json.dumps({}), status=200, content_type='application/json') final_response = requests.get(req_ex.DEST_URL) search.process_final_response(final_response) assert mocked_process_method.called
def start_search_tasks(): """ Before everything, kill if there is any running search tasks. Then start the search tasks concurrently. """ global SEARCH_TASKS logging.info( "(Re)populated config collections from config file. " "Cancelling previous loops and restarting them again with the new config." ) for looping_task in SEARCH_TASKS: logging.info("Cancelling this loop: %r", looping_task) looping_task.stop() SEARCH_TASKS = [] searches = CONFIG['Searches'].values() search_count = len(searches) logging.info("Search count: %d", search_count) reactor.suggestThreadPoolSize(search_count) try: for search in searches: search_obj = Search( SERVICE_CLASS_MAPPER.get(search['destination']['service']), search, CONFIG) do_search_concurrently(search_obj) except Exception as exception: logging.exception("Exception occurred while processing search. %s", exception.message)
def test_stop(): with patch.object(task.LoopingCall, 'stop') as mocked_stop: search = Search(DummyService, conf_ex.DUMMY_SEARCH_CONFIG, conf_ex.DUMMY_CONFIG) search.start() search.stop() assert mocked_stop.called