14

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

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

4

2 回答 2

21

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()

另见：

希望有帮助。

于 2013-09-12T11:04:24.833 回答
1

这件事有多种解决方案。但首先，我必须指出，单元测试是为了测试单元。单位是什么可能是查看事物的问题。在您的情况下，我会说一个单元是函数a()，因为您想测试您的单元是否正确运行，这意味着它b()在正确的位置调用函数。一个不同的观点是说函数a()和b()是一个单元。然后你不喜欢检查函数是否b()被调用，你只想测试函数调用的结果a()。因此，请决定最适合您的情况。

如果您真的想将功能a()作为一个单元进行测试，您应该准备好您的单元进行测试。在您的情况下，可以通过向其添加一个附加参数（默认为 function b）来完成，并将使用该参数而不是硬编码的函数b()调用 in 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
于 2013-09-12T11:17:00.927 回答