python - 如何在另一个函数中模拟一个函数？

Question

我一直试图在另一个函数中模拟这个函数调用，但没有成功。我如何成功地模拟这个？

from mock import patch
from path.to.a import function_a

@patch("class_b.function_c")
def test_method(self, method_to_mock):
    method_to_mock.return_value = 7890
    result = function_a() #error - type object 'class_b' has no attribute 'function_c'

#another module -- "path.to.a module"
def function_a():
    return class_b.function_c()

#another module
class class_b(class_c):
    pass

#another module
class class_c():
    @classmethod
    def function_c():
        return 123

Answer 1

您的代码有两个问题：

1）类方法没有正确声明

class class_c():
    @classmethod
    def function_c(cls):
        return 123

2) @patch 使用不正确。您需要将其更改为

def mock_method(cls):
    return 7890

# asssume the module name of class_b is modb
@patch("modb.class_b.function_c", new=classmethod(mock_method))
def test_method():
    result = function_a() 
    print result # check the result