Mock在Python单元测试中的使用(2)

之前定义的 rm 方法相当的简单.在盲目地删除之前,我们倾向于验证一个路径是否存在,并验证其是否是一个文件.让我们重构 rm 使其变得更加智能：

1. #!/usr/bin/env python
2. # -*- coding: utf-8 -*-
4. import os
5. import os.path
7. def rm(filename):
8. ? ?if os.path.isfile(filename):
9. ? ? ? ?os.remove(filename)

很好.现在,让我们调整测试用例来保持测试的覆盖率.

1. #!/usr/bin/env python
2. # -*- coding: utf-8 -*-
4. from mymodule import rm
6. import mock
7. import unittest
9. class RmTestCase(unittest.TestCase):
11. ? ?@mock.patch('mymodule.os.path')
12. ? ?@mock.patch('mymodule.os')
13. ? ?def test_rm(self, mock_os, mock_path):
14. ? ? ? ?# set up the mock
15. ? ? ? ?mock_path.isfile.return_value = False
17. ? ? ? ?rm("any path")
19. ? ? ? ?# test that the remove call was NOT called.
20. ? ? ? ?self.assertFalse(mock_os.remove.called, "Failed to not remove the file if not present.")
22. ? ? ? ?# make the file 'exist'
23. ? ? ? ?mock_path.isfile.return_value = True
25. ? ? ? ?rm("any path")
27. ? ? ? ?mock_os.remove.assert_called_with("any path")

我们的测试用例完全改变了.现在我们可以在没有任何副作用的情况下核实并验证方法的内部功能.

将文件删除作为服务

到目前为止,我们只是将 mock 应用在函数上,并没应用在需要传递参数的对象和实例的方法上.我们现在开始涵盖对象的方法.

首先,我们将?rm?方法重构成一个服务类.实际上将这样一个简单的函数转换成一个对象,在本质上这不是一个合理的需求,但它能够帮助我们了解?mock?的关键概念.让我们开始重构：

1. #!/usr/bin/env python
2. # -*- coding: utf-8 -*-
4. import os
5. import os.path
7. class RemovalService(object):
8. ? ?"""A service for removing objects from the filesystem."""
10. ? ?def rm(filename):
11. ? ? ? ?if os.path.isfile(filename):
12. ? ? ? ? ? ?os.remove(filename)

你会注意到我们的测试用例没有太大变化：

1. #!/usr/bin/env python
2. # -*- coding: utf-8 -*-
4. from mymodule import RemovalService
6. import mock
7. import unittest
9. class RemovalServiceTestCase(unittest.TestCase):
11. ? ?@mock.patch('mymodule.os.path')
12. ? ?@mock.patch('mymodule.os')
13. ? ?def test_rm(self, mock_path):
14. ? ? ? ?# instantiate our service
15. ? ? ? ?reference = RemovalService()
17. ? ? ? ?# set up the mock
18. ? ? ? ?mock_path.isfile.return_value = False
20. ? ? ? ?reference.rm("any path")
22. ? ? ? ?# test that the remove call was NOT called.
23. ? ? ? ?self.assertFalse(mock_os.remove.called, "Failed to not remove the file if not present.")
25. ? ? ? ?# make the file 'exist'
26. ? ? ? ?mock_path.isfile.return_value = True
28. ? ? ? ?reference.rm("any path")
30. ? ? ? ?mock_os.remove.assert_called_with("any path")

很好,我们知道?RemovalService?会如预期般的工作.接下来让我们创建另一个服务,将?RemovalService?声明为它的一个依赖：

1. #!/usr/bin/env python
2. # -*- coding: utf-8 -*-
4. import os
5. import os.path
7. class RemovalService(object):
8. ? ?"""A service for removing objects from the filesystem."""
10. ? ?def rm(self, filename):
11. ? ? ? ?if os.path.isfile(filename):
12. ? ? ? ? ? ?os.remove(filename)
15. class UploadService(object):
17. ? ?def __init__(self, removal_service):
18. ? ? ? ?self.removal_service = removal_service
20. ? ?def upload_complete(self, filename):
21. ? ? ? ?self.removal_service.rm(filename)

（编辑：ASP站长网）