test: add test

Author: taskingaijc

test/testcase/test_async/test_async_retrieval.py

@@ -212,23 +212,24 @@ async def test_a_get_record(self):
         pytest.assume(res_dict["record_id"] == self.record_id)
         # pytest.assume(res_dict["status"] == "ready")
 
-    # @pytest.mark.run(order=34)
-    # @pytest.mark.asyncio
-    # @pytest.mark.parametrize("text_splitter", text_splitter_list)
-    # async def test_a_update_record_by_text(self, text_splitter):
-    #     # Update a record.
-    #
-    #     update_record_data = {
-    #         "collection_id": self.collection_id,
-    #         "record_id": self.record_id,
-    #         "content": "TaskingAI is an AI-native application development platform that unifies modules like Model, Retrieval, Assistant, and Tool into one seamless ecosystem, streamlining the creation and deployment of applications for developers.",
-    #         "text_splitter": text_splitter,
-    #         "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("text_splitter", text_splitter_list)
+    async def test_a_update_record_by_text(self, text_splitter):
+        # Update a record.
+
+        update_record_data = {
+            "collection_id": self.collection_id,
+            "record_id": self.record_id,
+            "content": "TaskingAI is an AI-native application development platform that unifies modules like Model, Retrieval, Assistant, and Tool into one seamless ecosystem, streamlining the creation and deployment of applications for developers.",
+            "text_splitter": text_splitter,
+            "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)
+        records = list_records(collection_id=self.collection_id, order="desc", limit=20, after=None, before=None)
 
     # @pytest.mark.run(order=34)
     # @pytest.mark.asyncio

test/testcase/test_sync/test_sync_retrieval.py

@@ -231,23 +231,24 @@ def test_get_record(self, collection_id):
             pytest.assume(res_dict["record_id"] == record_id)
             # pytest.assume(res_dict["status"] == "ready")
 
-    # @pytest.mark.run(order=34)
-    # @pytest.mark.parametrize("text_splitter", text_splitter_list)
-    # def test_update_record_by_text(self, collection_id, record_id, text_splitter):
-    #     # Update a record.
-    #
-    #     update_record_data = {
-    #         "type": "text",
-    #         "title": "TaskingAI",
-    #         "collection_id": collection_id,
-    #         "record_id": record_id,
-    #         "content": "TaskingAI is an AI-native application development platform that unifies modules like Model, Retrieval, Assistant, and Tool into one seamless ecosystem, streamlining the creation and deployment of applications for developers.",
-    #         "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=34)
+    @pytest.mark.parametrize("text_splitter", text_splitter_list)
+    def test_update_record_by_text(self, collection_id, record_id, text_splitter):
+        # Update a record.
+
+        update_record_data = {
+            "type": "text",
+            "title": "TaskingAI",
+            "collection_id": collection_id,
+            "record_id": record_id,
+            "content": "TaskingAI is an AI-native application development platform that unifies modules like Model, Retrieval, Assistant, and Tool into one seamless ecosystem, streamlining the creation and deployment of applications for developers.",
+            "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)
+        records = list_records(collection_id=collection_id, order="desc", limit=20, after=None, before=None)
 
     # @pytest.mark.run(order=34)
     # @pytest.mark.parametrize("text_splitter", text_splitter_list)
@@ -295,7 +296,7 @@ def test_get_record(self, collection_id):
     @pytest.mark.run(order=79)
     def test_delete_record(self, collection_id):
         # List records.
-        time.sleep(Config.sleep_time)
+        # time.sleep(Config.sleep_time)
         records = list_records(collection_id=collection_id, order="desc", limit=20, after=None, before=None)
         old_nums = len(records)
         for index, record in enumerate(records):