使用 Python 中 re 模块对测试用例参数化，进行搜索 search、替换 sub

　　参数化的目的：运行自动化测试用例的时候参数都不需要改变，直接使用封装好的类进行参数化，发起请求时直接使用替换后参数；

　　自动化测试用例，如果一百个接口要在Excel写100个sheet表单，每个接口有10个字段，里面有5个都可能是变化的，需要使用参数化，先试用特定的字符在用例中进行站位，在发起请求构造参数时在进行替换占位符；----------我们可以每个接口分别创建一个参数化；

一、用例中手机号的替换，以字符串中的方法，使用 replace （译：瑞破类似）进行替换

# 原始字符串：{"mobilephone": "${not_existed_tel}", "pwd":"123456"}

# json 字符串

src_str = '{"mobilephone": "${not_existed_tel}", "pwd":"123456"}'

# 替换 json 字符串中 ${not_existed_tel} 为 18845820369

print(src_str.replace("${not_existed_tel}", ""))

# 结果：{"mobilephone": "18845820369", "pwd":"123456"}

二、使用 re 模块进行替换

　　re 正则表达式，是一个查找、搜索、替换文本的一种格式语言

搜索方法一，re 模块中的 match（译：马驰）方法，match方法是从头开始匹配

import re

# 1. 创建原始字符串(待替换的字符串)

src_str4 = '{"mobilephone": "${not_existed_tel}", "pwd":"123456"}'

# 2. 定义模式字符串去进行匹配

# 模式字符串 == 模子，以这个模板去原始字符串中进行比对

# 方法一，re 正则中的 match（译：马驰）方法，match方法是从头开始匹配

# 从 { 开始，匹配不上, 那么就回复None

res = re.match("{not_existed_tel}", src_str4)

print(res)  # 结果：None

# 从 { 开始，能匹配上, 会返回Match对象，加r不需要转义

res = re.match(r'{"mobilephone"', src_str4)

print(res)      # 返回Match对象结果：<re.Match object; span=(0, 14), match='{"mobilephone"'>

# 获取匹配的结果内容 group （译：歌如破）

a = res.group()

print(a)        # 结果：{"mobilephone"

搜索方法二：search 　　（译：涩吃）方法，只匹配一次

import re

# 1. 创建原始字符串(待替换的字符串)

src_str1 = '{"mobilephone": "${not_existed_tel}", "pwd":"123456"}'

# 2. 定义模式字符串去进行匹配

# 模式字符串 == 模子，以这个模板去原始字符串中进行比对："${not_existed_tel}"

# 方法二：search （译：涩吃）方法，只匹配一次

# 如果能匹配上会返回Match对象,匹配不上会返回None

# 美元符号需要转义加 r，特殊字符都需要转义

res1 = re.search(r"\${not_existed_tel}", src_str1)

print(res1)     # Match对象结果：<re.Match object; span=(17, 35), match='${not_existed_tel}'>

# 获取匹配到的结果，group（译：格如破）方法

b = res1.group()

print(b)    # 结果：${not_existed_tel}

搜索方法三：findall （译：范德奥）方法，会匹配很多次

src_str1 = '{"mobilephone": "${not_existed_tel}", "pwd":"123456"}'

res2 = re.findall(r'o', src_str1)

print(res2)     # 直接返回列表结果：['o', 'o', 'o']

替换方法： sub　　（译：萨博）方法

import re

# 1. 创建原始字符串(待替换的字符串)

src_str2 = '{"mobilephone": "${not_existed_tel}", "pwd":"123456"}'

# 第一个参数为模式字符串：${not_existed_tel}

# 第二个参数为新的字符串：18845820369

# 第三个参数为原始的字符串

# sub 方法如果能匹配上, 那么会返回替换之后的字符串

# sub 方法如果匹配不上, 那么直接返回原始的字符串

# 使用 search 方法先搜索，搜索到再用 sub 方法进行替换；count=0 默认匹配一次

if re.search(r"\${not_existed_tel}", src_str2):     # 先搜索

    res2 = re.sub(r"\${not_existed_tel}", "", src_str2)  # 在替换

    print(res2)

else:

    print("无法匹配原始字符串")

# 替换后结果：{"mobilephone": "18845820369", "pwd":"123456"}

# 使用正则匹配：\w+ 单词字符[A-Za-z0-9_]，+ 多次匹配；通过正则在原始字符串中匹配

if re.search(r"\${\w+}", src_str2):     # 先搜索

    res2 = re.sub(r"\${not_existed_tel}", "", src_str2)  # 在替换

    print(res2)

else:

    print("无法匹配原始字符串")

# 替换后结果：{"mobilephone": "18845820369", "pwd":"123456"}

　　re 模块基本参数化封装

import re

from scripts.handle_mysql import HandleMysql        # 数据库封装

from scripts.handle_config import HandleConfig      # 配置文件封装

from scripts.constants import CONFIGS_USER_ACCOUTS_DIR      # 要读取的配置文件路径

class Context:

    """

    处理上下文参数化

    """

    not_existed_tel_pattern = r'\${not_existed_tel}'

    invest_user_tel_pattern = r'\${invest_user_tel}'  # 配置${invest_user_tel} 的正则表达式

    invest_user_pwd_pattern = r'\${invest_user_pwd}'  # 配置${invest_user_pwd} 的正则表达式

    handle_config = HandleConfig(filename=CONFIGS_USER_ACCOUTS_DIR)     # 配置文件：需要指定用户账号所在的配置文件路径，为了读取数据

    @classmethod

    def not_existed_tel_replace(cls, data):

        """

        参数化未注册手机号（调用随机生成手机号方法）----常用替换方法以下面的方式

        :param data: 发起请求时传入的data

        :return: 参数化后data 或 匹配不上不需要参数化的原始data

        """

        if re.search(cls.not_existed_tel_pattern, data):          # 使用search方法在用例中匹配，匹配不上不需要替换，返回原始字符串

            do_mysql = HandleMysql()    # 创建会话

            data = re.sub(cls.not_existed_tel_pattern,            # 使用sub方法替换

                          do_mysql.create_not_existed_mobile(),   # 随机生成一个在数据库不存在的手机号

                          data)

            do_mysql.close()            # 关闭会话

        return data

    @classmethod

    def invest_user_tel_replace(cls, data):

        """

        参数化已注册的手机号（配置文件中获取已注册手机号码）

        :param data: 发起请求时传入的data

        :return: 参数化后data 或 匹配不上不需要参数化的原始data

        """

        if re.search(cls.invest_user_tel_pattern, data):            # 使用search方法在用例中匹配，匹配不上不需要替换，返回原始字符串

            invest_user_tel = cls.handle_config.get_value("invest_user", "mobilephone")     # 配置文件中获取手机号码

            data = re.sub(cls.invest_user_tel_pattern, invest_user_tel, data)               # 使用sub方法替换

        return data

    @classmethod

    def invest_user_pwd_replace(cls, data):

        """

        参数化已注册的密码（配置文件中账号的密码）

        :param data: 发起请求时传入的data

        :return: 参数化后data 或 匹配不上不需要参数化的原始data

        """

        if re.search(cls.invest_user_pwd_pattern, data):            # 使用search方法在用例中匹配，匹配不上不需要替换，返回原始字符串

            invest_user_pwd = cls.handle_config.get_value("invest_user", "pwd")     # 配置文件中获取密码

            data = re.sub(cls.invest_user_pwd_pattern, invest_user_pwd, data)       # 使用sub方法替换

        return data

    @classmethod

    def register_parameterization(cls, data):

        """

        对某个接口批量参数化

        :param data: 发起请求时传入的data

        :return:  参数化后data 或 匹配不上不需要参数化的原始data

        """

        data = cls.not_existed_tel_replace(data)  # 参数化未注册的手机号

        data = cls.invest_user_tel_replace(data)  # 参数化已注册的手机号

        data = cls.invest_user_pwd_replace(data)  # 再参数化已注册的用户密码

        return data

if __name__ == '__main__':

    one_str = '{"mobilephone": "${not_existed_tel}", "pwd":"${invest_user_pwd}", "regname": "KeYou"}'

    two_str = '{"mobilephone": "${not_existed_tel}", "pwd": ""}'

    three_str = '{"mobilephone": "", "pwd": "123456"}'          # 不需要参数化，返回原始data

    six_str = '{"mobilephone": "${invest_user_tel}", "pwd": "123456", "regname": "KeYou"}'

    # 参数化

    print(Context.register_parameterization(one_str))

    print(Context.register_parameterization(two_str))

    print(Context.register_parameterization(three_str))

    print(Context.register_parameterization(six_str))

    # 结果

    # {"mobilephone": "13218495762", "pwd":"123456", "regname": "KeYou"}

    # {"mobilephone": "13278293460", "pwd": ""}

    # {"mobilephone": "", "pwd": "123456"}

    # {"mobilephone": "13226834715", "pwd": "123456", "regname": "KeYou"}

*******请大家尊重原创，如要转载，请注明出处：转载自：https://www.cnblogs.com/shouhu/ 谢谢！！*******

巴特西

使用 Python 中 re 模块对测试用例参数化，进行搜索 search、替换 sub

最新文章

热门文章