Python接口自動化之接口依賴

Posted on by

一、場景說明

在面試接口自動化時，經常會問，其他接口調用的前提條件是當前用戶必須是登錄狀態，如何處理接口依賴？

在此之前我們介紹過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進行請求。

本篇文章就到這裡瞭，希望能給你帶來幫助，也希望您能夠多多關註WalkonNet的更多內容！

推薦閱讀：