Python接口自動化之接口依賴

VOL 129

28

2020-05

今天距2021年216天

這是ITester軟件測試小棧第129次推文

點擊上方藍字“ITester軟件測試小棧“關注我，每周一、三、五早上 07:30准時推送。

微信公眾號后台回復“資源”、“測試工具包”領取測試資源,回復“21天打卡”一起學習成長，打怪升級。

本文5030字，閱讀約需13分鍾

在上一篇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進行請求。

以上

That‘s all

更多系列文章

敬請期待

ITester軟件測試小棧

往期內容寵幸

1.Python接口自動化-接口基礎(一)

---

2.Python接口自動化-接口基礎(二)

---

3.Python接口自動化-requests模塊之get請求

---

4.Python接口自動化-requests模塊之post請求

---

5.Python接口自動化之cookie、session應用

---

6.Python接口自動化之Token詳解及應用

---

7.Python接口自動化之requests請求封裝

---

8.Python接口自動化之pymysql數據庫操作

---

9.Python接口自動化之logging日志

---

10.Python接口自動化之logging封裝及實戰

想獲取更多最新干貨內容

快來星標 置頂 關注我

每周一、三、五 07:30見

<<  滑動查看下一張圖片  >>

今日問題：

在接口自動化中，你是如何處理接口依賴的？ 

（歡迎在下方留言區發表你的看法）

留言福利：

抽取截止至6月1日留言點贊最多的兩位幸運er，可獲得：百度網盤一月VIP會員

（幸運er名單將在下周一推文的置頂留言處公布）

 后台 回復"資源"取干貨

回復"21天打卡"一起打怪升級

測試交流Q群：727998947

點亮一下在看，你更好看