Modify function name in ui_client.py

6 files changed, 15 insertions, 18 deletions

diff --git a/support/general_ui_utils/object/search_objects.py b/support/general_ui_utils/object/search_objects.py
index c47adee72..ff11d6d1f 100644
--- a/support/general_ui_utils/object/search_objects.py
+++ b/support/general_ui_utils/object/search_objects.py
@@ -102,7 +102,7 @@ class SearchObjects:
 
         return search_value
 
-    def get_object_uuids(self,name,object_element_position):
+    def get_objects_uuid(self,name,object_element_position):
         try:
             page_jump_element_position = object_element_position["page_jump"]
             # 打开该列表
diff --git a/support/ui_utils/objects/search_objects_example.py b/support/ui_utils/objects/search_objects_example.py
index 8a9bac514..c35782594 100644
--- a/support/ui_utils/objects/search_objects_example.py
+++ b/support/ui_utils/objects/search_objects_example.py
@@ -19,7 +19,7 @@ class SearchObjects:
 
     def search(self,name,policy_configuration,search_type):
         if "name" != "":
-            self.get_object_uuids(name,policy_configuration)
+            self.get_objects_uuid(name,policy_configuration)
         else:
             self.search_objects(policy_configuration,search_type)
 
@@ -112,7 +112,7 @@ class SearchObjects:
 
         return search_value
 
-    def get_object_uuids(self,name,object_configuration):
+    def get_objects_uuid(self,name,object_configuration):
         try:
             object_uuids_list = []
             object_uuids_temp_dict = {}
diff --git a/support/ui_utils/policies/edit_rules.py b/support/ui_utils/policies/edit_rules.py
index cc48890c6..a74f786c4 100644
--- a/support/ui_utils/policies/edit_rules.py
+++ b/support/ui_utils/policies/edit_rules.py
@@ -14,7 +14,7 @@ class EditRules:
     def __init__(self, driver):
         self.driver = driver
         
-    def edit_rules(self, policy_configuration, vsys_name=""):
+    def edit_rules(self, policy_configuration):
         try:
             type = policy_configuration["type"]
             name = policy_configuration["name"]
diff --git a/support/ui_utils/policies/search_rules.py b/support/ui_utils/policies/search_rules.py
index 148ccdcb1..7534fee2b 100644
--- a/support/ui_utils/policies/search_rules.py
+++ b/support/ui_utils/policies/search_rules.py
@@ -55,7 +55,7 @@ class SearchRules:
             print(datetime.now().strftime("%Y-%m-%d %H:%M:%S"), str(datetime.now().microsecond)[:3], "When searching rule, the exception error: ", str(e), flush=True)
             raise
 
-    def get_rule_uuids(self, policy_configuration):
+    def get_rules_uuid(self, policy_configuration):
         try:
             type = policy_configuration["type"]
             name = policy_configuration["name"]
diff --git a/support/ui_utils/ui_client.py b/support/ui_utils/ui_client.py
index bc8670f86..5b41fe7cb 100644
--- a/support/ui_utils/ui_client.py
+++ b/support/ui_utils/ui_client.py
@@ -36,15 +36,14 @@ class UIClient:
         error = objects.search_objects(name, policy_configuration)
         return error
     
-    def get_object_uuids(self, policy_configuration):
+    def get_objects_uuid(self, policy_configuration):
         objects = SearchObjects(self.driver)
-        objects_tuple, error = objects.get_object_uuids(policy_configuration)
-        # self.objects_tuple = objects_tuple
+        objects_tuple, error = objects.get_objects_uuid(policy_configuration)
         return objects_tuple, error
 
-    def edit_objects(self, objects_tuple, src_item,new_item):
+    def edit_objects(self, objects_tuple, src_item, new_item):
         objects = EditObjects(self.driver)
-        error = objects.edit_objects(objects_tuple, src_item,new_item)
+        error = objects.edit_objects(objects_tuple, src_item, new_item)
         return error
     
     def delete_objects(self, objects_tuple):
@@ -61,10 +60,9 @@ class UIClient:
         error = profiles.search_profiles(policy_configuration)
         return error
     
-    def get_profile_uuids(self, policy_configuration):
+    def get_profiles_uuid(self, policy_configuration):
         profiles = SearchProfiles(self.driver)
-        profiles_tuple, error = profiles.get_profile_uuids(policy_configuration)
-        # self.profiles_tuple = profiles_tuple
+        profiles_tuple, error = profiles.get_profiles_uuid(policy_configuration)
         return profiles_tuple, error
 
     def create_rules(self, policy_configuration):
@@ -78,10 +76,9 @@ class UIClient:
         error = rules.search_rules(policy_configuration)
         return error
     
-    def get_rule_uuids(self, policy_configuration):
+    def get_rules_uuid(self, policy_configuration):
         rules = SearchRules(self.driver)
-        rules_tuple, error = rules.get_rule_uuids(policy_configuration)
-        # self.rules_tuple = rules_tuple
+        rules_tuple, error = rules.get_rules_uuid(policy_configuration)
         return rules_tuple, error
 
     def edit_rules(self, policy_configuration):
diff --git a/tests/example/rule_with_traffic_example.py b/tests/example/rule_with_traffic_example.py
index 7f9a20ed4..db389c869 100644
--- a/tests/example/rule_with_traffic_example.py
+++ b/tests/example/rule_with_traffic_example.py
@@ -89,13 +89,13 @@ def run(parameter):
             ui_error = ui_client.create_objects(policy_configuration)
             if len(ui_error) > 0:
                 return ui_error
-            objects_tuple, ui_error = ui_client.get_object_uuids(policy_configuration)
+            objects_tuple, ui_error = ui_client.get_objects_uuid(policy_configuration)
             if len(ui_error) > 0:
                 return ui_error
             ui_error = ui_client.create_rules(policy_configuration)
             if len(ui_error) > 0:
                 return ui_error
-            rules_tuple, ui_error = ui_client.get_rule_uuids(policy_configuration)
+            rules_tuple, ui_error = ui_client.get_rules_uuid(policy_configuration)
             if len(ui_error) > 0:
                 return ui_error
         elif parameter["initiation_method"] == "api":