Python接口自动化浅析如何处理动态数据_Python

在上一篇Python接口自动化测试系列文章：Python接口自动化浅析logging封装及实战操作，主要介绍如何提取token、将token作为类属性全局调用及充值接口如何携带token进行请求。

以下主要介绍：接口自动化过程中，动态数据如何生成、动态数据与数据库数据进行对比并替换。

一、应用场景F

注册接口参数需要手机号，手机号如何动态生成？

生成的手机号如何与数据库数据进行对比？

未注册的手机号如何替换用例数据中的手机号？

二、动态手机号处理思路

编写函数，生成随机的手机号；
将生成的手机号进行数据库查询；
如手机号已存在，就重新生成手机号；
如手机号不存在，就将此手机号替换测试用例中的手机号。

三、动态手机号处理

1、注册接口测试用例

在data/cases.xlsx中，新建register工作簿，填充注册接口用例，其中mobile_phone是动态参数，

如果写死，在自动化过程中，会运行失败，所以这里用#new_phone#表示。

Python接口自动化浅析如何处理动态数据

2、动态生成手机号

在common目录下，新建文件helper.py，用于编写辅助函数，

实现特定的功能(类似于HttpRunner中的debugtalk.py)。

实现批量生成11位手机号，代码如下：

				?

									import random

									def generate_mobile():

									    """生成随机手机号"""

									    phone = "1" + random.choice(["3","5","7","8","9"])

									    for i in range(0,9):

									        num = random.randint(1,9)

									        phone += str(num)

									    return phone

									if __name__ == '__main__':

									    print(generate_mobile())

运行之后，结果为：

13889546979

上面代码生成批量手机号，比较简易，如对手机号格式要求更精确，可以自行按要求编写。

四、数据库查询并替换

1、replace()方法

描述：

replace()方法把字符串中的 old（旧字符串）替换成 new(新字符串)

replace语法：

str.replace(old, new[, max])

old -- 将被替换的字符串。

new -- 新字符串，用于替换old字符串。

max -- 可选字符串, 替换不超过 max 次

replace实战例子：

现有字符串如下：

				?

									Str = 'coco爱读书'

现在将Str中的coco改为vivi

				?

									Str = 'coco爱读书'

									print(Str.replace('coco', 'vivi'))

输出结果如下：

vivi爱读书

2、编写注册接口用例

接下来的注册接口用例代码，大多数代码其实和登录用例一样，只是新增了查询数据库操作。

大致思路如下：

从excel中读取用例数据；
判断用例数据中是否包含#new_phone#；
如包含#new_phone#，则随机生成手机号；
如随机生成的手机号在数据库中存在，则重新生成；
如随机生成的手机号在数据库中不存在，则用此手机号替换#new_phone#，进行注册。

				?

									import json

									import unittest

									from common.db_handler import DBHandler

									from common.helper import generate_mobile

									from common.logger_handler import logger

									from common.requests_handler import RequestHandler

									from common.excel_handler import ExcelHandler

									from config.setting import config

									from libs import ddt

									from middleware.yaml_handler import yaml_data

									@ddt.ddt

									class TestRegister(unittest.TestCase):

									    # 读取register sheet数据

									    excel = ExcelHandler(config.data_path)

									    data = excel.read_excel('register')

									    def setUp(self):

									        self.req = RequestHandler()

									        self.db = DBHandler(host=yaml_data['mysql']['host'], port=yaml_data['mysql']['port'],

									                            user=yaml_data['mysql']['user'], password=yaml_data['mysql']['password'],

									                            database=yaml_data['mysql']['db'], charset=yaml_data['mysql']['charset'])

									    def tearDown(self):

									        self.req.close_session()

									        self.db.close()

									    @ddt.data(*data)

									    def test_register(self,items):

									        # 判断#new_phone#是否在用例数据中

									        if "#new_phone#" in items['payload']:

									            while True:

									                # 使用自动生成手机号的函数

									                mobile = generate_mobile()

									                # 从数据库中查询此手机号是否存在

									                query_mobile = self.db.query("select * from member where mobile_phone=%s;",args=[mobile])

									                # 如果不存在，就跳出循环

									                if not query_mobile:

									                    break

									            # 将#new_phone#替换为生成的手机号        

									            items['payload'] = items['payload'].replace('#new_phone#', mobile)

									        logger.info('*'*30)

									        logger.info('测试第{}条测试用例:{}'.format(items['case_id'],items['case_title']))

									        logger.info('测试数据是:{}'.format(items))

									        # 访问注册接口，获取实际结果

									        res = self.req.visit(items['method'],config.host+items['url'],

									                             json=json.loads(items['payload']))

									        # 断言：预期结果与实际结果对比

									        try:

									            self.assertEqual(res['code'],items['expected_result'])

									            logger.info(res)

									            result = 'PASS'

									        except AssertionError as e:

									            logger.error("测试用例执行失败{}".format(e))

									            result = 'fail'

									            raise e

									        finally:

									            TestRegister.excel.write_excel(config.data_path,'register',items['case_id']+1,8,res['code'])

									            TestRegister.excel.write_excel(config.data_path,'register',items['case_id'] + 1,9, result)

									if __name__ == '__main__':

									    unittest.main()