pytest 常用方法介紹

 前言

　　之前一篇文章簡單介紹了 pytest 以及 pytest.fixture 裝飾器 ：https://www.cnblogs.com/shenh/p/11572657.html 。實際在寫自動化測試腳本中，還會有一些很實用的方法，下文就來講述下這些用法。

一.pytest.mark.parametrize 裝飾器

 pytest 内置裝飾器 @pytest.mark.parametrize 可以讓測試資料參數化，把測試資料單獨管理，類似 ddt 資料驅動的作用，友善代碼和測試資料分離。

1.一次傳多個參數

import pytest

@pytest.mark.parametrize('x,y',[(1,2),(3,4)])
def test_sum(x,y):
    sum = x + y
    print(sum)

if __name__ =="__main__":
    pytest.main(['test_sample.py','-s'])      

執行結果：

test_sample.py 
3
.
7
.

============================== 2 passed in 0.06s ==============================      

2.組合傳參：

 注意：這種方式一共傳遞了4組參數  （1,3）、（1,4）、（2,3）、（2,4）。這種方式可以簡化測試資料，不用手動再将參數組合。

import pytest

@pytest.mark.parametrize('x',[1,2])
@pytest.mark.parametrize('y',[3,4])
def test_sum(x,y):
    sum = x + y
    print(sum)

if __name__ =="__main__":
    pytest.main(['test_sample.py','-s'])      

 執行結果：

test_sample.py 
4
.
5
.
5
.
6
.

============================== 4 passed in 0.14s ==============================      

二、fixture傳回值

1.擷取被調用函數傳回值

import pytest

@pytest.fixture(scope='function')
def login():
    accesstoken = '197ce8083c38467f'

    return accesstoken


def test_sum(login):
    token = login
    print(token)

if __name__ =="__main__":
    pytest.main(['test_sample.py','-s'])      

test_sample.py 
197ce8083c38467f
.

============================== 1 passed in 0.04s ==============================      

 若被調用函數傳回多個參數：

import pytest

@pytest.fixture(scope='function')
def login():
    accesstoken = '197ce8083c38467f'
    customerguid = '096799f5-e040-11e9-8c01-0242ac11000d'

    return accesstoken,customerguid


def test_sum(login):
    token = login[0]
    guid = login[1]
    print(token)
    print(guid)

if __name__ =="__main__":
    pytest.main(['test_sample.py','-s'])      

test_sample.py 
197ce8083c38467f
096799f5-e040-11e9-8c01-0242ac11000d
.

============================== 1 passed in 0.07s ==============================      

2.單個用例調用多個函數

import pytest

@pytest.fixture(scope='function')
def login():
    print('登入')

@pytest.fixture(scope='function')
def conn():
    print('連接配接資料庫')

def test_1(login,conn):
    print('測試用例1')

def test_2():
    print('測試用例2')

if __name__ =="__main__":
    pytest.main(['test_sample.py','-s'])      

test_sample.py 
登入
連接配接資料庫
測試用例1
.
測試用例2
.

============================== 2 passed in 0.05s ==============================      

三、測試用例分類

 有時候我們隻需執行部分測試用例，比如從用例集當中挑選 smoke 測試，要怎麼做呢？通過裝飾器 @pytest.mark.smoke，smoke 是可以自定義的，運作時加上指令‘-m=smoke’，pytest 就會挑選帶有裝飾器的類或函數運作。

import pytest

@pytest.fixture(scope='function')
def login():
    accesstoken = '197ce8083c38467f'
    customerguid = '096799f5-e040-11e9-8c01-0242ac11000d'

    return accesstoken,customerguid

@pytest.mark.smoke
def test_sum(login):
    token = login[0]
    guid = login[1]
    print(token)
    print(guid)

def test_2():
    print('測試用例')

if __name__ =="__main__":
    pytest.main(['test_sample.py','-s','-m=smoke'])      

test_sample.py 
197ce8083c38467f
096799f5-e040-11e9-8c01-0242ac11000d
.

======================= 1 passed, 1 deselected in 0.02s =======================