Python中pytest命令行實現(xiàn)環(huán)境切換
前言
在自動化測試過程中經(jīng)常需要在不同的環(huán)境下進行測試驗證，所以寫自動化測試代碼時需要考慮不同環(huán)境切換的情況。pytest鉤子函數(shù)pytest_addoption可以很好幫我們解決這個痛點。
pytest_addoption(parser, pluginmanager)
注冊argparse樣式選項和ini樣式配置值，在測試運行開始時調(diào)用一次。
注意：
由于pytest在啟動過程中如何發(fā)現(xiàn)插件，因此該函數(shù)只能在位于測試根目錄的插件或conftest.py文件中實現(xiàn)。
參數(shù)
parser（pytest.parser）–若要添加命令行選項，請調(diào)用parser.addoption（…）。若要添加ini文件值，請調(diào)用解析器.addini（…）。
pluginmanager（pytest.PytestPluginManager）–pytest插件管理器，可用于安裝hookspec（）或hookpimpl（），并允許一個插件調(diào)用另一個插件的鉤子來更改命令行選項的添加方式。
以后可以分別通過配置對象訪問選項：
config.getoption（name）來檢索命令行選項的值。
config.getini（name）來檢索從ini樣式文件中讀取的值。
config對象通過.config屬性在許多內(nèi)部對象上傳遞，或者可以作為pytestconfig fixture檢索。
在conftest.py文件中定義命令行參數(shù)
def pytest_addoption(parser):
"""
添加命令行參數(shù)
parser.addoption為固定寫法
default 設(shè)置一個默認值，此處設(shè)置默認值為sit
choices 參數(shù)范圍，傳入其他值無效
help 幫助信息
"""
parser.addoption(
"--env", default="sit", choices=["dev", "sit", "uat"], help="環(huán)境參數(shù)"
)我們定義了不同環(huán)境下的命令參數(shù)：dev、sit、uat，我們怎么獲取運行的命令行參數(shù)呢？
獲取命令行參數(shù)
@pytest.fixture(scope="session")
def get_env(request):
return request.config.getoption("--env")設(shè)置不同環(huán)境的全局變量
在不同的測試環(huán)境下，URL、用戶信息等數(shù)據(jù)都是不一樣的，建議在conftest中給全局變量賦值可以減少代碼冗余。
先定義一個數(shù)據(jù)文件，data_util.py分別獲取用戶信息和URL信息
def get_env():
env = {
'sit': 'www.baidu.com',
'uat': 'www.hao123.com'
}
return env
def get_user():
users = {
'sit': ['user1', 'pwd1'],
'uat': ['user2', 'pwd2']
}
return users然后在conftest中根據(jù)環(huán)境設(shè)置全局變量值
# 設(shè)置不同環(huán)境下的全局變量
@pytest.fixture(scope="session")
def set_env(get_env):
if get_env == 'sit':
env_url = data_util.get_env()['sit']
user = data_util.get_user()['sit']
if get_env == 'uat':
env_url = data_util.get_env()['uat']
user = data_util.get_user()['uat']
return {'env_url': env_url, 'user': user}注意fixture的使用范圍為整個測試會話。
以下是完整的conftest
import pytest
import data_util
def pytest_addoption(parser):
"""
添加命令行參數(shù)
parser.addoption為固定寫法
default 設(shè)置一個默認值，此處設(shè)置默認值為sit
choices 參數(shù)范圍，傳入其他值無效
help 幫助信息
"""
parser.addoption(
"--env", default="sit", choices=["dev", "sit", "uat"], help="環(huán)境參數(shù)"
)
@pytest.fixture(scope="session")
def get_env(request):
return request.config.getoption("--env")
# 設(shè)置不同環(huán)境下的全局變量
@pytest.fixture(scope="session")
def set_env(get_env):
if get_env == 'sit':
env_url = data_util.get_env()['sit']
user = data_util.get_user()['sit']
if get_env == 'uat':
env_url = data_util.get_env()['uat']
user = data_util.get_user()['uat']
return {'env_url': env_url, 'user': user}定義測試類及測試方法
注意fixture不能在x-unit風格下的setup\teardown中引用，因此需要使用fixture定義setup、teardown方法才能引用到conftest里的fixture，一般我們在setup方法中初始化環(huán)境變量具體如下：
@pytest.fixture()
def class_fixture(set_env):
print('setup_class')
url = set_env.get('env_url')
user = set_env.get('user')
print(url, user)
yield
print('teardown class')這樣我們就在測試前把環(huán)境信息設(shè)置OK了。
測試驗證
以下是測試方法
import pytest
pytestmark = pytest.mark.usefixtures("module_fixture")
@pytest.fixture(scope="module", params=["test_fixture"])
def module_fixture(request):
param = request.param
print(" SETUP module", param)
yield param
print(" TEARDOWN module", param)
@pytest.fixture()
def class_fixture(set_env):
print('setup_class')
url = set_env.get('env_url')
user = set_env.get('user')
print(url, user)
yield
print('teardown class')
@pytest.fixture(scope="function", params=[1, 2])
def function_fixture(request):
param = request.param
print(" SETUP function", param)
yield param
print(" TEARDOWN function", param)
@pytest.mark.usefixtures('class_fixture')
class TestFixture:
def test_0(self, function_fixture):
print(" RUN test0 with function_fixture", function_fixture)
def test_1(self, module_fixture):
print(" RUN test1 with module_fixture", module_fixture)
def test_2(self, function_fixture, module_fixture):
print(f" RUN test2 with function_fixture {function_fixture} and module_fixture {module_fixture}")
def test_env(self, get_env):
print(f"The current environment is: get_env")
if __name__ == '__main__':
pytest.main(['-v', '-s','--env=uat', 'test_fixture.py::TestFixture::test_0'])我們首先填的uat命令運行，查看輸出：
可以看出輸出是正確的，我們再切換成sit試試：
if __name__ == '__main__':
pytest.main(['-v', '-s','--env=uat', 'test_fixture.py::TestFixture::test_0'])
可以看出在不同的命令下獲得的測試數(shù)據(jù)也不一樣，這樣我們就達到了環(huán)境切換的目的了~
到此這篇關(guān)于Python中pytest命令行實現(xiàn)環(huán)境切換的文章就介紹到這了,更多相關(guān)pytest 環(huán)境切換內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
這篇文章主要介紹了Python實現(xiàn)的IP端口掃描工具類,結(jié)合實例形式分析了Python基于socket模塊與多線程操作實現(xiàn)IP端口掃描的相關(guān)操作技巧,需要的朋友可以參考下2019-02-02
pygame多種方式實現(xiàn)屏保操作(自動切換、鼠標切換、鍵盤切換)
這篇文章主要介紹了pygame多種方式實現(xiàn)屏保操作(自動切換、鼠標切換、鍵盤切換),文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2021-04-04
Python unittest 簡單實現(xiàn)參數(shù)化的方法
今天小編就為大家分享一篇Python unittest 簡單實現(xiàn)參數(shù)化的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-11-11
這篇文章主要介紹了django從后臺返回html代碼的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-03-03
Python pywifi ERROR Open handle fai
這篇文章主要介紹了Python pywifi ERROR Open handle failed問題及解決方案，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-06-06
本文主要記錄了個人在項目中遇到的一些Python中對于文件、文件夾的操作需要涉及到的函數(shù)等內(nèi)容，非常的詳盡，有需要的可以參考下2014-10-10
本篇文章主要介紹了Python變量和字符串的相關(guān)資料。具有很好的參考價值。下面跟著小編一起來看下吧2017-04-04
這篇文章主要介紹了python實現(xiàn)計算資源圖標crc值的方法,通過解析資源文件找到icon的數(shù)據(jù),從而實現(xiàn)該功能,需要的朋友可以參考下2014-10-10
這篇文章主要介紹了python中如何設(shè)置list步長，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-05-05
這篇文章主要介紹了python中的正斜杠與反斜杠實例驗證,本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2023-03-03