test: add restore v2 test case (zilliztech#506)

Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
huanghaoyuanhhy · Jan 17, 2025 · cae2549 · cae2549
1 parent 312e7e9
commit cae2549
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/testcases/test_restore_backup.py b/tests/testcases/test_restore_backup.py
@@ -31,9 +31,11 @@ class TestRestoreBackup(TestcaseBase):
     @pytest.mark.parametrize("is_async", [True, False])
     @pytest.mark.parametrize("collection_need_to_restore", [3])
     @pytest.mark.parametrize("collection_type", ["all"])
+    @pytest.mark.parametrize("use_v2_restore", [True, False])
     @pytest.mark.tags(CaseLabel.L0)
     def test_milvus_restore_back(
         self,
+        use_v2_restore,
         collection_type,
         collection_need_to_restore,
         is_async,
@@ -110,6 +112,7 @@ def test_milvus_restore_back(
                 "collection_suffix": suffix,
                 "collection_names": restore_collections,
             }
+        payload["useV2Restore"] = use_v2_restore
         t0 = time.time()
         res = self.client.restore_backup(payload)
         restore_id = res["data"]["id"]