Python中pytest命令行實現環(huán)境切換
前言
在自動化測試過程中經常需要在不同的環(huán)境下進行測試驗證，所以寫自動化測試代碼時需要考慮不同環(huán)境切換的情況。pytest鉤子函數pytest_addoption可以很好幫我們解決這個痛點。
pytest_addoption(parser, pluginmanager)
注冊argparse樣式選項和ini樣式配置值，在測試運行開始時調用一次。
注意：
由于pytest在啟動過程中如何發(fā)現插件，因此該函數只能在位于測試根目錄的插件或conftest.py文件中實現。
參數
parser（pytest.parser）–若要添加命令行選項，請調用parser.addoption（…）。若要添加ini文件值，請調用解析器.addini（…）。
pluginmanager（pytest.PytestPluginManager）–pytest插件管理器，可用于安裝hookspec（）或hookpimpl（），并允許一個插件調用另一個插件的鉤子來更改命令行選項的添加方式。
以后可以分別通過配置對象訪問選項：
config.getoption（name）來檢索命令行選項的值。
config.getini（name）來檢索從ini樣式文件中讀取的值。
config對象通過.config屬性在許多內部對象上傳遞，或者可以作為pytestconfig fixture檢索。
在conftest.py文件中定義命令行參數
def pytest_addoption(parser):
"""
添加命令行參數
parser.addoption為固定寫法
default 設置一個默認值，此處設置默認值為sit
choices 參數范圍，傳入其他值無效
help 幫助信息
"""
parser.addoption(
"--env", default="sit", choices=["dev", "sit", "uat"], help="環(huán)境參數"
)我們定義了不同環(huán)境下的命令參數：dev、sit、uat，我們怎么獲取運行的命令行參數呢？
獲取命令行參數
@pytest.fixture(scope="session")
def get_env(request):
return request.config.getoption("--env")設置不同環(huán)境的全局變量
在不同的測試環(huán)境下，URL、用戶信息等數據都是不一樣的，建議在conftest中給全局變量賦值可以減少代碼冗余。
先定義一個數據文件，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中根據環(huán)境設置全局變量值
# 設置不同環(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):
"""
添加命令行參數
parser.addoption為固定寫法
default 設置一個默認值，此處設置默認值為sit
choices 參數范圍，傳入其他值無效
help 幫助信息
"""
parser.addoption(
"--env", default="sit", choices=["dev", "sit", "uat"], help="環(huán)境參數"
)
@pytest.fixture(scope="session")
def get_env(request):
return request.config.getoption("--env")
# 設置不同環(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)境信息設置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'])
可以看出在不同的命令下獲得的測試數據也不一樣，這樣我們就達到了環(huán)境切換的目的了~
到此這篇關于Python中pytest命令行實現環(huán)境切換的文章就介紹到這了,更多相關pytest 環(huán)境切換內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！
相關文章
這篇文章主要介紹了django 數據庫返回queryset實現封裝為字典，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-05-05
這篇文章主要為大家詳細介紹了如何利用python實現炫酷屏幕保護效果,文中的示例代碼講解詳細,具有一定的學習價值,感興趣的小伙伴可以跟隨小編一起了解一下2023-12-12
最近在做SEO的時候，為了讓發(fā)的外鏈能夠快速的收錄，想到了利用ping的功能，google和百度都有相關的ping介紹，有興趣的朋友可以去看看相關的知識。下面這篇文章主要介紹了利用Python腳本實現ping百度和google的方法,需要的朋友可以參考借鑒，一起來看看吧。2017-01-01
本文主要介紹了Selenium多窗口切換解決方案，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2022-07-07
這篇文章主要介紹了如何使用Python-VTK繪制線條，主要繪制直線和曲線，下面文章詳細實現過程需要的小伙伴可以參考一下2022-04-04
使用Python自動化生成PPT并結合LLM生成內容的代碼解析
PowerPoint是常用的文檔工具,但手動設計和排版耗時耗力,本文將展示如何通過 Python 自動化提取 PPT 樣式并生成新 PPT,同時結合大語言模型（LLM）生成內容（如自我介紹文本）,實現高效、個性化的 PPT 制作,需要的朋友可以參考下2025-05-05
解決python3報錯之takes?1?positional?argument?but?2?were?gi
這篇文章主要介紹了解決python3報錯之takes?1?positional?argument?but?2?were?given問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-03-03
本文通過實例代碼給大家介紹了opencv調整圖像亮度對比度,代碼簡單易懂，非常不錯，具有一定的參考借鑒價值，需要的朋友可以參考下2019-09-09
在數據處理與分析過程中,缺失值、重復值、異常值等問題是常見的挑戰(zhàn),本文總結了多種數據清洗與處理方法,文中的示例代碼簡潔易懂,有需要的小伙伴可以參考下2025-01-01
Python在for循環(huán)里處理大數據的推薦方法實例
這篇文章主要介紹了Python在for循環(huán)里處理大數據的推薦方法實例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2024-01-01