Question

我的应用程序代码如下所示。

# Filename: app.py
class Foo:
    def __init__(self):
        self.callback = None

    def set_handler(self, callback):
        self.callback = callback

    def run(self, details):
        name, age = details.split('/')
        if age.isdigit():
            age = int(age)
        else:
            age = -1
        return self.callback(name, age)

如您所见，它提供了set_handler方法来设置回调。稍后必须使用两个参数调用回调：字符串和整数。我试图在单位测试中确保这一点。

# Filename: test_app.py
import unittest
from app import Foo

class AppTest(unittest.TestCase):
    def f(self, a, b):
        # This callback should get called with the first argument as
        # string and the second argument as integer
        return repr(a) + ',' + repr(b)

    def test_callback(self):
        foo = Foo()
        foo.set_handler(self.f)
        self.assertEqual(foo.run('John/20'), "'John',20")
        self.assertEqual(foo.run('John/xyz'), "'John',-1")

if __name__ == '__main__':
    unittest.main()

本单元测试成功。但我不认为我的测试方式很强大。这个单元测试基本上是一个hack，因为我不知道如何正确测试是否使用正确类型的参数调用了回调。我觉得奇怪的是，AppTest的f方法通过尝试返回由repr()个参数组成的值来分担类型检查的责任，这并不是完全可靠的说法至少。

你能帮帮我吗？是否有可能减轻f方法测试类型的责任？

Answer 1

修改

尝试使用unittest.mock（Python 3.3上的标准库）。它允许您断言如何调用方法。例如：

import unittest from unittest.mock import Mock from app import Foo class AppTest(unittest.TestCase): def test_callback(self): foo = Foo() f = Mock() foo.set_handler(f) foo.run('John/20') f.assert_called_with('John', 20) foo.run('John/xyz') f.assert_called_with('John', -1) if __name__ == '__main__': unittest.main()

使用Python unittest测试回调调用的正确方法是什么？

1 个答案: