前言

我們?cè)谧鼋涌谧詣?dòng)化測(cè)試的時(shí)候，會(huì)經(jīng)常遇到這種場(chǎng)景：接口A的返回結(jié)果中的某個(gè)字段，是接口B的某個(gè)字段的入?yún)ⅰＨ绻鞘褂胮ostman，那我們可以通過設(shè)置后置變量，然后在需要使用的地方通過{{}}的方式來進(jìn)行調(diào)用。但是如果是使用自己寫的測(cè)試框架中要如何實(shí)現(xiàn)呢？我想到的是如下三種方法。

首先說明一下，以下三種方式均是通過python + pytest來實(shí)現(xiàn)的

一、通過conftest創(chuàng)建全局變量

conftest.py文件是pytest框架中很有用的一個(gè)東西，首先看下官方文檔中的解釋：

大概意思就是說，conftest.py文件供整個(gè)用例目錄（conftest.py文件可以有多個(gè)，并且只在當(dāng)前package下生效）使用而無需導(dǎo)入，也就是說在用例目錄是conftest中的信息是公用的，例如：

a/conftest.py:
    def pytest_runtest_setup(item):
        # called for running each test in 'a' directory
        print("setting up", item)
 
a/test_sub.py:
    def test_sub():
        pass
 
test_flat.py:
    def test_flat():
        pass
 

運(yùn)行后的結(jié)果：

pytest test_flat.py --capture=no? # will not show "setting up"

pytest a/test_sub.py --capture=no? # will show "setting up"

而我們就可以通過conftest + fixture函數(shù)來實(shí)現(xiàn)我們想要的效果了，具體代碼如下：

# conftest.py
 
# 定義一個(gè)全局變量，用于存儲(chǔ)內(nèi)容
global_data = {}
 
@pytest.fixture
def set_global_data():
    """
    設(shè)置全局變量，用于關(guān)聯(lián)參數(shù)
    :return:
    """
 
    def _set_global_data(key, value):
        global_data[key] = value
 
    return _set_global_data
 
@pytest.fixture
def get_global_data():
    """
    從全局變量global_data中取值
    :return:
    """
 
    def _get_global_data(key):
        return global_data.get(key)
 
    return _get_global_data
 
 

簡單說一下實(shí)現(xiàn)邏輯：

1. 首先定義一個(gè)變量global_data用于接收存儲(chǔ)用例返回的結(jié)果
2. set_global_data和get_global_data兩個(gè)fixture方法顧名思義，set方法是往global_data中存數(shù)據(jù)，get方法是從global_data中取數(shù)據(jù)

方法實(shí)現(xiàn)了，具體應(yīng)該怎么使用呢？如下：

 
# test_get_set.py
 
import requests
import pytest
 
 
 
def test_set(set_global_data):
    res = requests.get("http://www.baidu.com")
    status_code = res.status_code
    logger.info(f"請(qǐng)求返回狀態(tài)碼：{status_code}")
    set_global_data("status_code", status_code)
 
 
def test_get(get_global_data):
    data = get_global_data("status_code")
    logger.info(f'通過get_global_data方法獲取的值：{data}')
 
 
if __name__ == '__main__':
    pytest.main(['-sv', 'test_get_set.py'])

返回結(jié)果：

test_get_set.py::test_set PASSED

2021-12-24 17:58:37.642 | INFO? ? ?| cases.test_get_set:test_set:19 - 請(qǐng)求返回狀態(tài)碼：200

2021-12-24 17:58:37.643 | INFO? ? ?| cases.test_get_set:test_get:25 - 通過get_global_data方法獲取的值：200

test_get_set.py::test_get PASSED

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

通過這種方式，便實(shí)現(xiàn)了用例間的參數(shù)傳遞問題。

在實(shí)際工作中，因?yàn)樯婕暗降慕涌?、用例?huì)很多，所以可以根據(jù)需要使用不同的conftest進(jìn)行管理。并且存儲(chǔ)的數(shù)據(jù)結(jié)構(gòu)也需要進(jìn)行規(guī)范區(qū)分，如使用方法名作為字典的key。

二、使用tmpdir_factory方法

第二種方法，是使用pytest的tmpdir和tmpdir_factory兩個(gè)夾具函數(shù)，同樣是通過conftest文件來實(shí)現(xiàn)。仍然是先來看下官方文檔針對(duì)這兩個(gè)方法的說明：

簡單來說，這兩個(gè)方法的作用就是為每個(gè)測(cè)試方法創(chuàng)建一個(gè)臨時(shí)目錄用于存儲(chǔ)自定義的文件，這個(gè)臨時(shí)目錄會(huì)默認(rèn)保存3個(gè)sessions，之后就會(huì)按照創(chuàng)建的順序刪除舊的目錄?？聪鹿俜降睦樱?/p>

# content of test_tmpdir.py
def test_create_file(tmpdir):
    p = tmpdir.mkdir("sub").join("hello.txt")
    p.write("content")
    assert p.read() == "content"
    assert len(tmpdir.listdir()) == 1
    assert 0

# contents of conftest.py
import pytest
 
 
@pytest.fixture(scope="session")
def image_file(tmpdir_factory):
    img = compute_expensive_image()
    fn = tmpdir_factory.mktemp("data").join("img.png")
    img.save(str(fn))
    return fn
 
 
# contents of test_image.py
def test_histogram(image_file):
    img = load_image(image_file)
    # compute and test histogram

我在實(shí)際項(xiàng)目中的使用：

仍是在conftest.py文件中自定義一個(gè)夾具函數(shù)，返回結(jié)果是一個(gè)元組，p是tmpdir_factory方法返回的對(duì)象，轉(zhuǎn)為字符串之后就是文件存儲(chǔ)的路徑。

自定義一個(gè)名為“apitest-tmp-dir”的文件夾用于存儲(chǔ)文件

# conftest.py
 
@pytest.fixture
def tmp_factory(tmpdir_factory):
    """
    生成臨時(shí)目錄
    """
    p = tmpdir_factory.mktemp('apitest-tmp-dir')
    logger.info("當(dāng)前臨時(shí)文件的目錄為：" + str(p))
    return p, str(p)

在測(cè)試方法中的使用

 
# test_get_set.py
 
import requests
import pytest
import json
 
 
def test_set(tmp_factory):
    res = requests.get("http://www.baidu.com")
    status_code = res.status_code
    logger.info(f"返回狀態(tài)碼：{status_code}")
 
    logger.debug(tmp_factory)
    # 創(chuàng)建test_set.txt文件
    a = tmp_factory[0].join("test_set.txt")
    # 將需要的內(nèi)容寫入到文件中
    a.write({"status_code": status_code})
    
    # 使用read()方法獲取文件中的內(nèi)容
    logger.debug(a.read())
 
 
 
if __name__ == '__main__':
    pytest.main(['-sv', 'test_get_set.py'])

返回結(jié)果：?

test_get_set.py::test_set 2021-12-24 18:24:39.292 | INFO? ? ?| cases.conftest:tmp_factory:150 - 當(dāng)前臨時(shí)文件的目錄為：/private/var/folders/_f/1d0lt83x1599bf6mcfppbwp40000gn/T/pytest-of-j/pytest-19/apitest-tmp-dir0

2021-12-24 18:24:39.347 | INFO? ? ?| cases.test_get_set:test_set:32 - 返回狀態(tài)碼：200

2021-12-24 18:24:39.347 | DEBUG? ? | cases.test_get_set:test_set:34 - (local('/private/var/folders/_f/1d0lt83x1599bf6mcfppbwp40000gn/T/pytest-of-j/pytest-19/apitest-tmp-dir0'), '/private/var/folders/_f/1d0lt83x1599bf6mcfppbwp40000gn/T/pytest-of-j/pytest-19/apitest-tmp-dir0')

2021-12-24 18:24:39.348 | DEBUG? ? | cases.test_get_set:test_set:38 - {'status_code': 200}

PASSED

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

創(chuàng)建的文件：

可以看到，tmpdir_factory會(huì)自動(dòng)為我們創(chuàng)建一個(gè)目錄，名字是`tmp_factory`方法中自定義的名稱后面加0，同時(shí)它的上級(jí)目錄會(huì)自動(dòng)從pytest-0遞增

說下這個(gè)辦法的優(yōu)缺點(diǎn)：

• 先說優(yōu)點(diǎn)，這種數(shù)據(jù)存儲(chǔ)是直接寫入到文件，因此即使運(yùn)行結(jié)束后也可以訪問，而不像第一種方法存儲(chǔ)的內(nèi)容會(huì)隨著用例運(yùn)行的結(jié)束而消失
• 再說缺點(diǎn)，因?yàn)檫@個(gè)臨時(shí)目錄最多只能保存3個(gè)，因此如果用例很多時(shí)，那么就可能存在文件被自動(dòng)刪除的風(fēng)險(xiǎn)。不過這個(gè)貌似可以通過修改默認(rèn)配置來解決，可以繼續(xù)研究下。
• 缺點(diǎn)二，上面的例子中，直接通過a.read()就訪問到了文件中的內(nèi)容，這是因?yàn)閮?nèi)容的存儲(chǔ)與讀取全是在一個(gè)方法內(nèi)，因此可以直接調(diào)用。如果是A方法存結(jié)果，在B中需要讀取，那么便不能直接使用read()方法了（因?yàn)槊總€(gè)方法都會(huì)創(chuàng)建一個(gè)目錄，并且默認(rèn)的讀取地址都是這個(gè)方法自己創(chuàng)建的目錄）。就需要我們自己去單獨(dú)封裝一個(gè)讀取文件的方法了，實(shí)現(xiàn)起來也不難而且tmpdir本身的這些方法也是對(duì)os.path一些方法的二次封裝。

話說回來，都需要我自己去封裝一個(gè)讀取文件的方法了，為啥不干脆把讀、寫都自己來做呢？這樣是否刪除文件、刪除幾個(gè)、什么時(shí)候刪除就完全由自己定義了啊，貌似會(huì)更方便。。。。。三、總結(jié)

綜上所述，大家應(yīng)該對(duì)這兩種方法都有了一些了解，核心都是通過pytest的conftes.py文件來實(shí)現(xiàn)的。

總結(jié)

到此這篇關(guān)于pytest用例間參數(shù)傳遞的兩種實(shí)現(xiàn)的文章就介紹到這了,更多相關(guān)pytest例間參數(shù)傳遞內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論