pytest教程-46-钩子函数-pytest_sessionstart

领取资料，咨询答疑，请➕wei:  June__Go

上一小节我们学习了pytest_report_testitemFinished钩子函数的使用方法，本小节我们讲解一下pytest_sessionstart钩子函数的使用方法。

pytest_sessionstart 是 Pytest 提供的一个钩子函数，它在创建 Session 对象之后、执行收集测试用例之前调用。这个钩子可以用来执行一些全局的初始化操作，比如设置日志、初始化数据库连接等。

以下是一个具体的使用示例：

首先，在你的项目中创建一个 conftest.py 文件（如果还没有的话）。这个文件通常位于你的测试目录下，并且是 Pytest 用来收集钩子函数的地方。

然后，在 conftest.py 文件中定义 pytest_sessionstart 钩子函数：

# conftest.pydef pytest_sessionstart(session):# 这里可以执行一些全局的初始化操作print("Session is starting...")# 例如，设置日志import logginglogging.basicConfig(level=logging.INFO)# 或者初始化数据库连接# db_connection = create_db_connection()# session.db_connection = db_connection

在这个例子中，我们使用 logging 模块来设置日志记录。你也可以在这里执行其他任何需要在测试开始前完成的初始化任务。

当你运行 Pytest 时，pytest_sessionstart 钩子函数会被自动调用。如果你需要在测试结束后执行一些清理工作，可以使用 pytest_sessionfinish 钩子函数。

请注意，如果你在 conftest.py 中定义了 pytest_sessionstart 钩子，那么它将在整个测试会话开始时执行一次。如果你需要在每个测试文件开始前执行某些操作，你可能需要使用 pytest_collection_start 钩子。

当然，让我们创建一个更复杂的 pytest_sessionstart 钩子示例。在这个示例中，我们将执行以下操作：

1. 设置日志记录。
2. 创建一个全局的数据库连接。
3. 初始化一个测试数据准备服务。
4. 打印一条启动消息。

首先，确保你已经安装了 pytest 和 pytest-xdist（如果你打算并行运行测试）。如果没有安装，可以通过以下命令安装：

pip install pytest pytest-xdist

然后，在你的项目根目录下创建 conftest.py 文件，并添加以下代码：

# conftest.pyimport logging
import threading
import time
import sqlite3# 设置日志
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')# 全局数据库连接
class DatabaseConnection:def __init__(self):self.connection = sqlite3.connect('test.db')self.cursor = self.connection.cursor()def close(self):self.cursor.close()self.connection.close()def __enter__(self):return self.cursordef __exit__(self, exc_type, exc_val, exc_tb):self.connection.commit()if exc_type:self.connection.rollback()# 初始化数据库连接
def setup_database():db = DatabaseConnection()db.cursor.execute('''CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY AUTOINCREMENT,name TEXT NOT NULL,age INTEGER)''')db.commit()db.close()# 测试数据准备服务
class TestDataService:def __init__(self):self.data_prepared = threading.Event()def prepare_data(self):# 模拟数据准备过程time.sleep(2)logging.info("Test data prepared.")self.data_prepared.set()def wait_for_data(self):self.data_prepared.wait()# 初始化测试数据服务
def pytest_sessionstart(session):logging.info("Session is starting...")# 设置数据库setup_database()# 初始化测试数据服务test_data_service = TestDataService()prepare_thread = threading.Thread(target=test_data_service.prepare_data)prepare_thread.start()session._test_data_service = test_data_service  # 将服务实例保存到 session 对象中# 等待数据准备完成test_data_service.wait_for_data()logging.info("Session is ready to run tests.")

在这个示例中，我们首先定义了一个 DatabaseConnection 类来管理数据库连接。然后，我们创建了一个 TestDataService 类，它有一个 prepare_data 方法来模拟数据准备过程，并且使用一个线程来执行这个准备过程。我们还在 pytest_sessionstart 钩子中调用了这些初始化方法，并等待数据准备完成。

现在，当你运行 Pytest 时，pytest_sessionstart 钩子将被调用，并且会执行上述的初始化操作。这样，你就可以在测试用例中使用准备好的数据了。

请注意，这个示例假设你正在使用 SQLite 数据库，并且你的测试数据准备过程是同步的。在实际应用中，你可能需要根据你的具体需求调整数据库连接和数据准备逻辑。

最后感谢每一个认真阅读我文章的人，礼尚往来总是要有的，虽然不是什么很值钱的东西，如果你用得到的话可以直接拿走，希望可以帮助到大家！领取资料，咨询答疑，请➕wei:  June__Go