pytest+yaml實現接口自動化

點擊上方藍字關注我！

最近在學pytest的使用，不一樣於httprunner自動化測試框架，建立項目的時候就能把全部的項目文件目錄整理出來，須要手動建立文件分類，無論怎麼樣，先學會用，在去了解它的底層原理。

這篇文章我把pytest+yaml+參數化一塊兒講解了。

一個簡單的接口案例

# def test_movie_check():# url = 'http://api.douban.com/v2/movie/in_theaters'# params = {# 'apikey': '0df993c66c0c636e29ecbb5344252a4a',# 'start': 0,# 'count': 10# }# headers = {# "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_14_5) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/80.0.3987.149 Safari/537.36"# }# r = requests.get(url=url, params=params, headers=headers)# response = r.json()# assert response['count'] == params['count']# assert response['start'] == params['start']# assert response['title'] == '正在上映的電影-北京'

以上就是一個簡單的python自帶的requests的請求接口，並經過assert作斷言，判斷腳本運行是否正確。

pytest+yaml

看過我httprunner的應該都知道，httprunner主要就是經過yaml文件來進行接口調用的，原來pytest也能調用yaml文件，這裏咱們把yaml文件做爲咱們的接口數據準備文件，以下代碼所示：

yaml方法的使用：

       ◆   yaml.load() : 讀取yaml文件;

       ◆   yaml.load_all() :  生成迭代器，若存在多塊yaml文檔的時候，使用解析所有文檔;

       ◆  yaml.dump(params1,params2) : yaml.dump將python對象生成yaml文檔 params1 json格式的文檔 ,params2 打卡一個文本或二進制文件，將生成的yaml文件寫到文件中;

       ◆  yaml.dump_all() 將多個段輸出到一個文件;

第一步：編寫yaml文件，準備測試接口數據

test: - case: 驗證響應參數是否一致 requests: method: GET path: /v2/movie/in_theaters headers: User-Agent: Mozilla/5.0 (Macintosh; Intel Mac OS X 10_14_5) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/80.0.3987.149 Safari/537.36 params: apikey: 0df993c66c0c636e29ecbb5344252a4a start: 0 count: 10 validate: title: 正在上映的電影-北京 count: 10 start: 0

第二步：編寫test_movie.py文件，get_test_data方法返回測試數據

python的zip用法：

☆  將可迭代的對象做爲參數，打包成元祖，而後返回元祖組成的對象

☆ 若每一個迭代器的元素個數不同，返回列表長度最短的對象

☆ 利用zip(*)將元祖解壓成列表

☆ 經過list（zipped）來打印出壓縮結果

☆ 好處：節省內存

def get_test_data(filepath): case = [] # 存放測試用例名 requests = [] # 存放http請求對象    validate = []  # 存放斷言    with open(filepath, encoding='utf-8') as f: data = yaml.load(f.read(), Loader=yaml.SafeLoader) case.append(data.get('test')) requests.append(data.get('requests')) validate.append(data.get('validate')) params = zip(case, requests, validate) return case, paramscases, parameters = comm_function.get_test_data('D:/workFile/test/pytest_demo/data/test_movie.yaml')list_params = list(parameters)print(list_params[0][1])

第三步：判斷測試案例是否經過

@pytest.mark.parametrize 參數化，若yaml中有多條案例，會執行屢次。

class TestInTheaters(object): @pytest.mark.parametrize('case,http,validate', list(list_params), ids=cases) def test_in_theaters(self, case, http, validate): host = 'http://api.douban.com' res = requests.request(http['method'], url=host+http['path'], headers=http['headers'], params=http['params']) response = res.json() assert response['count'] == validate['count'] assert response['start'] == validate['start'] assert response['title'] == validate['title']

結果以下：

歡迎小夥伴們進羣一塊兒學習，一塊兒加油！

END

本文分享自微信公衆號 - 軟件測試小助手（gh_2282fef3410c）。
若有侵權，請聯繫 support@oschina.cn 刪除。
本文參與「OSC源創計劃」，歡迎正在閱讀的你也加入，一塊兒分享。