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):