python mock patch 的使用

港控/mmm° 2022-08-12 00:30 143阅读 0赞

关于python 的mock中使用patch时，遇到了一个可能出错的地方，特别记录下来。

首先使用patch.object，这里面是不会出错的。

test\_module.py

import os
    def rm(filename):
        os.remove(filename)
    
    
    def myfuction():
        return 2
    
    
    def fuction_uu():
        return myfuction()

test\_mock\_fuction.py

import nose
    from mock import MagicMock,patch
    from test_module import fuction_uu,rm
    import test_module
    import mock
    
    
    @patch.object(test_module,'myfuction')
    def test_faction_uu(mock_myfuction):
        mock_myfuction.return_value = 13
        assert  fuction_uu() == 13
        
    @patch.object(test_module,'os')
    def test_rm(mock_os):
        rm("any path")        # test that rm called os.remove with the right parameters
        mock_os.remove.assert_called_with("any path")

但是在使用patch的时候，出了问题。下面的代码在python->run的时候一切OK，但是我用nose框架的时候，就是run as unittest的时候，报错：“no module named test\_module"。

import nose
    from mock import MagicMock,patch
    from test_module import fuction_uu,rm
    import test_module
    import mock
    
    
    @patch('test_module.myfuction',MagicMock(return_value = 13))
    def test_faction_uu():
        print fuction_uu() == 13
        
    @patch('test_module.os')
    def test_rm(mock_os):
        rm("any path")        # test that rm called os.remove with the right parameters
        mock_os.remove.assert_called_with("any path")        
    
    
    if __name__ == '__main__':
        test_faction_uu()
        test_rm()

后来去查了一下mock的官方文档，才找到问题所在。第一个参数应该是”should be a string in the form‘package.module.ClassName’. “，改成下面的代码，一切OK！

import nose
    from mock import MagicMock,patch
    from test_module import fuction_uu,rm
    import test_module
    import mock
    
    
    @patch('test_mock_fuction.test_module.myfuction',MagicMock(return_value = 13))
    def test_faction_uu():
        assert fuction_uu() == 13
        
    @patch('test_mock_fuction.test_module.os')
    def test_rm(mock_os):
        rm("any path")        # test that rm called os.remove with the right parameters
        mock_os.remove.assert_called_with("any path")        
    
    
    if __name__ == '__main__':
        test_faction_uu()
        test_rm()