python接口自動化24-有token的接口項目使用unittest框架設計

時間 2019-11-19

標籤 python 接口自動化 token 項目使用 unittest 框架設計欄目 Python 简体版

原文原文鏈接

獲取token

在作接口自動化的時候，常常會遇到多個用例須要用同一個參數token，而且這些測試用例跨.py腳本了。
通常token只須要獲取一次就好了，而後其它使用unittest框架的測試用例所有調用這個參數，那麼如何實現呢？
雖然python裏面有個全局變量global，但這個只是針對於在同一個.py裏纔有效，跨腳本就不起做用了。html

解決思路

1.首先把公共數據單獨抽出來，用一個文件去管理，如yaml文件
2.寫一個讀yaml文件的get_token()函數放到a.py，去讀取須要的數據
3.其它腳本所有調用a.py裏面的get_token()函數，而後傳參
4.token動態獲取能夠寫個登陸函數放到run.py，獲取到以後把token值寫入到yaml文件，這樣保證每次token都是最新的
5.run.py裏面在運行全部用例以前先登陸一次並寫入token到yaml,而後執行全部用例，出報告結果

設計結構以下python

yaml配置

1.公共參數咱們能夠寫入到一個單獨的配置文件裏，這裏配置文件我用的yaml來管理數據框架

讀取token

1.寫個讀取yaml文件裏面的token值的函數，放到re_token.py,代碼以下：函數

# coding:utf-8
import yaml
import os
# 做者：上海-悠悠 QQ交流羣：588402570

cur = os.path.dirname(os.path.realpath(__file__))

def get_token(yamlName="token.yaml"):
    '''
    從token.yaml讀取token值
    :param yamlName: 配置文件名稱
    :return: token值
    '''
    p = os.path.join(cur, yamlName)
    f = open(p)
    a = f.read()
    t = yaml.load(a)
    f.close()
    return t["token"]

if __name__ =="__main__":
    print(get_token())

測試用例token參數關聯

1.常常用人問token如何參數關聯，問這種問題的，基本上沒認真學過python，估計日常寫的代碼都是東拼西湊處處複製過來的。測試

2.token的參數關聯用一箇中間變量【token】去接收上一步【get_token()函數)】獲取的token值就好了,再傳入到須要的地方，so easy!spa

3.如下代碼是test_01和test_02裏面寫入的演示案例設計

# coding:utf-8
import unittest
from common.re_token import get_token
# 做者：上海-悠悠 QQ交流羣：588402570

class Test_01(unittest.TestCase):


    @classmethod
    def setUpClass(cls):
        cls.token = get_token()
        print("獲取到當前用例token值：%s" % cls.token)

    def test_01(self):
        '''測試用例1'''
        body1 = {
            "a": "111111",
            "b": "111111",
            "token": self.token  # 參數關聯
        }
        print("用例1body：%s" % body1)

    def test_02(self):
        '''測試用例2'''
        body1 = {
            "a": "222222",
            "b": "2222222",
            "token": self.token  # 參數關聯
        }
        print("用例2body：%s" % body1)

if __name__ == "__main__":
    unittest.main()

4.執行結果：3d

..
獲取到當前用例token值：yoyoketang
用例1body：{'token': 'yoyoketang', 'a': '111111', 'b': '111111'}
用例2body：{'token': 'yoyoketang', 'a': '222222', 'b': '2222222'}
----------------------------------------------------------------------
Ran 2 tests in 0.000s
OK

執行策略

1.單個腳本運行沒問題了，那麼若是一次性執行多個用例腳本，都調用同一個token值呢？run.py腳本參考以下code

# coding=utf-8
import unittest
import os
import yaml
from common import HTMLTestRunner_cn
from ruamel import yaml

curpath = os.path.dirname(os.path.realpath(__file__))

# 做者：上海-悠悠 QQ交流羣：588402570

def login(user="yoyo", psw="123456"):
    '''
    先執行登陸，傳帳號和密碼兩個參數
    :return: 返回token值
    '''
    print("登陸的帳號名稱：%s" % user)
    p = psw
    print("輸入的密碼：**********")
    token = "xxxxxxxxx"     # 登陸後獲取到的token值
    return token

def write_yaml(value):
    '''
    把獲取到的token值寫入到yaml文件
    :param value:
    :return:
    '''
    ypath = os.path.join(curpath, "common", "token.yaml")
    print(ypath)
    # 需寫入的內容
    t = {"token": value}
    # 寫入到yaml文件
    with open(ypath, "w", encoding="utf-8") as f:
        yaml.dump(t, f, Dumper=yaml.RoundTripDumper)

def all_case(rule="test*.py"):
    '''加載全部的測試用例'''
    case_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "case")
    # 定義discover方法的參數
    discover = unittest.defaultTestLoader.discover(case_path,
                                                  pattern=rule,
                                                  top_level_dir=None)
    return discover


def run_case(all_case, reportName="report"):
    '''
    執行全部的用例, 並把結果寫入HTML測試報告
    '''
    curpath = os.path.dirname(os.path.realpath(__file__))
    report_path = os.path.join(curpath, reportName)  # 用例文件夾
    # 若是不存在這個report文件夾，就自動建立一個
    if not os.path.exists(report_path):os.mkdir(report_path)
    report_abspath = os.path.join(report_path, "result.html")
    print("report path:%s"%report_abspath)
    fp = open(report_abspath, "wb")
    runner = HTMLTestRunner_cn.HTMLTestRunner( stream=fp,
                                               verbosity=2,
                                               title=u'自動化測試報告,測試結果以下：',
                                               description=u'用例執行狀況：')

    # 調用add_case函數返回值
    runner.run(all_case)
    fp.close()

if __name__ == "__main__":
    token = login("admin", "111111")  # 1.登陸
    write_yaml(token)                   # 2.寫入yaml
    allcases = all_case()                  # 3.加載用例
    run_case(allcases)

2.執行完以後再report文件下生產一個測試報告htm