标签导航：

如何在python中与unitest一起测试方法呼叫订单

欢迎来到本期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()]