调用本类方法如何mock

在Python中，如果你想要mock一个类的某个方法，你可以使用`unittest.mock`模块中的`Mock`类或者`unittest.mock.patch`装饰器。以下是如何使用这两种方法来mock一个类的方法的示例。

使用`unittest.mock.patch`装饰器

`unittest.mock.patch`装饰器可以用来mock一个类的属性或方法。以下是一个简单的例子：

```python

from unittest.mock import patch

class MyClass:

def my_method(self):

return "original result"

使用patch装饰器来mock MyClass的my_method方法

with patch('module.MyClass.my_method') as mock_method:

mock_method.return_value = "mocked result"

result = MyClass().my_method()

print(result) 输出: mocked result

```

在这个例子中，`patch`装饰器被用来mock `MyClass` 的 `my_method` 方法，使得每次调用该方法时都会返回 `"mocked result"`。

使用`unittest.mock.Mock`类

你也可以创建一个`Mock`对象并直接设置它的`return_value`属性来mock一个方法：

```python

from unittest.mock import Mock

class MyClass:

def my_method(self):

return "original result"

创建一个Mock对象

mock_instance = Mock(spec=MyClass)

设置my_method的return_value

mock_instance.my_method.return_value = "mocked result"

使用mock实例调用方法

result = mock_instance.my_method()

print(result) 输出: mocked result

```

在这个例子中，我们创建了一个`Mock`对象，指定它的`spec`属性为`MyClass`，这样它就会模仿`MyClass`的行为。然后我们设置了`my_method`的`return_value`属性，使得每次调用该方法时都会返回 `"mocked result"`。

选择哪种方法取决于你的具体需求和个人偏好。`unittest.mock.patch`装饰器更简单，适合快速mock一个方法，而`Mock`类提供了更多的灵活性，可以用来mock整个实例。