Skip to content

Commit 00b0bc7

Browse files
committed
🧪 modify test_elasticsearch_service.py
1 parent 55cc9c0 commit 00b0bc7

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

‎test/backend/services/test_elasticsearch_service.py‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -297,8 +297,8 @@ def test_list_indices_without_stats(self, mock_get_knowledge):
297297
# Setup
298298
self.mock_es_core.get_user_indices.return_value = ["index1", "index2"]
299299
mock_get_knowledge.return_value = [
300-
{"index_name": "index1", "model_name": "test-model"},
301-
{"index_name": "index2", "model_name": "test-model"}
300+
{"index_name": "index1", "embedding_model_name": "test-model"},
301+
{"index_name": "index2", "embedding_model_name": "test-model"}
302302
]
303303

304304
# Execute
@@ -333,8 +333,8 @@ def test_list_indices_with_stats(self, mock_get_knowledge):
333333
"index2": {"base_info": {"doc_count": 20, "embedding_model": "test-model"}}
334334
}
335335
mock_get_knowledge.return_value = [
336-
{"index_name": "index1", "model_name": "test-model"},
337-
{"index_name": "index2", "model_name": "test-model"}
336+
{"index_name": "index1", "embedding_model_name": "test-model"},
337+
{"index_name": "index2", "embedding_model_name": "test-model"}
338338
]
339339

340340
# Execute
@@ -1322,8 +1322,8 @@ def test_list_indices_success_status_200(self, mock_response, mock_get_knowledge
13221322
self.mock_es_core.get_user_indices.return_value = ["index1", "index2"]
13231323
mock_response.status_code = 200
13241324
mock_get_knowledge.return_value = [
1325-
{"index_name": "index1", "model_name": "test-model"},
1326-
{"index_name": "index2", "model_name": "test-model"}
1325+
{"index_name": "index1", "embedding_model_name": "test-model"},
1326+
{"index_name": "index2", "embedding_model_name": "test-model"}
13271327
]
13281328

13291329
# Execute

0 commit comments

Comments
 (0)