如何在Python中与Unitest一起测试方法呼叫订单(呼叫.中与.订单.测试.方法...)

wufei1232025-02-15python17

如何在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()]

以上就是如何在Python中与Unitest一起测试方法呼叫订单的详细内容,更多请关注知识资源分享宝库其它相关文章!

发表评论

访客

◎欢迎参与讨论,请在这里发表您的看法和观点。