From 7feadf02e4239dd64669730b4895de3c4d74a671 Mon Sep 17 00:00:00 2001
From: zhuwenxing <wenxing.zhu@zilliz.com>
Date: Mon, 20 Dec 2021 19:22:14 +0800
Subject: [PATCH] [skip e2e]Update deploy test scripts (#13771)

Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
---
 .../python_client/deploy/scripts/action_after_reinstall.py  | 4 ++--
 tests/python_client/deploy/scripts/action_after_upgrade.py  | 6 ++++--
 .../python_client/deploy/scripts/action_before_reinstall.py | 5 +++--
 tests/python_client/deploy/scripts/action_before_upgrade.py | 5 +++--
 4 files changed, 12 insertions(+), 8 deletions(-)

diff --git a/tests/python_client/deploy/scripts/action_after_reinstall.py b/tests/python_client/deploy/scripts/action_after_reinstall.py
index 2ac9bf18b8..df490bb830 100644
--- a/tests/python_client/deploy/scripts/action_after_reinstall.py
+++ b/tests/python_client/deploy/scripts/action_after_reinstall.py
@@ -5,7 +5,7 @@ from utils import *
 def task_1():
     """
     task_1:
-        before reinstall: create collection and insert data
+        before reinstall: create collection and insert data, load and search
         after reinstall: get collection, load, search, create index, load, and search
     """
     prefix = "task_1_"
@@ -18,7 +18,7 @@ def task_1():
 def task_2():
     """
     task_2:
-        before reinstall: create collection, insert data and create index
+        before reinstall: create collection, insert data and create index, load and search
         after reinstall: get collection, load, search, insert data, create index, load, and search
     """
     prefix = "task_2_"
diff --git a/tests/python_client/deploy/scripts/action_after_upgrade.py b/tests/python_client/deploy/scripts/action_after_upgrade.py
index 52adf56c84..e664812501 100644
--- a/tests/python_client/deploy/scripts/action_after_upgrade.py
+++ b/tests/python_client/deploy/scripts/action_after_upgrade.py
@@ -5,7 +5,7 @@ from utils import *
 def task_1():
     """
     task_1:
-        before upgrade: create collection and insert data
+        before upgrade: create collection and insert data, load and search
         after upgrade: get collection, load, search, create index, load, and search
     """
     prefix = "task_1_"
@@ -13,12 +13,14 @@ def task_1():
     get_collections(prefix)
     load_and_search(prefix)
     create_collections_and_insert_data(prefix)
+    create_index(prefix)
+    load_and_search(prefix)
 
 
 def task_2():
     """
     task_2:
-        before upgrade: create collection, insert data and create index
+        before upgrade: create collection, insert data and create index, load and search
         after upgrade: get collection, load, search, insert data, create index, load, and search
     """
     prefix = "task_2_"
diff --git a/tests/python_client/deploy/scripts/action_before_reinstall.py b/tests/python_client/deploy/scripts/action_before_reinstall.py
index 766cde0586..842fdb0b04 100644
--- a/tests/python_client/deploy/scripts/action_before_reinstall.py
+++ b/tests/python_client/deploy/scripts/action_before_reinstall.py
@@ -5,7 +5,7 @@ from utils import *
 def task_1():
     """
     task_1:
-        before reinstall: create collection and insert data
+        before reinstall: create collection and insert data, load and search
         after reinstall: get collection, load, search, create index, load, and search
     """
     prefix = "task_1_"
@@ -13,12 +13,13 @@ def task_1():
     get_collections(prefix)
     load_and_search(prefix)
     create_collections_and_insert_data(prefix)
+    load_and_search(prefix)
 
 
 def task_2():
     """
     task_2:
-        before reinstall: create collection, insert data and create index
+        before reinstall: create collection, insert data and create index,load and search
         after reinstall: get collection, load, search, insert data, create index, load, and search
     """
     prefix = "task_2_"
diff --git a/tests/python_client/deploy/scripts/action_before_upgrade.py b/tests/python_client/deploy/scripts/action_before_upgrade.py
index 70f834a178..b151faa840 100644
--- a/tests/python_client/deploy/scripts/action_before_upgrade.py
+++ b/tests/python_client/deploy/scripts/action_before_upgrade.py
@@ -5,7 +5,7 @@ from utils import *
 def task_1():
     """
     task_1:
-        before upgrade: create collection and insert data
+        before upgrade: create collection and insert data, load and search
         after upgrade: get collection, load, search, create index, load, and search
     """
     prefix = "task_1_"
@@ -13,12 +13,13 @@ def task_1():
     get_collections(prefix)
     load_and_search(prefix)
     create_collections_and_insert_data(prefix)
+    load_and_search(prefix)
 
 
 def task_2():
     """
     task_2:
-        before upgrade: create collection, insert data and create index
+        before upgrade: create collection, insert data and create index, load and search
         after upgrade: get collection, load, search, insert data, create index, load, and search
     """
     prefix = "task_2_"