在軟件測試中，測試環境的切換是個令人頭疼的問題。不同環境的配置不同，如何高效切換測試環境成為許多測試開發人員關注的重點。你是否希望在運行測試用例時，能夠動態選擇測試環境，而不是繁瑣地手動修改配置？

Pytest 測試框架的 hook 函數可以輕松實現這一需求！ 本文將帶你詳細了解如何借助 pytest 的強大功能，實現測試環境的動態切換。

在自動化測試過程中，有時候我們需要切換不同的測試環境用于驗證功能是否正常（比如在測試環境中功能是正常的，但是到了預發布環境中就出現了問題）。往往我們需要手動修改測試腳本中的配置適配在不同的環境執行，十分不優雅！

如果能在執行測試腳本之前指定對應環境進行自動化配置那就不需要手動一個個修改配置了，比如像pytest執行時可以通過傳入豐富的內置命令行參數改變其測試行為。同樣pytest也能支持自定義命令行參數，通過對應的參數選取對應的環境配置進行執行，實現這一需求可以通過pytest的hook函數。

Pytest 是 Python 社區最受歡迎的測試框架之一，其靈活的插件機制和豐富的 hook 函數，讓開發者能夠根據需求自定義測試行為。在動態切換測試環境的場景中，以下思路和方法非常實用：

1. 使用 pytest 的 pytest_addoption 添加命令行參數

通過為 pytest 添加命令行參數，我們可以實現環境配置的動態傳遞。

實現步驟：

• 在 conftest.py 文件中，定義一個選項用于傳遞環境信息。

def pytest_addoption(parser):parser.addoption("--env", action="store", default="dev", help="Specify test environment: dev/staging/prod")

關于pytest hook函數

Hook函數（鉤子函數），它是pytest框架的開發者為了讓用戶更好的去擴展pytest的功能而預留的一些函數。而預留的這些函數在整個測試執行的生命周期中特定的階段會被pytest自動調用執行。

關于pytest中的預留鉤子，可以通過開發插件，和在conftest.py去實現這些鉤子。pytest中的鉤子函數按功能一共分為6類：引導鉤子，初始化鉤子、用例收集鉤子、用例執行鉤子、報告鉤子、調試鉤子，當然動態切換測試環境這個需求我們不需要熟悉所有的鉤子函數，這里我們只需要知道pytest_addoption這個初始化鉤子函數即可：

函數

pytest_addoption

參數

• parser ：參數解析器
• pluginmanager ：插件管理器

觸發時機：

• conftest文件加載完之后執行， 在測試運行開始時調用一次。

作用：

• 添加運行命令的命令行參數，pytest.ini的配置參數

實現步驟

1：新增conftest.py文件，添加pytest_addoption鉤子函數

def pytest_addoption(parser):"""注冊自定義參數命令行參數"""parser.addoption("--env", default="test", choices=['dev', 'test', 'pre'],help="命令行參數 '--env' 設置環境切換")

參數說明：

• --env 代表需要傳入 --env 參數，后續通過pytest --env=test 或者 pytest --env test去運行
• default 代表默認值，不傳--env參數的情況下默認使用test環境
• choices 設置參數范圍，僅限于dev、test、pre這三個環境
• help 代表命令行參數的幫助信息

需要注意：鉤子函數名字是固定的，不能隨意更改

2：定義fixture函數，用于獲取命令行參數

@pytest.fixture(scope='session')
def get_env(request):option = request.config.getoption("--env")if option == 'test':print("當前環境為測試環境")elif option == "dev":print("當前環境為開發環境")elif option == "pre":print("當前環境為預發布環境")

?

3：在測試函數上使用fixture

我們可以看到通過--env參數可以切換不同的環境執行，不加參數的情況下使用的是test環境。

4：fixture函數解析不同的配置

上述的步驟能夠完成基本的命令行參數解析，但是離動態切換環境還差一步：我們需要通過不同的參數讀取對應的配置文件，將配置文件信息解析出來傳遞給測試用例使用。

在項目的config目錄下保存項目的不同環境配置文件：dev-env.ini、test-env.ini、pre-env.ini。文件中存儲不同的環境參數值（比如URL、賬號信息、數據庫配置信息等），如下所示：

[host]
baseurl = http://192.168.3.200:8080[account]
username = admin
password = 123456abc[dbinfo]
url = http://db.192.168.3.100:8081
port = 3306
username = root
password = 123456abc

[host] baseurl = http://test.lemonban.com [account] username = lemon_auto password = lemon123456 [dbinfo] url = http://db.lemonban.com port = 3306 username = student password = 123456a

通過在測試函數中引用fixture函數get_env，即可獲取對應的配置參數值。

在敏捷開發和 DevOps 的背景下，動態切換測試環境已經成為軟件測試流程中不可或缺的一部分。傳統的手動切換環境方式效率低下，甚至可能導致配置錯誤，進而影響測試結果的準確性。 借助 Pytest 的靈活機制，動態切換測試環境不僅能提升測試效率，還能更好地適應多環境并行測試的需求。

Pytest 的 hook 函數為動態切換測試環境提供了強大的支持。通過簡單的配置和代碼調整，你可以輕松實現多環境測試的需求，讓測試流程更加高效和規范化。

“測試環境切換不再是難題，Pytest 帶你邁向自動化測試的新高度！”