领取资料，咨询答疑，请➕wei: June__Go
上一小节我们学习了pytest_runtest_setup钩子函数的使用方法，本小节我们讲解一下pytest_runtest_call钩子函数的使用方法。
pytest_runtest_call 钩子函数在 pytest 调用测试函数（即测试用例的执行阶段）之前被调用。这个钩子可以用来在测试函数执行之前进行一些准备工作，或者在测试函数执行之后进行一些清理工作。以下是如何使用这个钩子函数的具体方法和代码示例：
首先，确保你的项目中有一个 conftest.py 文件。然后，在 conftest.py 文件中定义 pytest_runtest_call 钩子函数：
# conftest.py
import pytest
import logging
# 设置日志
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
def pytest_runtest_call(item):
# 在测试函数执行之前执行的代码
logging.info(f"Running test: {item.name}")
# 假设我们需要在每个测试用例执行前进行一些准备工作
# 例如，设置测试数据、初始化资源等
# ...
# 执行测试函数
yield # 这是 pytest_runtest_call 钩子的一个特殊之处，它允许我们使用 yield
# 来暂停执行，等待测试函数执行完成后继续执行后续代码
# 在测试函数执行之后执行的代码
# 例如，清理资源、验证测试结果等
# ...
# 如果测试函数执行成功，我们可以在这里进行一些后续处理
# 如果测试函数执行失败，pytest 会调用 pytest_runtest_teardown 钩子来处理清理工作
# ...
# 你也可以在 pytest_runtest_call 钩子中捕获异常，并进行相应的处理
def pytest_runtest_call(item, call):
try:
# 在这里执行测试前的准备工作
# ...
# 调用测试函数
call()
# 在这里执行测试后的清理工作
# ...
except Exception as e:
# 如果在测试执行过程中发生异常，我们可以在这里捕获并处理
logging.error(f"An error occurred during the test: {e}")
# 如果你想要在测试执行后立即执行某些操作，可以使用 pytest_runtest_teardown 钩子
def pytest_runtest_teardown(item, nextitem):
# 在这里执行测试执行后的清理工作
# ...在这个示例中，我们首先设置了日志系统，以便在测试过程中记录重要信息。然后，我们在 pytest_runtest_call 钩子函数中使用 yield 关键字来暂停执行，等待测试函数执行完成后继续执行后续代码。这样，我们可以在测试函数执行前后执行一些操作，例如设置和清理测试数据。
请注意，pytest_runtest_call 钩子函数的参数 call 是一个函数，它代表了要执行的测试函数。我们通过调用 call() 来执行测试函数。如果在测试执行过程中发生异常，我们可以在 pytest_runtest_call 钩子中捕获并处理这些异常。
这个示例展示了如何在测试用例执行前后执行一系列操作，并处理可能出现的异常情况。在实际应用中，你可能需要根据具体的测试需求来调整这些操作。
让我们通过一个更复杂的示例来展示 pytest_runtest_call 钩子函数的使用方法。在这个示例中，我们将模拟一个场景，其中我们需要在每个测试用例执行前后进行数据库操作的模拟，配置日志记录器，以及在测试用例执行失败时记录详细的错误信息。我们还将展示如何在测试用例执行后进行断言验证。
首先，确保你的项目中有一个 conftest.py 文件。然后，在 conftest.py 文件中定义 pytest_runtest_call 钩子函数：
# conftest.py
import pytest
import logging
import os
from some_database_module import DatabaseConnection, execute_query # 假设这是我们的数据库操作模块
from some_logging_module import configure_logger # 假设这是我们的日志配置模块
# 设置日志
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
# 假设我们有一个全局数据库连接对象
db_connection = None
def pytest_runtest_call(item):
# 在测试用例执行前执行的代码
logging.info(f"Setting up for test: {item.name}")
# 配置日志记录器
configure_logger(item.name)
# 初始化数据库连接
try:
global db_connection
db_connection = DatabaseConnection()
execute_query(db_connection, "CREATE TABLE IF NOT EXISTS test_table (id INT PRIMARY KEY, value VARCHAR(255))")
logging.info("Database connection initialized and test table created.")
except Exception as e:
logging.error(f"Failed to initialize database connection for test: {item.name} - {e}")
# 如果数据库初始化失败，标记测试用例为预期失败
pytest.xfail(f"Skipping test {item.name} due to database initialization failure.")
return True # 阻止其他钩子函数执行
# 执行测试函数
yield # 允许测试函数执行
# 在测试用例执行后执行的代码
if db_connection:
try:
# 执行一些清理数据库的操作
execute_query(db_connection, "TRUNCATE TABLE test_table")
logging.info("Test table cleaned up.")
except Exception as e:
logging.error(f"Failed to clean up test table for test: {item.name} - {e}")
# 如果清理失败，记录错误信息，但允许测试继续执行
# 关闭数据库连接
finally:
db_connection.close()
logging.info("Database connection closed.")
# 在这里执行其他清理工作，例如删除临时文件等
# ...
# 如果测试用例执行成功，进行断言验证
# 假设我们有一个断言验证函数
if item.name == "test_assertion":
assert item.test_data['key1'] == "value1", "Assertion failed in test_assertion"
logging.info("Assertion check passed.")
# 在测试用例执行后执行的代码
def pytest_runtest_teardown(item, nextitem):
# 在这里执行测试执行后的清理工作
# 例如，验证测试结果、记录测试日志等
# ...在这个示例中，我们在 pytest_runtest_call 钩子函数中首先配置了日志记录器，然后尝试初始化数据库连接并创建一个测试表。如果数据库初始化失败，我们使用 pytest.xfail 跳过当前测试用例。在测试用例执行后，我们执行数据库清理操作，并关闭数据库连接。
我们还定义了一个 pytest_runtest_teardown 钩子函数来执行测试用例执行后的清理工作，例如验证测试结果和记录测试日志。
请注意，这个示例中的数据库操作和日志配置都是假设的，你需要根据你的项目实际情况来实现这些功能。这个示例展示了如何在测试用例执行前后执行一系列复杂的操作，并处理可能出现的异常情况。同时，它还展示了如何在测试用例执行后进行断言验证。
最后感谢每一个认真阅读我文章的人，礼尚往来总是要有的，虽然不是什么很值钱的东西，如果你用得到的话可以直接拿走，希望可以帮助到大家！领取资料，咨询答疑，请➕wei: June__Go
