Question

我正在使用mock Python模块执行测试。

有时候我在模拟一个类，但是我只想模拟它的一些方法和属性，而不是全部。

假设以下情况：

# module.py
class SomeClass:
    def some_method(self):
        return 100

    def another_method(self):
        return 500

# test.py
class Tests(unittest.TestCase):
    @patch('module.SomeClass')
    def test_some_operation(self, some_class_mock):
        some_class_instance = some_class_mock.return_value

        # I'm mocking only the some_method method.
        some_class_instance.some_method.return_value = 25

        # This is ok, the specific method I mocked returns the value I wished.
        self.assertEquals(
            25,
            SomeClass().some_method()
        )

        # However, another_method, which I didn't mock, returns a MagicMock instance
        # instead of the original value 500
        self.assertEquals(
            500,
            SomeClass().another_method()
        )

在上面的代码中，我修补了SomeClass类后，将调用其return_values 我没有明确设置将返回MagicMock个对象。

我的问题是：我该如何仅模拟某些类方法而又不影响其他类？

我可以想到两种方法，但是没有一种是真的好。

一种方法是将模拟方法设置为原始类方法，如下所示：
```
some_class_instance.another_method = SomeClass.another_method
```
这并不是真正希望的，因为该类可能具有很多方法和属性来 “取消模拟”。

另一种方法是显式修补我想要的每个方法，例如：

 @patch('module.SomeClass.some_method')
 def test_some_operation(self, some_method_mock):

但是，如果我想模拟类本身，因为它无法模拟对例如初始化器。无论如何，下面的代码将覆盖所有SomeClass的方法。

 @patch('module.SomeClass.some_method')
 @patch('module.SomeClass')
 def test_some_operation(self, some_class_mock, some_method_mock):

这是一个更具体的示例：

class Order:
    def process_event(self, event, data):
        if event == 'event_a':
            return self.process_event_a(data)

        elif event == 'event_b':
            return self.process_event_b(data)

        else:
            return None

    def process_event_a(self, data):
        # do something with data

    def process_event_b(self, data):
        # do something different with data

在这种情况下，我有一个通用方法process_event，它根据所提供的事件来调用特定的处理事件。

我只想测试方法process_event。我只想知道是否根据我提供的事件调用了适当的特定事件。

因此，在我的测试案例中，我想要做的是仅模拟process_event_a和process_event_b，使用特定参数调用原始的process_event，然后断言{{1} }或process_event_a用适当的参数调用。

Answer 1

必须修补对象，而不是修补整个类。即，创建您的类的实例，然后修补该实例的方法。

请注意，您也可以使用装饰器class SomeClass: def some_method(self): return 100 def another_method(self): return 500代替我的方法。

test.py

在您的from unittest import mock class Tests(unittest.TestCase): def test_some_operation(self): some_class_instance = SomeClass() # I'm mocking only the some_method method. with mock.patch.object(some_class_instance, 'some_method', return_value=25) as cm: # This is gonna be ok self.assertEquals( 25, SomeClass().some_method() ) # The other methods work as they were supposed to. self.assertEquals( 500, SomeClass().another_method() )

    def create_card_on_the_board_call(self, id_list, card_count, card_name):
    global create_card
    for i in range(card_count):
        create_card = self.rest_api_helper.post_call(
            self.base_url + '/cards?' + 'key=' + test_config[
                'key'] + '&' + 'token=' + test_config[
                'token'] + '&' + f'idList={id_list}' + '&' + f'name={card_name}', headers=None)
    return create_card.json()

模拟Python类的方法和属性的子集

1 个答案: