Python接口自动化之动态数据处理
本文3575字，阅读约需10分钟
在上一篇Python接口自动化测试系列文章：Python接口自动化之接口依赖，主要介绍如何提取token、将token作为类属性全局调用及充值接口如何携带token进行请求。
以下主要介绍：接口自动化过程中，动态数据如何生成、动态数据与数据库数据进行对比并替换。
一
应用场景
- 注册接口参数需要手机号，手机号如何动态生成？
- 生成的手机号如何与数据库数据进行对比？
- 未注册的手机号如何替换用例数据中的手机号？
二
动态手机号处理思路
①编写函数，生成随机的手机号；
②将生成的手机号进行数据库查询；
③如手机号已存在，就重新生成手机号；
④如手机号不存在，就将此手机号替换测试用例中的手机号。
三
动态手机号处理
1
注册接口测试用例
在data/cases.xlsx中，新建register工作簿，填充注册接口用例，其中mobile_phone是动态参数，如果写死，在自动化过程中，会运行失败，所以这里用#new_phone#表示。
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()
今日问题：
在接口自动化过程中，你是如何处理动态数据的？
（欢迎在下方留言区发表你的看法）