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

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() gets被调用？

Answer 1

您可以使用模拟模块模拟函数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()

另见：

• 断言在Python单元测试中调用了一个方法
• 模拟一个类：Mock（）或patch（）？

希望有所帮助。

Answer 2

这件事可能有各种解决方案。 但首先，我必须指出，单元测试是针对测试单元的; 什么单位可以是观看事物的问题。 在你的情况下，我会说一个单位是函数a()因为你想测试你的单位行为正确，这意味着它在正确的位置调用函数b() 。 另一种观点是，函数a()和b()是一个单位。 然后，你不喜欢，检查功能是否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