Mock在Python单元测试中的使用

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

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)

（编辑：网站开发网_盐城站长网 ）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!