如何在Python中与Unitest一起测试方法呼叫订单
欢迎来到本期Piko教程!
在编写单元测试时,除了验证方法是否被调用外,有时还需要验证方法调用的顺序。本文将演示如何在Python中使用unittest来实现这一目标。import unittestfrom 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()]