httprunner/ate/testcase.py

from ate import utils

class TestcaseParser(object):

    def __init__(self, variables_binds={}):
        self.variables_binds = variables_binds

    def update_variables_binds(self, variables_mapping):
        """ update variables binds with new mapping.
        """
        if variables_mapping:
            self.variables_binds.update(variables_mapping)

    def parse(self, testcase_template):
        """ parse testcase_template, replace all variables with bind value.
        variables marker: ${variable}.
        @param testcase_template
            "request": {
                "url": "http://127.0.0.1:5000/api/users/${uid}",
                "method": "POST",
                "headers": {
                    "Content-Type": "application/json",
                    "authorization": "${authorization}",
                    "random": "${random}"
                },
                "body": "${json}"
            },
            "response": {
                "status_code": "${expected_status}",
                "headers": {
                    "Content-Type": "application/json"
                },
                "body": {
                    "success": True,
                    "msg": "user created successfully."
                }
            }
        """
        return self.substitute(testcase_template)

    def substitute(self, content):
        """ substitute content recursively, each variable will be replaced with bind value.
            variables marker: ${variable}.
        """
        if isinstance(content, str):
            return utils.parse_content_with_variables(content, self.variables_binds)

        if isinstance(content, list):
            return [self.substitute(item) for item in content]

        if isinstance(content, dict):
            parsed_content = {}
            for key, value in content.items():
                parsed_content[key] = self.substitute(value)

            return parsed_content

        return content