Skip to content

Commit bb18fb2

Browse files
[DAGE-20] updated test names
1 parent 2b58168 commit bb18fb2

3 files changed

Lines changed: 11 additions & 14 deletions

File tree

rre-dataset-generator/src/search_engine/elasticsearch_search_engine.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -51,10 +51,7 @@ def fetch_for_query_generation(self,
5151
for field, values in dict_field.items():
5252
if not values:
5353
continue
54-
if len(values) == 1:
55-
filter_clauses.append({"term": {field: values[0]}})
56-
else:
57-
filter_clauses.append({"terms": {field: values}})
54+
filter_clauses.append({"terms": {field: values}})
5855

5956
# Wrap in a bool query if there are any filters
6057
if filter_clauses:

rre-dataset-generator/tests/unit/test_elasticsearch_search_engine.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def mock_dict(mock_doc):
3838
'fields': mock_doc["_source"]
3939
}
4040

41-
def test_elasticsearch_search_engine_fetch_for_query_generation(monkeypatch, elasticsearch_config, mock_doc, mock_dict):
41+
def test_elasticsearch_search_engine_fetch_for_query_generation__expect__result_returned(monkeypatch, elasticsearch_config, mock_doc, mock_dict):
4242
url = "https://fakeurl"
4343
search_engine = ElasticsearchSearchEngine(url)
4444

@@ -54,7 +54,7 @@ def test_elasticsearch_search_engine_fetch_for_query_generation(monkeypatch, ela
5454
doc_fields=elasticsearch_config.doc_fields)
5555
assert result[0] == Document(**mock_dict)
5656

57-
def test_elasticsearch_search_engine_fetch_for_evaluation(monkeypatch, elasticsearch_config, mock_doc, mock_dict):
57+
def test_elasticsearch_search_engine_fetch_for_evaluation__expect__result_returned(monkeypatch, elasticsearch_config, mock_doc, mock_dict):
5858
url = "https://fakeurl"
5959
search_engine = ElasticsearchSearchEngine(url)
6060

@@ -70,7 +70,7 @@ def test_elasticsearch_search_engine_fetch_for_evaluation(monkeypatch, elasticse
7070
doc_fields=elasticsearch_config.doc_fields)
7171
assert result[0] == Document(**mock_dict)
7272

73-
def test_elasticsearch_search_engine_negative_post_fetch_for_query_generation_EXPECTED_HttpError(monkeypatch, elasticsearch_config):
73+
def test_elasticsearch_search_engine_negative_post_fetch_for_query_generation__expect__HttpError(monkeypatch, elasticsearch_config):
7474
for status_code in [400, 401, 402, 403, 500]:
7575
monkeypatch.setattr(requests, "post", lambda *args, **kwargs: MockResponseElasticsearchEngine([],
7676
status_code=status_code))
@@ -85,7 +85,7 @@ def test_elasticsearch_search_engine_negative_post_fetch_for_query_generation_EX
8585
)
8686

8787

88-
def test_elasticsearch_search_engine_negative_post_fetch_for_evaluation_EXPECTED_HttpError(monkeypatch, elasticsearch_config):
88+
def test_elasticsearch_search_engine_negative_post_fetch_for_evaluation__expect__HttpError(monkeypatch, elasticsearch_config):
8989
for status_code in [400, 401, 402, 403, 500]:
9090
monkeypatch.setattr(requests, "post", lambda *args, **kwargs: MockResponseElasticsearchEngine([],
9191
status_code=status_code))
@@ -99,6 +99,6 @@ def test_elasticsearch_search_engine_negative_post_fetch_for_evaluation_EXPECTED
9999
doc_fields=elasticsearch_config.doc_fields
100100
)
101101

102-
def test_elasticsearch_search_engine_bad_url_EXPECTED_ValidationError():
102+
def test_elasticsearch_search_engine_bad_url__expect__ValidationError():
103103
with pytest.raises(ValidationError):
104104
_ = ElasticsearchSearchEngine("fake-NONurl")

rre-dataset-generator/tests/unit/test_solr_search_engine.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def mock_dict(mock_doc):
3434
'fields': {k: v for k, v in mock_doc.items() if k != 'mock_id'}
3535
}
3636

37-
def test_solr_search_engine_fetch_for_query_generation(monkeypatch, solr_config, mock_doc, mock_dict):
37+
def test_solr_search_engine_fetch_for_query_generation__expect__result_returned(monkeypatch, solr_config, mock_doc, mock_dict):
3838
monkeypatch.setattr(requests, "get", lambda *args, **kwargs: MockResponseUniqueKey(ident="mock_id"))
3939
search_engine = SolrSearchEngine("https://fakeurl")
4040

@@ -49,7 +49,7 @@ def test_solr_search_engine_fetch_for_query_generation(monkeypatch, solr_config,
4949
doc_fields=solr_config.doc_fields)
5050
assert result[0] == Document(**mock_dict)
5151

52-
def test_solr_search_engine_fetch_for_evaluation(monkeypatch, solr_config, mock_doc, mock_dict):
52+
def test_solr_search_engine_fetch_for_evaluation__expect__result_returned(monkeypatch, solr_config, mock_doc, mock_dict):
5353
monkeypatch.setattr(requests, "get", lambda *args, **kwargs: MockResponseUniqueKey(ident="mock_id"))
5454
search_engine = SolrSearchEngine("https://fakeurl")
5555

@@ -64,7 +64,7 @@ def test_solr_search_engine_fetch_for_evaluation(monkeypatch, solr_config, mock_
6464
doc_fields=solr_config.doc_fields)
6565
assert result[0] == Document(**mock_dict)
6666

67-
def test_solr_search_engine_negative_post_fetch_for_query_generation_EXPECTED_HttpError(monkeypatch, solr_config):
67+
def test_solr_search_engine_negative_post_fetch_for_query_generation__expect__HttpError(monkeypatch, solr_config):
6868
for status_code in [400, 401, 402, 403, 500]:
6969
monkeypatch.setattr(requests, "get", lambda *args, **kwargs: MockResponseUniqueKey(ident="identifier"))
7070
monkeypatch.setattr(requests, "post", lambda *args, **kwargs: MockResponseSolrEngine([], status_code=status_code))
@@ -79,7 +79,7 @@ def test_solr_search_engine_negative_post_fetch_for_query_generation_EXPECTED_Ht
7979
)
8080

8181

82-
def test_solr_search_engine_negative_post_fetch_for_evaluation_EXPECTED_HttpError(monkeypatch, solr_config):
82+
def test_solr_search_engine_negative_post_fetch_for_evaluation__expect__HttpError(monkeypatch, solr_config):
8383
for status_code in [400, 401, 402, 403, 500]:
8484
monkeypatch.setattr(requests, "get", lambda *args, **kwargs: MockResponseUniqueKey(ident="identifier"))
8585
monkeypatch.setattr(requests, "post", lambda *args, **kwargs: MockResponseSolrEngine([], status_code=status_code))
@@ -137,6 +137,6 @@ def test_template_to_json_payload(monkeypatch):
137137
}
138138
assert solr_engine._template_to_json_payload(template) == expected_payload
139139

140-
def test_solr_search_engine_bad_url_EXPECTED_ValidationError():
140+
def test_solr_search_engine_bad_url__expect__ValidationError():
141141
with pytest.raises(ValidationError):
142142
_ = SolrSearchEngine("fake-NONurl")

0 commit comments

Comments
 (0)