test: update record

Author: taskingaijc

test/files/test.pdf

@@ -250,31 +250,31 @@ async def test_a_update_record_by_web(self, text_splitter):
 res_dict = vars(res)
 assume_record_result(update_record_data, res_dict)
 
-@pytest.mark.run(order=34)
-@pytest.mark.asyncio
-@pytest.mark.parametrize("upload_file_data", upload_file_data_list[2:3])
-async def test_a_update_record_by_file(self, upload_file_data):
-# upload file
-upload_file_res = await a_upload_file(**upload_file_data)
-upload_file_dict = vars(upload_file_res)
-file_id = upload_file_dict["file_id"]
-pytest.assume(file_id is not None)
-
-# Update a record.
-
-update_record_data = {
-"type": "file",
-"title": "Machine learning",
-"collection_id": self.collection_id,
-"record_id": self.record_id,
-"file_id": file_id,
-"text_splitter": TokenTextSplitter(chunk_size=200, chunk_overlap=100),
-"metadata": {"test": "test"},
-}
-res = await a_update_record(**update_record_data)
-logger.info(f"a_update_record:{res}")
-res_dict = vars(res)
-assume_record_result(update_record_data, res_dict)
+# @pytest.mark.run(order=34)
+# @pytest.mark.asyncio
+# @pytest.mark.parametrize("upload_file_data", upload_file_data_list[2:3])
+# async def test_a_update_record_by_file(self, upload_file_data):
+# # upload file
+# upload_file_res = await a_upload_file(**upload_file_data)
+# upload_file_dict = vars(upload_file_res)
+# file_id = upload_file_dict["file_id"]
+# pytest.assume(file_id is not None)
+#
+# # Update a record.
+#
+# update_record_data = {
+# "type": "file",
+# "title": "Machine learning",
+# "collection_id": self.collection_id,
+# "record_id": self.record_id,
+# "file_id": file_id,
+# "text_splitter": TokenTextSplitter(chunk_size=200, chunk_overlap=100),
+# "metadata": {"test": "test"},
+# }
+# res = await a_update_record(**update_record_data)
+# logger.info(f"a_update_record:{res}")
+# res_dict = vars(res)
+# assume_record_result(update_record_data, res_dict)
 
 @pytest.mark.run(order=79)
 @pytest.mark.asyncio

test/testcase/test_async/test_async_retrieval.py

@@ -267,30 +267,30 @@ def test_update_record_by_web(self, collection_id, record_id, text_splitter):
 res_dict = vars(res)
 assume_record_result(update_record_data, res_dict)
 
-@pytest.mark.run(order=35)
-@pytest.mark.parametrize("upload_file_data", upload_file_data_list[2:3])
-def test_update_record_by_file(self, collection_id, record_id, upload_file_data):
-# upload file
-upload_file_res = upload_file(**upload_file_data)
-upload_file_dict = vars(upload_file_res)
-file_id = upload_file_dict["file_id"]
-pytest.assume(file_id is not None)
-
-# Update a record.
-text_splitter = TokenTextSplitter(chunk_size=200, chunk_overlap=20)
-
-update_record_data = {
-"type": "file",
-"title": "TaskingAI",
-"collection_id": collection_id,
-"record_id": record_id,
-"file_id": file_id,
-"text_splitter": text_splitter,
-"metadata": {"test": "test"},
-}
-res = update_record(**update_record_data)
-res_dict = vars(res)
-assume_record_result(update_record_data, res_dict)
+# @pytest.mark.run(order=35)
+# @pytest.mark.parametrize("upload_file_data", upload_file_data_list[2:3])
+# def test_update_record_by_file(self, collection_id, record_id, upload_file_data):
+# # upload file
+# upload_file_res = upload_file(**upload_file_data)
+# upload_file_dict = vars(upload_file_res)
+# file_id = upload_file_dict["file_id"]
+# pytest.assume(file_id is not None)
+#
+# # Update a record.
+# text_splitter = TokenTextSplitter(chunk_size=200, chunk_overlap=20)
+#
+# update_record_data = {
+# "type": "file",
+# "title": "TaskingAI",
+# "collection_id": collection_id,
+# "record_id": record_id,
+# "file_id": file_id,
+# "text_splitter": text_splitter,
+# "metadata": {"test": "test"},
+# }
+# res = update_record(**update_record_data)
+# res_dict = vars(res)
+# assume_record_result(update_record_data, res_dict)
 
 @pytest.mark.run(order=79)
 def test_delete_record(self, collection_id):