pytestconftest——Python測試之conftest文件操作

一、pytest和conftest文件簡介

pytest是Python中一個非常流行的測試框架，它支持各種類型的測試，包括單元測試、功能測試和端到端測試。pytest使用靈活的fixture系統管理測試所需的資源。fixture是一個函數，它可以在測試運行之前和之後執行一些代碼，從而管理測試依賴項。conftest.py是pytest的一個特殊文件，它包含在pytest測試運行過程中自動查找的fixture和插件。

二、在conftest文件中定義fixture

在conftest文件中定義fixture，就可以在整個測試套件中使用該fixture。當任何測試運行之前，pytest會在conftest文件中搜索fixture定義，並確保它們在測試運行之前執行。下面是一個簡單的示例：

 # conftest.py

import pytest

@pytest.fixture
def my_fixture():
    return "my fixture value"

在上面的示例中，我們定義了一個名為my_fixture的fixture。fixture返回一個簡單的字元串，這是測試中的測試數據。

三、使用fixture進行測試

有了fixture之後，我們可以在測試函數中使用該fixture。下面是一個示例：

# test_example.py

def test_my_fixture(my_fixture):
    assert my_fixture == "my fixture value"
    

在這個例子中，我們導入了my_fixture fixture並使用它在測試函數中進行斷言。pytest會自動調用fixture並將它作為參數傳遞給測試函數。這種方式可以避免在測試函數中重複編寫相同的代碼，並幫助測試代碼保持整潔和可讀。

四、使用fixture管理瀏覽器會話

在Web應用程序測試中，使用瀏覽器是常見的。由於瀏覽器會話是資源密集型的，因此管理它們是非常重要的。fixture可以用來自動啟動和關閉瀏覽器會話，這樣測試就可以專註於測試場景，而不必擔心細節。下面是一個使用Selenium和pytest-selenium插件的示例：

 # conftest.py

import pytest
from selenium import webdriver

@pytest.fixture(scope="session")
def browser():
    browser = webdriver.Chrome()
    yield browser
    browser.quit()

# test_example.py

def test_google_search(browser):
    browser.get("https://www.google.com")
    assert "google" in browser.title
    

在上面的示例中，我們使用Selenium和Chrome瀏覽器啟動瀏覽器會話，並將其聲明為作用域為session的fixture。這意味著該fixture將在整個測試套件中只執行一次，在所有測試結束後關閉瀏覽器。

五、使用fixture進行資料庫測試

在測試資料庫應用程序時，使用fixture可以使測試更容易編寫和管理。pytest提供了許多支持不同類型資料庫的插件，例如pytest-postgresql、pytest-mongodb和pytest-mysql。下面是一個使用pytest-mysql插件的示例：

# conftest.py

import pytest
import mysql.connector

@pytest.fixture(scope="session")
def mysql_connection():
    connection = mysql.connector.connect(
        host="localhost",
        user="testuser",
        password="testpass",
        database="testdb"
    )
    yield connection
    connection.close()

# test_example.py

def test_mysql_query(mysql_connection):
    cursor = mysql_connection.cursor()
    cursor.execute("SELECT COUNT(*) FROM my_table")
    result = cursor.fetchone()
    assert result[0] == 10
    

在上面的示例中，我們使用pytest-mysql插件創建了一個名為mysql_connection的fixture。fixture將建立一個MySQL連接，進行一些操作，然後關閉連接。我們還編寫了一個測試函數來查詢資料庫，並進行斷言。pytest會自動調用fixture並將其作為參數傳遞給測試函數。