@@ -44,7 +44,7 @@ def test_filter_auth_invalid(self, invalid_auth, expected_code, expected_fragmen
4444 @pytest .mark .p2
4545 @pytest .mark .parametrize ("invalid_auth, expected_code, expected_fragment" , INVALID_AUTH_CASES )
4646 def test_infos_auth_invalid (self , invalid_auth , expected_code , expected_fragment ):
47- res = document_infos (invalid_auth , {"doc_ids" : ["doc_id" ]})
47+ res = document_infos (invalid_auth , "kb_id" , {"doc_ids" : ["doc_id" ]})
4848 assert res ["code" ] == expected_code , res
4949 assert expected_fragment in res ["message" ], res
5050
@@ -91,11 +91,12 @@ def test_filter(self, WebApiAuth, add_dataset_func):
9191
9292 @pytest .mark .p2
9393 def test_infos (self , WebApiAuth , add_document_func ):
94- _ , doc_id = add_document_func
95- res = document_infos (WebApiAuth , { "doc_ids " : [doc_id ]})
94+ dataset_id , doc_id = add_document_func
95+ res = document_infos (WebApiAuth , dataset_id , { "ids " : [doc_id ]})
9696 assert res ["code" ] == 0 , res
97- assert len (res ["data" ]) == 1 , res
98- assert res ["data" ][0 ]["id" ] == doc_id , res
97+ docs = res ["data" ]["docs" ]
98+ assert len (docs ) == 1 , docs
99+ assert docs [0 ]["id" ] == doc_id , res
99100
100101 ## The inputs has been changed to add 'doc_ids'
101102 ## TODO:
@@ -138,20 +139,22 @@ def test_infos(self, WebApiAuth, add_document_func):
138139
139140 @pytest .mark .p2
140141 def test_change_status (self , WebApiAuth , add_document_func ):
141- _ , doc_id = add_document_func
142+ dataset_id , doc_id = add_document_func
142143 res = document_change_status (WebApiAuth , {"doc_ids" : [doc_id ], "status" : "1" })
144+
143145 assert res ["code" ] == 0 , res
144146 assert res ["data" ][doc_id ]["status" ] == "1" , res
145- info_res = document_infos (WebApiAuth , {"doc_ids" : [doc_id ]})
147+ info_res = document_infos (WebApiAuth , dataset_id , {"ids" : [doc_id ]})
148+
146149 assert info_res ["code" ] == 0 , info_res
147- assert info_res ["data" ][0 ]["status" ] == "1" , info_res
150+ assert info_res ["data" ]["docs" ][ 0 ]["status" ] == "1" , info_res
148151
149152
150153class TestDocumentMetadataNegative :
151154 @pytest .mark .p2
152155 def test_filter_missing_kb_id (self , WebApiAuth , add_document_func ):
153156 kb_id , doc_id = add_document_func
154- res = document_filter (WebApiAuth , "" , {"doc_ids " : [doc_id ]})
157+ res = document_filter (WebApiAuth , "" , {"ids " : [doc_id ]})
155158 assert res ["code" ] == 100 , res
156159 assert "<MethodNotAllowed '405: Method Not Allowed'>" == res ["message" ], res
157160
@@ -228,26 +231,6 @@ def _allow_kb(self, module, monkeypatch, kb_id="kb1", tenant_id="tenant1"):
228231 monkeypatch .setattr (module .UserTenantService , "query" , lambda ** _kwargs : [SimpleNamespace (tenant_id = tenant_id )])
229232 monkeypatch .setattr (module .KnowledgebaseService , "query" , lambda ** _kwargs : True if _kwargs .get ("id" ) == kb_id else False )
230233
231-
232- def test_infos_meta_fields (self , document_app_module , monkeypatch ):
233- module = document_app_module
234- monkeypatch .setattr (module .DocumentService , "accessible" , lambda * _args , ** _kwargs : True )
235-
236- class _Docs :
237- def dicts (self ):
238- return [{"id" : "doc1" }]
239-
240- monkeypatch .setattr (module .DocumentService , "get_by_ids" , lambda _ids : _Docs ())
241- monkeypatch .setattr (module .DocMetadataService , "get_document_metadata" , lambda _doc_id : {"author" : "alice" })
242-
243- async def fake_request_json ():
244- return {"doc_ids" : ["doc1" ]}
245-
246- monkeypatch .setattr (module , "get_request_json" , fake_request_json )
247- res = _run (module .doc_infos ())
248- assert res ["code" ] == 0
249- assert res ["data" ][0 ]["meta_fields" ]["author" ] == "alice"
250-
251234 def test_metadata_update_missing_kb_id (self , document_app_module , monkeypatch ):
252235 module = document_app_module
253236
0 commit comments