1919from ralph .conf import settings
2020from ralph .exceptions import BackendException , BackendParameterException
2121
22- from tests .fixtures .backends import ES_TEST_HOSTS , ES_TEST_INDEX , ES_TEST_INDEX_2
22+ from tests .fixtures .backends import (
23+ ES_TEST_FORWARDING_INDEX ,
24+ ES_TEST_HOSTS ,
25+ ES_TEST_INDEX ,
26+ )
2327
2428
2529def test_backends_database_es_database_instantiation (es ):
@@ -465,7 +469,9 @@ def mock_search(**_):
465469 assert caplog .record_tuples == [(logger_name , logging .ERROR , msg )]
466470
467471
468- def test_backends_database_es_query_statements_by_ids_with_multiple_indexes (es , es2 ):
472+ def test_backends_database_es_query_statements_by_ids_with_multiple_indexes (
473+ es , es_forwarding
474+ ):
469475 """Tests the ES query_statements_by_ids method, given a valid search
470476 query, should execute the query uniquely on the specified index and return the
471477 expected results.
@@ -474,17 +480,21 @@ def test_backends_database_es_query_statements_by_ids_with_multiple_indexes(es,
474480
475481 # Insert documents
476482 index_1_document = {"_index" : ES_TEST_INDEX , "_id" : "1" , "_source" : {"id" : "1" }}
477- index_2_document = {"_index" : ES_TEST_INDEX_2 , "_id" : "2" , "_source" : {"id" : "2" }}
483+ index_2_document = {
484+ "_index" : ES_TEST_FORWARDING_INDEX ,
485+ "_id" : "2" ,
486+ "_source" : {"id" : "2" },
487+ }
478488 bulk (es , [index_1_document ])
479- bulk (es2 , [index_2_document ])
489+ bulk (es_forwarding , [index_2_document ])
480490
481491 # As we bulk insert documents, the index needs to be refreshed before making queries
482492 es .indices .refresh (index = ES_TEST_INDEX )
483- es2 .indices .refresh (index = ES_TEST_INDEX_2 )
493+ es_forwarding .indices .refresh (index = ES_TEST_FORWARDING_INDEX )
484494
485495 # Instantiate ES Databases
486496 database = ESDatabase (hosts = ES_TEST_HOSTS , index = ES_TEST_INDEX )
487- database_2 = ESDatabase (hosts = ES_TEST_HOSTS , index = ES_TEST_INDEX_2 )
497+ database_2 = ESDatabase (hosts = ES_TEST_HOSTS , index = ES_TEST_FORWARDING_INDEX )
488498
489499 # Check the expected search query results
490500 index_1_document = index_1_document | {"_score" : 1.0 }
0 commit comments