Python接口自动化之动态数据处理

在上一篇Python接口自动化测试系列文章：Python接口自动化之logging封装及实战，

主要介绍将logging常用配置放入yaml配置文件、logging日志封装及结合登录用例讲解日志如何在接口测试中运用。

 

以下主要介绍如何提取token、将token作为类属性全局调用及充值接口如何携带token进行请求。

 

一 场景说明

 

在面试接口自动化时，经常会问，其他接口调用的前提条件是当前用户必须是登录状态，如何处理接口依赖？

 

在此之前我们介绍过session管理器保存会话状态。如果接口请求需要携带token，那么又如果处理呢？以下详细介绍。

 

未登录状态下，直接请求充值接口的异常场景：

import unittest import requests class TestRecharge(unittest.TestCase):  def setUp(self):  pass  def tearDown(self):  pass  def test_recharge(self):  url = 'http://127.0.0.1:8888/recharge'  payload = {  "member_id": 100004312,  "amount": 80  }  res = requests.post(url,json=payload)  print(res.json()) if __name__ == '__main__':  unittest.main()

运行结果如下：

{'code': 1003, 'msg': '未授权或token已过期'}

 

 

 

二 token处理思路

 

在处理之前，token灵魂三连问：

• 如何获取token？
• 获取的token如何管理？
• 其他接口如何携带token？

思路如下：

1.抽取登录接口返回值中的token；

2.使用全局变量存储token。token可以存到yaml或者json或者ini的配置文件里，以下介绍将token作为类属性；

3.其他接口将token值放入请求头，发送请求；

 

 

三 JsonPath基本用法

 

我们需要从登录接口中提取出token，这里介绍JsonPath基本用法。

 

1 JsonPath介绍

JsonPath为Json文档提供了解析能力，通过使用JsonPath，你可以方便的查找节点、获取想要的数据，JsonPath是Json版的XPath。

 

 

2 JsonPath语法

 

JsonPath	说明
$	文档根元素
@	当前元素
.或[]	匹配下级元素
..	递归匹配所有子元素
*	通配符，匹配下级元素
[]	下标运算符，根据索引获取元素，JsonPath索引从0开始
[,]	连接操作符，将多个结果拼接成数组返回，可以使用索引或别名
[start:end:step]	数据切片操作
?()	过滤表达式

 

这么多语法规范，大家会觉得一脸懵逼吧，细心的小伙伴会发现语法中 .. 表示递归匹配所有子元素，简单粗暴，用..获取所有匹配的子元素，通过索引拿到想要的值，以下以实战进行演练。

 

 

3 JsonPath源码

def jsonpath(obj, expr, result_type='VALUE', debug=0, use_eval=True):  """traverse JSON object using jsonpath expr, returning values or paths"""

 

obj: 需要解析的对象，比如接口的返回值。expr: 匹配的表达式。

 

 

4 JsonPath实战

以下编写代码进行演示。

from jsonpath import jsonpath # json格式结果 res = {  "code": 0,  "msg": "OK",  "data": {  "token_info": {  "token_type": "Bearer",  "expires_in": "2020-05-28 13:07:29",  "token": "eyJhbGciOiJIUzUxAiJ9.eyJtZW1iZZJfaWQiOjEwMDA1MTMxNSwiZXhwIjoxNTkwNjQyNDQ5fQ.s6A7pzLILf9tqpEDAU9wIPGGLGkgKEQ6EBHq26l-eUM5seCb48DWalkE7u16iRZv3uzD5hIDFbw41Jmi9V0T_Q"  }  } } # 获取res中的token token = jsonpath(res,'$..token')[0] print(token)

响应结果为：

eyJhbGciOiJIUzUxBiJ9.eyJtZW1iZZJfaWQiOpEwMDA1MTMxNSwiZXhwIjoxNTkwNjQyNDQ5fQ.s6A7pzLILf9tqpEDAU9wIPGGLGkgKEQ6EBHq26l-eUM5seCb48SWalkE8u16iRZv3uzD5hIDFbw41Jmi9V0T_Q

 

 

四 token依赖实例

 

①在config/setting.py配置文件中，配置全局的域名或IP。

class DevConfig():  # 项目的域名或IP  host='http://127.0.0.1:8888' config = DevConfig()

 

②在config/config.yaml文件中，存入登录账号。

config.yaml

user:  mobile_phone: '155********'  pwd: '12345678'

③在项目根目录下，新建文件夹middleware，文件夹下新建文件helper.py，用于处理token。

 

④首先编写登录接口，返回接口数据。

helper.py

from common.requests_handler import RequestsHandler from config.setting import config from common.yaml_handler import yaml_data def login():  """登录，接口返回token"""  req = RequestsHandler()  res = req.visit('post', config.host+'/login', json=yaml_data['user'])  return res print(login())

运行结果如下：

{  'code': 0,  'msg': 'OK',  'data': {  'token_info': {  'token_type': 'Bearer',  'expires_in': '2020-05-26 13:31:41',  'token': 'eyJhbGciOiJIUpUxMiJ9.eyJtZW1iZXJfaWQiOjEwMDA1MTMxNSwiZXhwIjoxNTkwNDcxMTAxfQ.4QDNdQ-Wyecvpv7yRnMlCx1ZXpAlb8vpohxHht5OfNcjGn_yTaISyhmn7OMzjMGlz68sJ_uFbCh2NhIaO2p_Lg'  }  } }

 

⑤接下来编写函数，获取登录接口返回值并提取token和token_type进行拼接 。这里要使用jsonpath模块提取返回的token和token_type。

helper.py

def save_token():  """保存token信息"""  res = login()  token = jsonpath(res,'$..token')[0]  token_type = jsonpath(res,'$..token_type')[0]  token = " ".join([token_type, token])  return token print(save_token())

运行结果如下：

Bearer eyJhbGciOJIPzUxMiJ9.eyJtZW1iZXJfaWQiOjEwMDA1MTMxNSwiZXhwIjoxNTkwN

 

⑥最后定义类，将token作为类属性，整体代码如下：helper.py

from common.requests_handler import RequestsHandler from config.setting import config from common.yaml_handler import yaml_data from jsonpath import jsonpath def login():  """登录，返回token信息"""  req = RequestsHandler()  res = req.visit('post', config.host+'/login', json=yaml_data['user'])  return res def save_token():  """保存token信息"""  res = login()  token = jsonpath(res,'$..token')[0]  token_type = jsonpath(res,'$..token_type')[0]  token = " ".join([token_type, token])  Context.token = token  return token class Context:  """将token作为类属性"""  token = '' if __name__ == '__main__':  print(save_token())

 

运行结果如下：

Bearer eyJhbGciOiJIUzUxMiJ9.eyJtZW1iZXJfaWQiOjEwMDA1MTMxNSwiZXhwIjoxNTkwNDcz

 

⑦充值接口携带token请求。在上一篇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()

 

运行结果为：

{  'code': 0,  'msg': 'OK',  'data': {  'leave_amount': 240.0,  'mobile_phone': '155********',  } }

 

总结：本文主要介绍如何提取token、将token作为类属性全局调用及充值接口如何携带token进行请求。

 

更多系列文章，可以关注微信公众号：ITester软件测试小栈