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軟件測試小棧