Performance testing

Framework/Built_In_Automation/Performance_Testing/BuiltInFunctions.py

@@ -0,0 +1,159 @@
+from Framework.Utilities.decorators import logger, deprecated
+import inspect,sys,random
+from Framework.Utilities import CommonUtil, ConfigModule
+import traceback
+global sr
+from Framework.Built_In_Automation.Shared_Resources import BuiltInFunctionSharedResources as sr
+
+MODULE_NAME = inspect.getmodulename(__file__)
+
+@logger
+def locust_config(data_set):
+    """
+    Save locust configuration to a variable
+    Args:
+        data_set:
+          swarm              | input parameter    | integer
+          spawn              | input parameter    | integer
+          locust config      | performance action | variable_name
+
+    Returns:
+        "passed" if success.
+        "zeuz_failed" otherwise.
+    """
+
+    sModuleInfo = inspect.currentframe().f_code.co_name + " : " + MODULE_NAME
+
+    try:
+        locust_var = losust_var_name = swarm = spawn = None
+        try:
+            for left, mid, right in data_set:
+                left = left.strip().lower()
+                if mid.strip().lower() == "input parameter":
+                    if "swarm" == left:
+                        swarm = float(right.strip().lower())
+                    elif "spawn" == left:
+                        spawn = float(right.strip().lower())
+                elif "action" == mid.strip().lower():
+                    if "locust config" == left:
+                        losust_var_name = right.strip()
+            if None in [losust_var_name,swarm,spawn]: 
+                CommonUtil.ExecLog(sModuleInfo,  f"dataset is inaccurate", 3)
+                return "zeuz_failed"
+            locust_var = {
+                            "locust_config": {
+                                "swarm": swarm,
+                                "spawn": spawn
+                            },
+                            "users": {}
+                    }    
+        except:
+            CommonUtil.ExecLog(sModuleInfo, "Failed to parse data.", 1)
+            traceback.print_exc()
+            return "zeuz_failed"
+        sr.Set_Shared_Variables(losust_var_name, locust_var)
+        return "passed"
+    except:
+        return CommonUtil.Exception_Handler(sys.exc_info())
+
+
+@logger
+def assign_locust_user(data_set):
+    """
+    Add Locust user configuration to the existing locust variable
+    This action provides that data to creates a user class into locust file
+    Args:
+        data_set:
+          wait_time          | input parameter    | between(int,int)
+          host               | input parameter    | url of host
+          name               | input parameter    | user/class name Case sensitive
+          type               | input parameter    | type of user/class
+          assign locust user | performance action | existing locust variable name
+    Returns:
+        "passed" if success.
+        "zeuz_failed" otherwise.
+    """
+
+    sModuleInfo = inspect.currentframe().f_code.co_name + " : " + MODULE_NAME
+
+    try:
+        locust_var = losust_var_name = name = user_type = wait_time = host = None
+        try:
+            for left, mid, right in data_set:
+                left = left.strip().lower()
+                if mid.strip().lower() == "input parameter":
+                    if "name" == left:
+                        name = right.strip()
+                    if "type" == left:
+                        user_type = right.strip().lower()
+                        user_type = 'HttpUser' if user_type == 'httpuser' else "User" if user_type == "user" else None
+                    elif "wait_time" == left:
+                        wait_time = right.strip().lower()
+                    elif "host" == left:
+                        host = right.strip().lower()
+                elif mid.strip().lower() == "action":
+                    if "assign locust user" == left:
+                        losust_var_name = right.strip()
+            if None in [losust_var_name,name,user_type,wait_time,host]: 
+                CommonUtil.ExecLog(sModuleInfo,  f"dataset is inaccurate", 3)
+                return "zeuz_failed"
+            locust_var = sr.Get_Shared_Variables(losust_var_name,log=False)
+            locust_var['users'][name] = {'type':user_type,'wait_time' : wait_time,'host':host,'tasks':[]}
+        except:
+            CommonUtil.ExecLog(sModuleInfo, "Failed to parse data.", 1)
+            traceback.print_exc()
+            return "zeuz_failed"
+        sr.Set_Shared_Variables(losust_var_name, locust_var)
+        return "passed"
+    except:
+        return CommonUtil.Exception_Handler(sys.exc_info())
+
+@logger
+def assign_locust_task(data_set):
+    """
+    Add task configuration to the existing locust user
+    This action provides that data to define the tasks performed by locust user/class
+    Args:
+        data_set:
+          action             | input parameter    | get/post etc
+          data               | input parameter    | url of endpoint
+          task name          | input parameter    | string
+          name               | input parameter    | existing user/class name Case sensitive
+          assign locust task | performance action | existing locust variable name
+    Returns:
+        "passed" if success.
+        "zeuz_failed" otherwise.
+    """
+
+    sModuleInfo = inspect.currentframe().f_code.co_name + " : " + MODULE_NAME
+
+    try:
+        locust_var = locust_var_name = action = data = name = task_name = None
+        try:
+            for left, mid, right in data_set:
+                left = left.strip().lower()
+                if mid.strip().lower() == "input parameter":
+                    if "action" == left.strip():
+                        action = right.strip().lower()
+                    elif "data" == left.strip():
+                        data = right.strip().lower()
+                    elif "name" == left.strip():
+                        name = right.strip()
+                    elif "task name" == left:
+                        task_name = right.strip().lower()
+                elif mid.strip().lower() == "action":
+                    if "assign locust task" == left:
+                        locust_var_name = right.strip()
+            if None in [locust_var_name,action,data,name,task_name]:
+                CommonUtil.ExecLog(sModuleInfo,  f"dataset is inaccurate", 3)
+                return "zeuz_failed"
+            locust_var = sr.Get_Shared_Variables(locust_var_name,log=False)
+            locust_var['users'][name]['tasks'].append({'action':action,'data':data,'name':task_name}) 
+        except:
+            CommonUtil.ExecLog(sModuleInfo, "Failed to parse data.", 1)
+            traceback.print_exc()
+            return "zeuz_failed"
+        sr.Set_Shared_Variables(locust_var_name, locust_var)
+        return "passed"
+    except:
+        return CommonUtil.Exception_Handler(sys.exc_info())

Framework/Built_In_Automation/Performance_Testing/tests/json_to_zeuz_json.py

@@ -0,0 +1,61 @@
+import json
+import pprint
+
+# Check if action is disabled
+def check_if_disabled(action):
+    if action.get('action_disabled') == 'false':
+        return True
+    else:
+        return False
+# Check if the action is a locust action
+def selected_locust_action(action):
+    all_locust_actions = ['locust config','assign locust user','assign locust task']
+    locust_actions =  [ac[0] for ac in action['Action data'] if ac[0] in all_locust_actions]
+    if(locust_actions):
+        return locust_actions[0]
+    else:
+        return False
+
+# Get the values for the action paramteres
+def get_values(step_actions,action_val_map):
+    data = dict()
+    for sa in step_actions:
+        variable_name = action_val_map.get(sa[0])
+        if(variable_name):
+            variable_value = sa[2]
+            data[variable_name] = variable_value
+    return data
+
+def convert_json(input_json_file_path,output_json_file_path):
+    input_json_file_path = os.path.dirname(os.path.realpath(__file__)) + '/' + input_json_file_path
+    output_json_file_path = os.path.dirname(os.path.realpath(__file__)) + '/' + output_json_file_path
+    input_json_data = json.load(open(input_json_file_path))
+    testcase = input_json_data["TestCases"]
+    steps = testcase[0]["Steps"]
+    actions = []
+    for step in steps:
+        actions = actions + step["Step actions"]
+    all_locusts = dict()
+    for action in actions:
+        locust_action = selected_locust_action(action)
+        if(check_if_disabled(action) == False and locust_action):
+                step_actions = action['Action data']
+                if(locust_action == 'locust config'):
+                    params = get_values(step_actions,{'swarm':'swarm','spawn':'spawn','locust config':'variable_name'})
+                    all_locusts[params['variable_name']] = ({"locust_config": {"swarm": params['swarm'],"spawn": params['spawn']},"users": {}})
+                if(locust_action == 'assign locust user'):
+                    params = get_values(step_actions,{'type':'type','name':'name','host':'host','wait_time':'wait_time','assign locust user':'variable_name'})
+                    all_locusts[params['variable_name']]['users'][params['name']] = {'type':params['type'],'wait_time' : params['wait_time'],'host':params['host'],'tasks':[]}
+
+                if(locust_action == 'assign locust task'):
+                    params = get_values(step_actions,{'action':'http_method','data':'data','name':'name','assign locust task':'variable_name'})
+                    all_locusts[params['variable_name']]['users'][params['name']]['tasks'].append({'action': 'client.' + params['http_method'],'data':params['data'],'name':params['name']}) 
+
+    pprint.pprint(all_locusts)
+
+    with open(output_json_file_path, "w") as outfile:
+        json.dump(all_locusts,outfile,indent=4 )
+
+import os
+if __name__ == '__main__':
+    convert_json(input_json_file_path="sample2.json", output_json_file_path="sample_zeuz_json.json")

Framework/Built_In_Automation/Performance_Testing/tests/sample1.json

@@ -0,0 +1,143 @@
+[
+    {
+        "action_name": "Locust config",
+        "action_disabled": "true",
+        "step_actions": [
+            [
+                "swarm",
+                "input parameter",
+                "10"
+            ],
+            [
+                "spawn",
+                "input parameter",
+                "100"
+            ],
+            [
+                "locust config",
+                "performance action",
+                "PERF_VARIABLE"
+            ]
+        ]
+    },
+    {
+        "action_name": "Define user",
+        "action_disabled": "true",
+        "step_actions": [
+            [
+                "type",
+                "input parameter",
+                "HttpUser"
+            ],
+            [
+                "name",
+                "input parameter",
+                "WebsiteUser"
+            ],
+            [
+                "host",
+                "input parameter",
+                "https://xyz.com"
+            ],
+            [
+                "wait_time",
+                "input parameter",
+                "between(5, 15)"
+            ],
+            [
+                "assign locust user",
+                "performance action",
+                "PERF_VARIABLE"
+            ]
+        ]
+    },
+    {
+        "action_name": "Define task",
+        "action_disabled": "true",
+        "step_actions": [
+            [
+                "action",
+                "input parameter",
+                "get"
+            ],
+            [
+                "data",
+                "input parameter",
+                "/"
+            ],
+            [
+                "name",
+                "input parameter",
+                "WebsiteUser"
+            ],
+            [
+                "assign locust task",
+                "performance action",
+                "PERF_VARIABLE"
+            ]
+        ]
+    },
+    {
+        "action_name": "Define task",
+        "action_disabled": "true",
+        "step_actions": [
+            [
+                "action",
+                "input parameter",
+                "get"
+            ],
+            [
+                "data",
+                "input parameter",
+                "/static/assets.js"
+            ],
+            [
+                "name",
+                "input parameter",
+                "WebsiteUser"
+            ],
+            [
+                "assign locust task",
+                "performance action",
+                "PERF_VARIABLE"
+            ]
+        ]
+    },
+    {
+        "action_name": "Define task",
+        "action_disabled": "true",
+        "step_actions": [
+            [
+                "action",
+                "input parameter",
+                "post"
+            ],
+            [
+                "data",
+                "input parameter",
+                "/submit/form"
+            ],
+            [
+                "name",
+                "input parameter",
+                "WebsiteUser"
+            ],
+            [
+                "assign locust task",
+                "performance action",
+                "PERF_VARIABLE"
+            ]
+        ]
+    },
+    {
+        "action_name": "Generate locust python file + run locust file",
+        "action_disabled": "true",
+        "step_actions": [
+            [
+                "run performance test",
+                "performance action",
+                "PERF_VARIABLE"
+            ]
+        ]
+    }
+]