“function”对象没有属性“assert”“called”“once”“with”

Question

我正在尝试使用pytest和pytest模拟运行以下测试

def rm(filename):
    helper(filename, 5)

def helper(filename):
    pass

def test_unix_fs(mocker):
    mocker.patch('module.helper')
    rm('file')
    helper.assert_called_once_with('file', 5)

但我有个例外AttributeError: 'function' object has no attribute 'assert_called_once_with'

我做错什么了？

Answer 1

在面向对象的情况下：

class Foo:
    def rm(self, filename):
        self.helper(filename, 5)

    def helper(self, filename, number):
        pass

def test_unix_fs(mocker):
    mocker.patch.object(Foo, 'helper')
    foo = Foo()
    foo.rm('file')
    helper.assert_called_once_with('file', 5)