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并将其作为参数传递给测试函数。