test: add test

Author: taskingaijc

test/testcase/test_sync/test_sync_retrieval.py

@@ -248,8 +248,7 @@ def test_update_record_by_text(self, collection_id, record_id, text_splitter):
         res = update_record(**update_record_data)
         res_dict = vars(res)
         assume_record_result(update_record_data, res_dict)
-        time.sleep(2)
-        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)
@@ -268,8 +267,7 @@ def test_update_record_by_web(self, collection_id, record_id, text_splitter):
         res = update_record(**update_record_data)
         res_dict = vars(res)
         assume_record_result(update_record_data, res_dict)
-        time.sleep(2)
-        records = list_records(collection_id=collection_id, order="desc", limit=20, after=None, before=None)
+
 
     @pytest.mark.run(order=35)
     @pytest.mark.parametrize("upload_file_data", upload_file_data_list[2:3])
@@ -295,8 +293,7 @@ def test_update_record_by_file(self, collection_id, record_id, upload_file_data)
         res = update_record(**update_record_data)
         res_dict = vars(res)
         assume_record_result(update_record_data, res_dict)
-        time.sleep(2)
-        records = list_records(collection_id=collection_id, order="desc", limit=20, after=None, before=None)
+
 
     @pytest.mark.run(order=79)
     def test_delete_record(self, collection_id):
@@ -339,6 +336,7 @@ def test_query_chunks(self, collection_id):
 
     @pytest.mark.run(order=42)
     def test_create_chunk(self, collection_id):
+        time.sleep(2)
         # Create a chunk.
         create_chunk_data = {
             "collection_id": collection_id,
@@ -399,23 +397,32 @@ def test_update_chunk(self, collection_id, chunk_id):
         res = update_chunk(**update_chunk_data)
         res_dict = vars(res)
         assume_chunk_result(update_chunk_data, res_dict)
-        records = list_records(collection_id=collection_id, order="desc", limit=20, after=None, before=None)
+
 
     @pytest.mark.run(order=46)
-    def test_delete_chunk(self, collection_id):
+    def test_delete_chunk(self, collection_id, chunk_id):
         # List chunks.
 
-        chunks = list_chunks(collection_id=collection_id, limit=5)
-        for index, chunk in enumerate(chunks):
-            chunk_id = chunk.chunk_id
-
-            # Delete a chunk.
-
-            delete_chunk(collection_id=collection_id, chunk_id=chunk_id)
+        # chunks = list_chunks(collection_id=collection_id, limit=5)
+        # for index, chunk in enumerate(chunks):
+        #     chunk_id = chunk.chunk_id
+        #
+        #     # Delete a chunk.
+        #
+        #     delete_chunk(collection_id=collection_id, chunk_id=chunk_id)
+        #
+        #     # List chunks.
+        #
+        #     new_chunks = list_chunks(collection_id=collection_id)
+        #     chunk_ids = [chunk.chunk_id for chunk in new_chunks]
+        #     pytest.assume(chunk_id not in chunk_ids)
+        #     records = list_records(collection_id=collection_id, order="desc", limit=20, after=None, before=None)
+        #     for record in records:
+        delete_chunk(collection_id=collection_id, chunk_id=chunk_id)
 
-            # List chunks.
+        # List chunks.
 
-            new_chunks = list_chunks(collection_id=collection_id)
-            chunk_ids = [chunk.chunk_id for chunk in new_chunks]
-            pytest.assume(chunk_id not in chunk_ids)
-            records = list_records(collection_id=collection_id, order="desc", limit=20, after=None, before=None)
+        new_chunks = list_chunks(collection_id=collection_id)
+        chunk_ids = [chunk.chunk_id for chunk in new_chunks]
+        pytest.assume(chunk_id not in chunk_ids)
+        records = list_records(collection_id=collection_id, order="desc", limit=20, after=None, before=None)