Python - 模拟和存根

Python 模拟(Mocking) 和存根(Stubbing)是单元测试中的重要技术，它们通过用受控的替代品替换真实对象或方法来帮助隔离正在测试的功能。在本章中，我们将详细了解 Mocking 和 Stubbing -

Python 模拟

模拟是一种测试技术，其中创建 mock 对象以模拟真实对象的行为。

当测试一段与复杂、不可预测或缓慢的组件（如数据库、Web 服务或硬件设备）交互的代码时，这非常有用。

模拟的主要目的是隔离被测代码，并确保独立于其依赖项评估其行为。

模拟的主要特征

以下是 python 中模拟的主要特征 -

• 行为模拟：可以对 Mock 对象进行编程，以返回特定值、引发异常或模拟各种条件下真实对象的行为。
• 交互验证：Mock 可以通过允许测试人员验证特定方法是否使用预期的参数来记录它们的使用方式。
• 测试隔离：通过用 mock 替换真实对象，测试可以专注于被测代码的逻辑，而无需担心外部依赖项的复杂性或可用性。

Python 模拟示例

下面是 database.get_user 方法的示例，该方法被模拟以返回预定义的用户字典。然后，测试可以验证该方法是否使用正确的参数调用 -

from unittest.mock import Mock

# Create a mock object
database = Mock()

# Simulate a method call
database.get_user.return_value = {"name": "Prasad", "age": 30}

# Use the mock object
user = database.get_user("prasad_id")
print(user) 	

# Verify the interaction
database.get_user.assert_called_with("prasad_id")	

输出

Python 存根

存根是一种相关的测试技术，其中某些方法或函数被替换为返回固定的、预先确定的响应的“存根”。

存根比模拟更简单，因为它通常不涉及记录或验证交互。相反，存根侧重于通过确保一致和可重复的结果，为被测代码提供受控的输入。

Stubbing 的主要特征

以下是 python 中 Stubbing 的主要特征 -

• 固定响应：存根返回特定的预定义值或响应，而不管它们的调用方式如何。
• 简化的依赖项：通过用存根替换复杂的方法，测试可以避免设置或管理复杂的依赖关系。
• 关注输入：Stubbing 强调通过允许测试人员专注于被测试代码的逻辑和输出来为被测代码提供已知的输入。

Python 存根示例

以下是 get_user_from_db 函数的示例，该函数被存根以始终返回预定义的用户字典。该测试不需要与真实数据库交互，以简化设置并确保结果一致 -

from unittest.mock import patch

# Define the function to be stubbed
def get_user_from_db(user_id):
	 	# Simulate a complex database operation
	 	pass

# Test the function with a stub
with patch('__main__.get_user_from_db', return_value={"name": "Prasad", "age": 25}):
	 	user = get_user_from_db("prasad_id")
	 	print(user) 	

输出

Python 模拟(Mocking) vs. 存根(Stubbing)

对 Mocking 和 Stubbing 主要功能、用途和用例的比较清楚地说明了何时使用每种方法。通过探索这些区别，开发人员可以创建更有效且可维护的测试，最终获得更高质量的软件。

下表显示了基于不同标准的模拟和存根之间的主要区别 -