Question

我正在测试一个继承自另一个非常复杂的类，使用DB连接方法和一堆依赖。我想模拟它的基类，以便我可以很好地使用子类中定义的方法，但是当我从一个模拟类继承时，对象本身会变成一个模拟并丢失它的所有方法。

如何模拟超类？

或多或少的情况可归纳为：

import mock

ClassMock = mock.MagicMock()

class RealClass(ClassMock):

    def lol(self):
        print 'lol'

real = RealClass()
real.lol()  # Does not print lol, but returns another mock

print real # prints <MagicMock id='...'>

这是一个简化的案例。实际发生的是RealClass扩展AnotherClass，但我设法拦截AnotherClass并将其替换为模拟。

Answer 1

这是我长期以来一直在努力的事情，但我想我终于找到了解决方案。

正如您已经注意到的，如果您尝试用Mock替换基类，那么您尝试测试的类只会变成模拟，这会使您无法测试它。解决方案是仅模拟基类的方法而不是整个基类本身，但说起来容易做起来难：在测试中逐个模拟每个方法都非常容易出错按试验依据。

我所做的是创建一个扫描另一个类的类，并为自己Mock()分配与另一个类的方法匹配的类。然后，您可以在测试中使用此类代替真正的基类。

这是假类：

class Fake(object):
    """Create Mock()ed methods that match another class's methods."""

    @classmethod
    def imitate(cls, *others):
        for other in others:
            for name in other.__dict__:
                try:
                    setattr(cls, name, Mock())
                except (TypeError, AttributeError):
                    pass
        return cls

所以例如你可能有一些像这样的代码（道歉这有点人为，只是假设BaseClass和SecondClass做了非平凡的工作并且包含很多方法并且不是＆＃39 ;甚至必须由你定义）：

class BaseClass:
    def do_expensive_calculation(self):
        return 5 + 5

class SecondClass:
    def do_second_calculation(self):
        return 2 * 2

class MyClass(BaseClass, SecondClass):
    def my_calculation(self):
        return self.do_expensive_calculation(), self.do_second_calculation()

然后你可以编写一些这样的测试：

class MyTestCase(unittest.TestCase):
    def setUp(self):
        MyClass.__bases__ = (Fake.imitate(BaseClass, SecondBase),)

    def test_my_methods_only(self):
        myclass = MyClass()
        self.assertEqual(myclass.my_calculation(), (
            myclass.do_expensive_calculation.return_value, 
            myclass.do_second_calculation.return_value,
        ))
        myclass.do_expensive_calculation.assert_called_once_with()
        myclass.do_second_calculation.assert_called_once_with()

因此，基类上存在的方法仍然可以作为可以与之交互的模拟，但是您的类本身不会成为模拟。

我一直小心翼翼地确保它在python2和python3中都有效。

Answer 2

这应该适合你。

import mock

ClassMock = mock.MagicMock

class RealClass(ClassMock):

    def lol(self):
        print 'lol'

real = RealClass()
real.lol()  # Does not print lol, but returns another mock

print real # prints <MagicMock id='...'>

你应该像你一样传递一个类的实例。 mock.MagicMock是一个类，所以你直接传递它。

In [2]: inspect.isclass(mock.MagicMock)
Out[2]: True

Python mock：用于继承的模拟基类

2 个答案: