def test_context_creation_es6(): from Elasticsearch_v2 import results_to_context, get_total_results base_page = 0 size = 2 total_dict, total_results = get_total_results(ES_V6_RESPONSE) query = 'incident' index = 'users' search_context, meta_headers, hit_tables, hit_headers = results_to_context( index, query, base_page, size, total_dict, ES_V6_RESPONSE) assert str(search_context) == MOCK_ES6_SEARCH_CONTEXT assert str( meta_headers ) == "['Query', 'took', 'timed_out', 'total', 'max_score', 'Server', 'Page', 'Size']" assert str(hit_tables) == MOCK_ES6_HIT_CONTEXT assert str(hit_headers) == "['_id', '_index', '_type', '_score', 'Date']"
def test_context_creation_es6(params, mocker): mocker.patch.object(demisto, 'params', return_value=params) importlib.reload( Elasticsearch_v2 ) # To reset the Elasticsearch client with the OpenSearch library from Elasticsearch_v2 import results_to_context, get_total_results base_page = 0 size = 2 total_dict, total_results = get_total_results(ES_V6_RESPONSE) query = 'incident' index = 'users' search_context, meta_headers, hit_tables, hit_headers = results_to_context( index, query, base_page, size, total_dict, ES_V6_RESPONSE) assert str(search_context) == MOCK_ES6_SEARCH_CONTEXT assert str( meta_headers ) == "['Query', 'took', 'timed_out', 'total', 'max_score', 'Server', 'Page', 'Size']" assert str(hit_tables) == MOCK_ES6_HIT_CONTEXT assert str(hit_headers) == "['_id', '_index', '_type', '_score', 'Date']"