
欢迎来到本期Piko教程!
在编写单元测试时,除了验证方法是否被调用外,有时还需要验证方法调用的顺序。本文将演示如何在Python中使用unittest来实现这一目标。
import unittest
from unittest.mock import MagicMock, call
# 被测试函数
def function(obj):
obj.say_hello()
obj.say_goodbye()
# 单元测试
class TestFunctionCallOrder(unittest.TestCase):
def test_function_call_order(self):
# 创建模拟对象
mock_obj = MagicMock()
# 使用模拟对象调用被测试函数
function(mock_obj)
# 定义预期调用顺序
expected_calls = [call.say_hello(), call.say_goodbye()]
# 验证实际调用顺序与预期调用顺序是否匹配
mock_obj.assert_has_calls(expected_calls)
如果修改函数调用顺序,例如:
def function(obj):
obj.say_goodbye()
obj.say_hello()
上述测试将立即捕获此回归:
AssertionError: Calls not found. Expected: [call.say_hello(), call.say_goodbye()] Actual: [call.say_goodbye(), call.say_hello()]










