在上一篇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請求。
import unittest import requests from middleware.helper import Context, save_token class TestRecharge(unittest.TestCase): def setUp(self): pass def tearDown(self): pass def test_recharge(self): save_token() token = Context.token url = 'http://127.0.0.1:88888/recharge' payload = { "member_id": 100051315, "amount": 80 } res = requests.post(url,json=payload) print(res.json()) if __name__ == '__main__': unittest.main()
運行結果為：
{ 'code': 0, 'msg': 'OK', 'data': { 'leave_amount': 240.0, 'mobile_phone': '155********', } }
總結：本文主要介紹如何提取token、將token作為類屬性全局調用及充值接口如何攜帶token進行請求。
更多系列文章，可以關注微信公眾號：ITester軟件測試小棧
