# encoding: utf-8 import ast import os import re from httprunner import exceptions, utils, validator from httprunner.compat import basestring, builtin_str, numeric_types, str # TODO: change variable notation from $var to {{var}} # $var_1 variable_regex_compile = re.compile(r"\$(\w+)") # ${func1($var_1, $var_3)} function_regex_compile = re.compile(r"\$\{(\w+)\(([\$\w\.\-/\s=,]*)\)\}") def parse_string_value(str_value): """ parse string to number if possible e.g. "123" => 123 "12.2" => 12.3 "abc" => "abc" "$var" => "$var" """ try: return ast.literal_eval(str_value) except ValueError: return str_value except SyntaxError: # e.g. $var, ${func} return str_value def is_variable_exist(content): if not isinstance(content, basestring): return False return True if variable_regex_compile.search(content) else False def is_function_exist(content): if not isinstance(content, basestring): return False return True if function_regex_compile.search(content) else False def regex_findall_variables(content): """ extract all variable names from content, which is in format $variable Args: content (str): string content Returns: list: variables list extracted from string content Examples: >>> regex_findall_variables("$variable") ["variable"] >>> regex_findall_variables("/blog/$postid") ["postid"] >>> regex_findall_variables("/$var1/$var2") ["var1", "var2"] >>> regex_findall_variables("abc") [] """ try: return variable_regex_compile.findall(content) except TypeError: return [] def regex_findall_functions(content): """ extract all functions from string content, which are in format ${fun()} Args: content (str): string content Returns: list: functions list extracted from string content Examples: >>> regex_findall_functions("${func(5)}") ["func(5)"] >>> regex_findall_functions("${func(a=1, b=2)}") ["func(a=1, b=2)"] >>> regex_findall_functions("/api/1000?_t=${get_timestamp()}") ["get_timestamp()"] >>> regex_findall_functions("/api/${add(1, 2)}") ["add(1, 2)"] >>> regex_findall_functions("/api/${add(1, 2)}?_t=${get_timestamp()}") ["add(1, 2)", "get_timestamp()"] """ try: return function_regex_compile.findall(content) except TypeError: return [] def parse_parameters(parameters, variables_mapping=None, functions_mapping=None): """ parse parameters and generate cartesian product. Args: parameters (list) parameters: parameter name and value in list parameter value may be in three types: (1) data list, e.g. ["iOS/10.1", "iOS/10.2", "iOS/10.3"] (2) call built-in parameterize function, "${parameterize(account.csv)}" (3) call custom function in debugtalk.py, "${gen_app_version()}" variables_mapping (dict): variables mapping loaded from testcase config functions_mapping (dict): functions mapping loaded from debugtalk.py Returns: list: cartesian product list Examples: >>> parameters = [ {"user_agent": ["iOS/10.1", "iOS/10.2", "iOS/10.3"]}, {"username-password": "${parameterize(account.csv)}"}, {"app_version": "${gen_app_version()}"} ] >>> parse_parameters(parameters) """ variables_mapping = variables_mapping or {} functions_mapping = functions_mapping or {} parsed_parameters_list = [] parameters = utils.ensure_mapping_format(parameters) for parameter_name, parameter_content in parameters.items(): parameter_name_list = parameter_name.split("-") if isinstance(parameter_content, list): # (1) data list # e.g. {"app_version": ["2.8.5", "2.8.6"]} # => [{"app_version": "2.8.5", "app_version": "2.8.6"}] # e.g. {"username-password": [["user1", "111111"], ["test2", "222222"]} # => [{"username": "user1", "password": "111111"}, {"username": "user2", "password": "222222"}] parameter_content_list = [] for parameter_item in parameter_content: if not isinstance(parameter_item, (list, tuple)): # "2.8.5" => ["2.8.5"] parameter_item = [parameter_item] # ["app_version"], ["2.8.5"] => {"app_version": "2.8.5"} # ["username", "password"], ["user1", "111111"] => {"username": "user1", "password": "111111"} parameter_content_dict = dict(zip(parameter_name_list, parameter_item)) parameter_content_list.append(parameter_content_dict) else: # (2) & (3) parsed_variables_mapping = parse_variables_mapping( variables_mapping ) parsed_parameter_content = eval_lazy_data( parameter_content, parsed_variables_mapping, functions_mapping ) if not isinstance(parsed_parameter_content, list): raise exceptions.ParamsError("parameters syntax error!") parameter_content_list = [] for parameter_item in parsed_parameter_content: if isinstance(parameter_item, dict): # get subset by parameter name # {"app_version": "${gen_app_version()}"} # gen_app_version() => [{'app_version': '2.8.5'}, {'app_version': '2.8.6'}] # {"username-password": "${get_account()}"} # get_account() => [ # {"username": "user1", "password": "111111"}, # {"username": "user2", "password": "222222"} # ] parameter_dict = {key: parameter_item[key] for key in parameter_name_list} elif isinstance(parameter_item, (list, tuple)): # {"username-password": "${get_account()}"} # get_account() => [("user1", "111111"), ("user2", "222222")] parameter_dict = dict(zip(parameter_name_list, parameter_item)) elif len(parameter_name_list) == 1: # {"user_agent": "${get_user_agent()}"} # get_user_agent() => ["iOS/10.1", "iOS/10.2"] parameter_dict = { parameter_name_list[0]: parameter_item } parameter_content_list.append(parameter_dict) parsed_parameters_list.append(parameter_content_list) return utils.gen_cartesian_product(*parsed_parameters_list) ############################################################################### ## parse content with variables and functions mapping ############################################################################### def get_mapping_variable(variable_name, variables_mapping): """ get variable from variables_mapping. Args: variable_name (str): variable name variables_mapping (dict): variables mapping Returns: mapping variable value. Raises: exceptions.VariableNotFound: variable is not found. """ try: return variables_mapping[variable_name] except KeyError: raise exceptions.VariableNotFound("{} is not found.".format(variable_name)) def get_mapping_function(function_name, functions_mapping): """ get function from functions_mapping, if not found, then try to check if builtin function. Args: variable_name (str): variable name variables_mapping (dict): variables mapping Returns: mapping function object. Raises: exceptions.FunctionNotFound: function is neither defined in debugtalk.py nor builtin. """ if function_name in functions_mapping: return functions_mapping[function_name] elif function_name in ["parameterize", "P"]: from httprunner import loader return loader.load_csv_file elif function_name in ["environ", "ENV"]: return utils.get_os_environ try: # check if HttpRunner builtin functions from httprunner import loader built_in_functions = loader.load_builtin_functions() return built_in_functions[function_name] except KeyError: pass try: # check if Python builtin functions item_func = eval(function_name) if callable(item_func): # is builtin function return item_func except (NameError, TypeError): # is not builtin function raise exceptions.FunctionNotFound("{} is not found.".format(function_name)) def parse_function_params(params): """ parse function params to args and kwargs. Args: params (str): function param in string Returns: dict: function meta dict { "args": [], "kwargs": {} } Examples: >>> parse_function_params("") {'args': [], 'kwargs': {}} >>> parse_function_params("5") {'args': [5], 'kwargs': {}} >>> parse_function_params("1, 2") {'args': [1, 2], 'kwargs': {}} >>> parse_function_params("a=1, b=2") {'args': [], 'kwargs': {'a': 1, 'b': 2}} >>> parse_function_params("1, 2, a=3, b=4") {'args': [1, 2], 'kwargs': {'a':3, 'b':4}} """ function_meta = { "args": [], "kwargs": {} } params_str = params.strip() if params_str == "": return function_meta args_list = params_str.split(',') for arg in args_list: arg = arg.strip() if '=' in arg: key, value = arg.split('=') function_meta["kwargs"][key.strip()] = parse_string_value(value.strip()) else: function_meta["args"].append(parse_string_value(arg)) return function_meta class LazyFunction(object): """ call function lazily. """ def __init__(self, function_meta, functions_mapping=None, check_variables_set=None): """ init LazyFunction object with function_meta Args: function_meta (dict): function name, args and kwargs. { "func_name": "func", "args": [1, 2] "kwargs": {"a": 3, "b": 4} } """ self.functions_mapping = functions_mapping or {} self.check_variables_set = check_variables_set or set() self.cache_key = None self.__parse(function_meta) def __parse(self, function_meta): """ init func as lazy functon instance Args: function_meta (dict): function meta including name, args and kwargs """ self._func = get_mapping_function( function_meta["func_name"], self.functions_mapping ) self._args = prepare_lazy_data( function_meta.get("args", []), self.functions_mapping, self.check_variables_set ) self._kwargs = prepare_lazy_data( function_meta.get("kwargs", {}), self.functions_mapping, self.check_variables_set ) if self._func.__name__ == "load_csv_file": if len(self._args) != 1 or self._kwargs: raise exceptions.ParamsError("P() should only pass in one argument!") self._args = [self._args[0]] elif self._func.__name__ == "get_os_environ": if len(self._args) != 1 or self._kwargs: raise exceptions.ParamsError("ENV() should only pass in one argument!") self._args = [self._args[0]] def __repr__(self): return "LazyFunction({})".format(self._func.__name__) def __prepare_cache_key(self, args, kwargs): return (self._func.__name__, repr(args), repr(kwargs)) def to_value(self, variables_mapping=None): """ parse lazy data with evaluated variables mapping. Notice: variables_mapping should not contain any variable or function. """ variables_mapping = variables_mapping or {} args = parse_lazy_data(self._args, variables_mapping) kwargs = parse_lazy_data(self._kwargs, variables_mapping) self.cache_key = self.__prepare_cache_key(args, kwargs) return self._func(*args, **kwargs) cached_functions_mapping = {} """ cached function calling results. """ class LazyString(object): """ evaluate string lazily. """ def __init__(self, raw_string, functions_mapping=None, check_variables_set=None, cached=False): """ make raw_string as lazy object with functions_mapping check if any variable undefined in check_variables_set """ self.raw_string = raw_string self.functions_mapping = functions_mapping or {} self.check_variables_set = check_variables_set or set() self.cached = cached self.__parse(raw_string) def __parse(self, raw_string): """ parse raw string, replace function and variable with {} Args: raw_string(str): string with functions or varialbes e.g. "ABC${func2($a, $b)}DE$c" Returns: string: "ABC{}DE{}" args: ["${func2($a, $b)}", "$c"] """ self._string = raw_string args_mapping = {} # Notice: functions must be handled before variables # search function like ${func($a, $b)} func_match_list = regex_findall_functions(self._string) match_start_position = 0 for func_match in func_match_list: func_str = "${%s(%s)}" % (func_match[0], func_match[1]) match_start_position = raw_string.index(func_str, match_start_position) self._string = self._string.replace(func_str, "{}", 1) function_meta = parse_function_params(func_match[1]) function_meta = { "func_name": func_match[0] } function_meta.update(parse_function_params(func_match[1])) lazy_func = LazyFunction( function_meta, self.functions_mapping, self.check_variables_set ) args_mapping[match_start_position] = lazy_func # search variable like $var var_match_list = regex_findall_variables(self._string) match_start_position = 0 for var_name in var_match_list: # check if any variable undefined in check_variables_set if var_name not in self.check_variables_set: raise exceptions.VariableNotFound(var_name) var = "${}".format(var_name) match_start_position = raw_string.index(var, match_start_position) # TODO: escape '{' and '}' # self._string = self._string.replace("{", "{{") # self._string = self._string.replace("}", "}}") self._string = self._string.replace(var, "{}", 1) args_mapping[match_start_position] = var_name self._args = [args_mapping[key] for key in sorted(args_mapping.keys())] def __repr__(self): return "LazyString({})".format(self.raw_string) def to_value(self, variables_mapping=None): """ parse lazy data with evaluated variables mapping. Notice: variables_mapping should not contain any variable or function. """ variables_mapping = variables_mapping or {} args = [] for arg in self._args: if isinstance(arg, LazyFunction): if self.cached and arg.cache_key and arg.cache_key in cached_functions_mapping: value = cached_functions_mapping[arg.cache_key] else: value = arg.to_value(variables_mapping) cached_functions_mapping[arg.cache_key] = value args.append(value) else: # variable var_value = get_mapping_variable(arg, variables_mapping) args.append(var_value) if self._string == "{}": return args[0] else: return self._string.format(*args) def prepare_lazy_data(content, functions_mapping=None, check_variables_set=None, cached=False): """ make string in content as lazy object with functions_mapping Raises: exceptions.VariableNotFound: if any variable undefined in check_variables_set """ # TODO: refactor type check if content is None or isinstance(content, (numeric_types, bool, type)): return content elif isinstance(content, (list, set, tuple)): return [ prepare_lazy_data( item, functions_mapping, check_variables_set, cached ) for item in content ] elif isinstance(content, dict): parsed_content = {} for key, value in content.items(): parsed_key = prepare_lazy_data( key, functions_mapping, check_variables_set, cached ) parsed_value = prepare_lazy_data( value, functions_mapping, check_variables_set, cached ) parsed_content[parsed_key] = parsed_value return parsed_content elif isinstance(content, basestring): # content is in string format here if not (is_variable_exist(content) or is_function_exist(content)): # content is neither variable nor function return content functions_mapping = functions_mapping or {} check_variables_set = check_variables_set or set() content = content.strip() content = LazyString(content, functions_mapping, check_variables_set, cached) return content def parse_lazy_data(content, variables_mapping=None): """ parse lazy data with evaluated variables mapping. Notice: variables_mapping should not contain any variable or function. """ # TODO: refactor type check if content is None or isinstance(content, (numeric_types, bool, type)): return content elif isinstance(content, LazyString): variables_mapping = utils.ensure_mapping_format(variables_mapping or {}) return content.to_value(variables_mapping) elif isinstance(content, (list, set, tuple)): return [ parse_lazy_data(item, variables_mapping) for item in content ] elif isinstance(content, dict): parsed_content = {} for key, value in content.items(): parsed_key = parse_lazy_data(key, variables_mapping) parsed_value = parse_lazy_data(value, variables_mapping) parsed_content[parsed_key] = parsed_value return parsed_content return content def eval_lazy_data(content, variables_mapping=None, functions_mapping=None): """ evaluate data instantly. Notice: variables_mapping should not contain any variable or function. """ variables_mapping = variables_mapping or {} check_variables_set = set(variables_mapping.keys()) return parse_lazy_data( prepare_lazy_data( content, functions_mapping, check_variables_set ), variables_mapping ) def extract_variables(content): """ extract all variables in content recursively. """ if isinstance(content, (list, set, tuple)): variables = set() for item in content: variables = variables | extract_variables(item) return variables elif isinstance(content, dict): variables = set() for key, value in content.items(): variables = variables | extract_variables(value) return variables elif isinstance(content, LazyString): return set(regex_findall_variables(content.raw_string)) return set() def parse_variables_mapping(variables_mapping, ignore=False): """ eval each prepared variable and function in variables_mapping. Args: variables_mapping (dict): { "varA": LazyString(123$varB), "varB": LazyString(456$varC), "varC": LazyString(${sum_two($a, $b)}), "a": 1, "b": 2, "c": {"key": LazyString($b)}, "d": [LazyString($a), 3] } ignore (bool): If set True, VariableNotFound will be ignored. This is used when initializing tests. Returns: dict: parsed variables_mapping should not contain any variable or function. { "varA": "1234563", "varB": "4563", "varC": "3", "a": 1, "b": 2, "c": {"key": 2}, "d": [1, 3] } """ variables_mapping = variables_mapping or {} ref_variables_set = set() parsed_variables_mapping = {} while len(parsed_variables_mapping) != len(variables_mapping): for var_name in variables_mapping: if var_name in parsed_variables_mapping: continue value = variables_mapping[var_name] variables = extract_variables(value) # check if reference variable itself if var_name in variables: # e.g. # var_name = "token" # variables_mapping = {"token": LazyString($token)} # var_name = "key" # variables_mapping = {"key": [LazyString($key), 2]} if ignore: parsed_variables_mapping[var_name] = value continue raise exceptions.VariableNotFound(var_name) if variables: # reference other variable, or function call with other variable # e.g. {"varA": "123$varB", "varB": "456$varC"} # e.g. {"varC": "${sum_two($a, $b)}"} if any([_var_name not in parsed_variables_mapping for _var_name in variables]): # reference variable not parsed continue parsed_value = parse_lazy_data(value, parsed_variables_mapping) parsed_variables_mapping[var_name] = parsed_value return parsed_variables_mapping def _extend_with_api(test_dict, api_def_dict): """ extend test with api definition, test will merge and override api definition. Args: test_dict (dict): test block, this will override api_def_dict api_def_dict (dict): api definition Examples: >>> api_def_dict = { "name": "get token 1", "request": {...}, "validate": [{'eq': ['status_code', 200]}] } >>> test_dict = { "name": "get token 2", "extract": {"token": "content.token"}, "validate": [{'eq': ['status_code', 201]}, {'len_eq': ['content.token', 16]}] } >>> _extend_with_api(test_dict, api_def_dict) >>> print(test_dict) { "name": "get token 2", "request": {...}, "extract": {"token": "content.token"}, "validate": [{'eq': ['status_code', 201]}, {'len_eq': ['content.token', 16]}] } """ # override api name test_dict.setdefault("name", api_def_dict.pop("name", "")) # override variables def_variables = api_def_dict.pop("variables", []) test_dict["variables"] = utils.extend_variables( def_variables, test_dict.get("variables", {}) ) # merge & override validators TODO: relocate def_raw_validators = api_def_dict.pop("validate", []) def_validators = [ validator.uniform_validator(_validator) for _validator in def_raw_validators ] ref_validators = test_dict.pop("validate", []) test_dict["validate"] = validator.extend_validators( def_validators, ref_validators ) # merge & override extractors def_extrators = api_def_dict.pop("extract", {}) test_dict["extract"] = utils.extend_variables( def_extrators, test_dict.get("extract", {}) ) # merge & override request test_dict["request"] = api_def_dict.pop("request", {}) # base_url & verify: priority api_def_dict > test_dict if api_def_dict.get("base_url"): test_dict["base_url"] = api_def_dict["base_url"] if "verify" in api_def_dict: test_dict["request"]["verify"] = api_def_dict["verify"] # merge & override setup_hooks def_setup_hooks = api_def_dict.pop("setup_hooks", []) ref_setup_hooks = test_dict.get("setup_hooks", []) extended_setup_hooks = list(set(def_setup_hooks + ref_setup_hooks)) if extended_setup_hooks: test_dict["setup_hooks"] = extended_setup_hooks # merge & override teardown_hooks def_teardown_hooks = api_def_dict.pop("teardown_hooks", []) ref_teardown_hooks = test_dict.get("teardown_hooks", []) extended_teardown_hooks = list(set(def_teardown_hooks + ref_teardown_hooks)) if extended_teardown_hooks: test_dict["teardown_hooks"] = extended_teardown_hooks # TODO: extend with other api definition items, e.g. times test_dict.update(api_def_dict) def _extend_with_testcase(test_dict, testcase_def_dict): """ extend test with testcase definition test will merge and override testcase config definition. Args: test_dict (dict): test block testcase_def_dict (dict): testcase definition Returns: dict: extended test dict. """ # override testcase config variables testcase_def_dict["config"].setdefault("variables", {}) testcase_def_variables = utils.ensure_mapping_format(testcase_def_dict["config"].get("variables", {})) testcase_def_variables.update(test_dict.pop("variables", {})) testcase_def_dict["config"]["variables"] = testcase_def_variables # override base_url, verify # priority: testcase config > testsuite tests test_base_url = test_dict.pop("base_url", "") if not testcase_def_dict["config"].get("base_url"): testcase_def_dict["config"]["base_url"] = test_base_url # override name test_name = test_dict.pop("name") or testcase_def_dict["config"].pop("name") or "Undefined name" # override testcase config name, output, etc. testcase_def_dict["config"].update(test_dict) testcase_def_dict["config"]["name"] = test_name test_dict.clear() test_dict.update(testcase_def_dict) def __prepare_config(config, project_mapping): """ parse testcase/testsuite config. """ # get config variables raw_config_variables = config.pop("variables", {}) raw_config_variables_mapping = utils.ensure_mapping_format(raw_config_variables) override_variables = utils.deepcopy_dict(project_mapping.get("variables", {})) functions = project_mapping.get("functions", {}) # override config variables with passed in variables raw_config_variables_mapping.update(override_variables) if raw_config_variables_mapping: config["variables"] = raw_config_variables_mapping check_variables_set = raw_config_variables_mapping.keys() prepared_config = prepare_lazy_data(config, functions, check_variables_set, cached=True) return prepared_config def __prepare_testcase_tests(tests, config, project_mapping): """ override tests with testcase config variables, base_url and verify. test maybe nested testcase. variables priority: testcase config > testcase test > testcase_def config > testcase_def test > api base_url priority: testcase test > testcase config > testsuite test > testsuite config > api verify priority: testcase teststep (api) > testcase config > testsuite config Args: tests (list): config (dict): project_mapping (dict): """ config_variables = config.get("variables", {}) config_base_url = config.get("base_url", "") config_verify = config.get("verify", True) functions = project_mapping.get("functions", {}) prepared_testcase_tests = [] session_variables = {} for test_dict in tests: # 1, testcase config => testcase tests # override test_dict variables test_dict_variables = utils.extend_variables( test_dict.pop("variables", {}), config_variables ) test_dict["variables"] = test_dict_variables # base_url & verify: priority test_dict > config if (not test_dict.get("base_url")) and config_base_url: test_dict["base_url"] = config_base_url # unify validators' format if "validate" in test_dict: ref_raw_validators = test_dict.pop("validate", []) test_dict["validate"] = [ validator.uniform_validator(_validator) for _validator in ref_raw_validators ] if "testcase_def" in test_dict: # test_dict is nested testcase # 2, testcase test_dict => testcase_def config testcase_def = test_dict.pop("testcase_def") _extend_with_testcase(test_dict, testcase_def) # verify priority: nested testcase config > testcase config test_dict["config"].setdefault("verify", config_verify) # 3, testcase_def config => testcase_def test_dict test_dict = _parse_testcase(test_dict, project_mapping) elif "api_def" in test_dict: # test_dict has API reference # 2, test_dict => api api_def_dict = test_dict.pop("api_def") _extend_with_api(test_dict, api_def_dict) # verify priority: testcase teststep > testcase config if "request" in test_dict and "verify" not in test_dict["request"]: test_dict["request"]["verify"] = config_verify # move extracted variable to session variables if "extract" in test_dict: extract_mapping = utils.ensure_mapping_format(test_dict["extract"]) session_variables.update(extract_mapping) check_variables_set = set(test_dict_variables.keys()) \ | set(session_variables.keys()) | {"request", "response"} # convert validators to lazy function validators = test_dict.pop("validate", []) prepared_validators = [] for _validator in validators: function_meta = { "func_name": _validator["comparator"], "args": [ _validator["check"], _validator["expect"] ], "kwargs": {} } prepared_validators.append( LazyFunction( function_meta, functions, check_variables_set ) ) test_dict["validate"] = prepared_validators # convert variables and functions to lazy object. # raises VariableNotFound if undefined variable exists in test_dict prepared_test_dict = prepare_lazy_data( test_dict, functions, check_variables_set ) prepared_testcase_tests.append(prepared_test_dict) return prepared_testcase_tests def _parse_testcase(testcase, project_mapping): """ parse testcase Args: testcase (dict): { "config": {}, "teststeps": [] } """ testcase.setdefault("config", {}) prepared_config = __prepare_config( testcase["config"], project_mapping ) prepared_testcase_tests = __prepare_testcase_tests( testcase["teststeps"], prepared_config, project_mapping ) return { "config": prepared_config, "teststeps": prepared_testcase_tests } def __get_parsed_testsuite_testcases(testcases, testsuite_config, project_mapping): """ override testscases with testsuite config variables, base_url and verify. variables priority: parameters > testsuite config > testcase config > testcase_def config > testcase_def tests > api base_url priority: testcase_def tests > testcase_def config > testcase config > testsuite config Args: testcases (dict): { "testcase1 name": { "testcase": "testcases/create_and_check.yml", "weight": 2, "variables": { "uid": 1000 }, "parameters": { "uid": [100, 101, 102] }, "testcase_def": { "config": {}, "teststeps": [] } }, "testcase2 name": {} } testsuite_config (dict): { "name": "testsuite name", "variables": { "device_sn": "${gen_random_string(15)}" }, "base_url": "http://127.0.0.1:5000" } project_mapping (dict): { "env": {}, "functions": {} } """ testsuite_base_url = testsuite_config.get("base_url") testsuite_config_variables = testsuite_config.get("variables", {}) functions = project_mapping.get("functions", {}) parsed_testcase_list = [] for testcase_name, testcase in testcases.items(): parsed_testcase = testcase.pop("testcase_def") parsed_testcase.setdefault("config", {}) parsed_testcase["path"] = testcase["testcase"] parsed_testcase["config"]["name"] = testcase_name if "weight" in testcase: parsed_testcase["config"]["weight"] = testcase["weight"] # base_url priority: testcase config > testsuite config parsed_testcase["config"].setdefault("base_url", testsuite_base_url) # 1, testsuite config => testcase config # override test_dict variables testcase_config_variables = utils.extend_variables( testcase.pop("variables", {}), testsuite_config_variables ) # 2, testcase config > testcase_def config # override testcase_def config variables overrided_testcase_config_variables = utils.extend_variables( parsed_testcase["config"].pop("variables", {}), testcase_config_variables ) if overrided_testcase_config_variables: parsed_testcase["config"]["variables"] = overrided_testcase_config_variables # parse config variables parsed_config_variables = parse_variables_mapping( overrided_testcase_config_variables, functions) # parse parameters if "parameters" in testcase and testcase["parameters"]: cartesian_product_parameters = parse_parameters( testcase["parameters"], parsed_config_variables, functions ) for parameter_variables in cartesian_product_parameters: # deepcopy to avoid influence between parameters testcase_copied = utils.deepcopy_dict(parsed_testcase) parsed_config_variables_copied = utils.deepcopy_dict(parsed_config_variables) testcase_copied["config"]["variables"] = utils.extend_variables( parsed_config_variables_copied, parameter_variables ) parsed_testcase_copied = _parse_testcase(testcase_copied, project_mapping) parsed_testcase_copied["config"]["name"] = parse_lazy_data( parsed_testcase_copied["config"]["name"], testcase_copied["config"]["variables"] ) parsed_testcase_list.append(parsed_testcase_copied) else: parsed_testcase = _parse_testcase(parsed_testcase, project_mapping) parsed_testcase_list.append(parsed_testcase) return parsed_testcase_list def _parse_testsuite(testsuite, project_mapping): testsuite.setdefault("config", {}) prepared_config = __prepare_config(testsuite["config"], project_mapping) parsed_testcase_list = __get_parsed_testsuite_testcases( testsuite["testcases"], prepared_config, project_mapping ) return parsed_testcase_list def parse_tests(tests_mapping): """ parse tests and load to parsed testcases tests include api, testcases and testsuites. Args: tests_mapping (dict): project info and testcases list. { "project_mapping": { "PWD": "XXXXX", "functions": {}, "variables": {}, # optional, priority 1 "env": {} }, "testsuites": [ { # testsuite data structure "config": {}, "testcases": { "testcase1 name": { "variables": { "uid": 1000 }, "parameters": { "uid": [100, 101, 102] }, "testcase_def": { "config": {}, "teststeps": [] } }, "testcase2 name": {} } } ], "testcases": [ { # testcase data structure "config": { "name": "desc1", "path": "testcase1_path", "variables": {}, # optional, priority 2 }, "teststeps": [ # test data structure { 'name': 'test step desc1', 'variables': [], # optional, priority 3 'extract': [], 'validate': [], 'api_def': { "variables": {} # optional, priority 4 'request': {}, } }, test_dict_2 # another test dict ] }, testcase_dict_2 # another testcase dict ], "api": { "variables": {}, "request": {} } } """ project_mapping = tests_mapping.get("project_mapping", {}) parsed_tests_mapping = { "project_mapping": project_mapping, "testcases": [] } for test_type in tests_mapping: if test_type == "testsuites": # load testcases of testsuite testsuites = tests_mapping["testsuites"] for testsuite in testsuites: parsed_testcases = _parse_testsuite(testsuite, project_mapping) for parsed_testcase in parsed_testcases: parsed_tests_mapping["testcases"].append(parsed_testcase) elif test_type == "testcases": for testcase in tests_mapping["testcases"]: parsed_testcase = _parse_testcase(testcase, project_mapping) parsed_tests_mapping["testcases"].append(parsed_testcase) elif test_type == "apis": # encapsulate api as a testcase for api_content in tests_mapping["apis"]: testcase = { "teststeps": [api_content] } parsed_testcase = _parse_testcase(testcase, project_mapping) parsed_tests_mapping["testcases"].append(parsed_testcase) return parsed_tests_mapping