Question

我正在尝试为项目设置一些自动单元测试。我有一些功能，作为副作用偶尔会调用另一个功能。我想写一个单元测试，测试第二个函数被调用，但我很难过。下面是伪代码示例：

def a(self):
    data = self.get()
    if len(data) > 3500:
        self.b()

    # Bunch of other magic, which is easy to test.

def b(self):
    serial.write("\x00\x01\x02")

如何测试b() - 被调用？

Answer 1

您可以使用mock模块模拟函数b并检查它是否被调用。这是一个例子：

import unittest
from mock import patch


def a(n):
    if n > 10:
        b()

def b():
    print "test"


class MyTestCase(unittest.TestCase):
    @patch('__main__.b')
    def test_b_called(self, mock):
        a(11)
        self.assertTrue(mock.called)

    @patch('__main__.b')
    def test_b_not_called(self, mock):
        a(10)
        self.assertFalse(mock.called)

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

另见：

希望有所帮助。

Answer 2

这件事可能有各种解决方案。但首先，我必须指出，单元测试是针对测试单元的;什么单位可以是观看事物的问题。在你的情况下，我会说一个单位是函数a()，因为你想测试你的单位行为正确，这意味着它在正确的位置调用函数b()。另一种观点是，函数a()和b()是一个单位。那么你 not 就像检查函数b()是否被调用一样，你只想测试函数a()的调用结果。因此，最好选择适合自己情况的想法。

如果你真的想要将函数a()作为一个单元进行测试，你应该准备你的单元进行测试。在您的情况下，可以通过向其添加一个额外的参数（默认为函数b）来完成，并且将使用该参数而不是b()中的硬编码函数a()调用：

def a(self, subfunction=None):
    is subfunction is None:  # for testing
        subfunction = self.b
    data = self.get()
    if len(data) > 3500:
        subfunction()

现在你可以注入（在测试期间）一个帮助函数，它只是通知测试它被调用：

store = None
def injected(*args, **kwargs):
    global store
    store = args, kwargs

obj.a(subfunction=injected)
# now check if global store is set properly

如何测试在具有nosetests的函数中调用函数

2 个答案: