python+pytest介面自動化引數關聯

2022-06-23 18:00:17

前言

今天呢，筆者想和大家來聊聊python+pytest介面自動化測試的引數關聯，筆者這邊就不多說廢話了，咱們直接進入正題。

一、什麼是引數關聯？

引數關聯，也叫介面關聯，即介面之間存在引數的聯絡或依賴。在完成某一功能業務時，有時需要按順序請求多個介面，此時在某些介面之間可能會存在關聯關係。比如：B介面的某個或某些請求引數是通過呼叫A介面獲取的，即需要先請求A介面，從A介面的返回資料中拿到需要的欄位值，在請求B介面時作為請求引數傳入。

二、有哪些場景？

最常見的場景之一便是：請求登入介面後獲取到token值，後續其他介面請求時需要將token作為請求引數傳入。

再比如下單-->支付場景，呼叫下單介面生成訂單後會返回訂單號，訂單號傳給支付介面進行支付。

三、引數關聯場景

以最常見的網上購物為例，對應的場景及請求我們可以大致簡化如下（可聯想某寶購物流程）：

使用者在購物車選擇商品點選【去結算】進入訂單確認頁，訂單確認頁點選【提交訂單】這時會先請求下單介面建立訂單
緊接著會拿建立的訂單去請求獲取支付憑證介面，這個介面會調起支付頁面，也就是輸入密碼的支付介面
輸入支付密碼後會請求支付服務的支付介面進行實際支付，支付的結果會返回給請求方，告知是否支付成功

這個流程中涉及到的介面其實都是有關聯的，我們要對整個流程進行測試那麼就需要按順序呼叫所有這些涉及到的介面。

但這裡我們只需要弄明白引數關聯，那麼以下單介面和獲取支付憑證介面為例，舉例說明就夠了，即先請求下單介面生成訂單號，再拿這個訂單號去請求獲取支付憑證介面，才能調起支付介面並進行支付。

下單介面如下：

介面地址：<伺服器>/trade/order/purchase
請求型別：post

請求引數：

{
    "goodsId": 10,  //商品id
    "goodsSkuId": 33,   //sku id
    "num": 2,   //購買數量
    "tradePromotion": { //選擇的優惠項
        "type": 1,  //型別<1：優惠券>
        "promotionId": 1    //優惠id
    }
}

返回值data：

{
    "code": 0,
    "msg": "成功",
    "data": {
        "tradeNo": "0020220116204344962706666"  //交易訂單號
    },
    "t": 1639658625474
}

獲取支付憑證介面如下：

介面地址：<伺服器>/pay/pre/consum
請求型別：post

請求引數：

{
    "orderNo":"0020220116204344962706666",    //交易訂單號
    "product":"alipayWapClient"    //支付渠道<alipayWapClient：支付寶手機網頁支付>
}

返回值 data:

{
    "code": 0,
    "msg": "成功",
    "data": {
        "payNo":"123213213219379213",
        "certificate": "<form name="punchout_form" method="post" action="https://openapi.alipay.com/gateway.do?charset=UTF-8&method=alipay.trade.wap.pay&sign=aTKlfEnYgR7x9xs1Eyjipo0S%2BFtQ6AKu9d%2Brb7iieMThz2Dq7zp4h8QH4lelTKovOloT%2FPiNXR5miwKgOWW3K6pl0TFO5XX5NSZNBmU%2BPd5ogeo0YT0vCqWUM60rqbYLNtZmvyx%2BAR4Z2SOnqs0CYqVIbZAhpn1Bd5HsdcCCYVgsgOdbEE60Cfu3LG3Z%2FQ0GQIdW24uTyr%2BojRc25ipOC9NIYwtba8UjRw18Z3e7sj75qoIg%2FipICH7FCJBJEdlgBGlNxIjKzhYj4OBg93D&return_url=https%3A%2F%2Fblog.csdn.net%2Fa032788aotify_url=http%3A%2F%2F82.157.145.132%3A8089%2Ftest%2Fnotify%2Fgateway&version=1.0&app_id=2021001105644746&sign_type=RSA2&timestamp=2021-150&alipay_sdk=alipay-sdk-java-4.9.5.ALL&format=json">n<input type="hidden" name="biz_content" value="{&quot;time_expire&quot;:&quot;2022-12-31 22:00:00&quot;,&quot;out_trade_no&quot;:&quot;123213213219379213&quot;,&quot;total_amount&quot;:0.01,&quot;subject&quot;:&quot;測試商品&quot;,&quot;product_code&quot;:&quot;QUICK_WAP_WAY&quot;}">n<input type="submit" value="立即支付" style="display:none" >n</form>n<script>document.forms[0].submit();</script>"
    },
    "t": 1639659171031
}

其中orderNo欄位使這兩個介面關聯了起來。因為每次生成的訂單號都不一樣，所以在測試這個場景的時候，就需要使這兩個介面的引數進行關聯，才能走通。

四、指令碼編寫

那麼在pytest框架的自動化測試中，引數關聯可以怎樣處理呢？這裡提供兩種思路，如下：

根據業務場景的呼叫時序，在用例中按順序呼叫介面
將依賴的介面編寫成fixture函數，並使用yield返回下個介面需要的引數

1、在用例中按順序呼叫

程式碼範例如下：

import requests
import json
import pytest
 
def test_order_pay():
    '''
    建立訂單->獲取支付憑證，調起支付介面
    :return:
    '''
    # 先呼叫下單介面生成訂單
    url_order = "https://gouwu.com/trade/order/purchase"
    data_order = {
        "goodsId": 10,
        "goodsSkuId": 33,
        "num": 2,
        "tradePromotion": {
            "type": 1,
            "promotionId": 1
        },
        "tradeDirectionArticle": {
            "articleId": 1
        }
    }
    res_order = requests.post(url=url_order, json=data_order).text
    tradeNo = json.loads(res_order)["tradeNo"]
 
    # 再請求獲取支付憑證介面
    url_pay = "https://gouwu.com/pay/pre/consum"
    data_pay = {
        "orderNo": tradeNo, # tradeNo通過下單介面獲取
        "product": "alipayWapClient"
    }
    res_pay = requests.post(url=url_pay, json=data_pay).text
    res_pay = json.loads(res_pay)
    # 斷言
    assert res_pay["code"]==0
    assert res_pay["data"]["payNo"]
    assert res_pay["data"]["certificate"]
    
    
if __name__ == '__main__':
    pytest.main()

上面的程式碼只是流水式的進行呼叫，我們還可以先將每個介面請求封裝成單獨的函數，在測試用例中只需按照順序呼叫這些函數即可，這個我們會在後續的文章中進行說明。

2、使用Fixture函數

定義Fixture函數，程式碼範例如下：

@pytest.fixture()
def get_order():
    '''
    請求下單介面，建立訂單
    :return:
    '''
    url_order = "https://gouwu.com/trade/order/purchase"
    data_order = {
        "goodsId": 10,
        "goodsSkuId": 33,
        "num": 2,
        "tradePromotion": {
            "type": 1,
            "promotionId": 1
        },
        "tradeDirectionArticle": {
            "articleId": 1
        }
    }
    res_order = requests.post(url=url_order, json=data_order).text
    tradeNo = json.loads(res_order)["tradeNo"]
    yield tradeNo

在測試函數中呼叫上面定義的fixture函數，程式碼範例如下：

def test_pay(get_order):
    '''
    下單->支付場景校驗
    :param get_order: 呼叫上面的Fixture函數，函數名get_order即返回的tradeNo
    :return:
    '''
    url_pay = "https://gouwu.com/pay/pre/consum"
    data_pay = {
        "orderNo": get_order,  # get_order即為上面定義的fixture函數返回值
        "product": "alipayWapClient"
    }
    res_pay = requests.post(url=url_pay, json=data_pay).text
    res_pay = json.loads(res_pay)
    # 斷言
    assert res_pay["code"] == 0
    assert res_pay["data"]["payNo"]
    assert res_pay["data"]["certificate"]