【脚本语言系列】关于Python测试工具mock, 你需要知道的事

如何使用mock

假设求阶乘会引起一系列的反应，导致发生核灾难的后果；使用mock包，用一个模拟对象来模拟阶乘计算。

1. 安装mock

easy_install mock
pip install mock

2. 编写阶乘函数

class NuclearReactor():
    def __init__(self, n):
        self.n = n

    def do_work(self, msg):
        print "Working"

        return self.factorial(self.n, msg)

    def factorial(self, n, msg):
        print msg

        if n == 0:
            return 1

        if n < 0:
            raise ValueError, "Core meltdown"

        return numpy.arange(1, n+1).cumprod()

3. 创建一个模拟对象
   核反应堆类有一个do_work方法，该方法会调用危险的factorial方法；
   这将确保模拟对象被调用后，其返回值是6.

reactor.factorial = MagicMock(return_value=6)

class NuclearReactorTest(unittest.TestCase):
    def test_called(self):
        reactor = NuclearReactor(3)
        reactor.factorial = MagicMock(return_value= 6)
        result = reactor.do_work("mocked")
        self.assertEqual(6, result)
        reactor.factorial.assert_called_with(3, "mocked")

    def test_unmocked(self):
        reactor = NuclearReactor(3)
        reactor.factorial(3, "unmocked")
        numpy.testing.assert_raises(ValueError)

4. 断言行为
   可以用若干种方法，检查模拟对象的行为，进而测试真实对象的行为。
   可以断言，用正确的参数调用了潜在的有爆炸危险的factorial方法。

self.assertEqual(6, result)
reactor.factorial.assert_called_with(3, "mocked")

5. 运行结果

   1. 测试代码
      

   2. 测试结果
      

什么是mock

模拟对象(mock)是真实对象的替代物，用来测试真实对象的部分行为。
一般来说，只有在不方便创建真实对象（例如数据库连接）时，或者对真实对象的测试会产生不希望有的副作用时，才需要用到模拟对象。
例如，我们可能不希望在测试过程中，把数据写入到文件系统或数据库中。