Python-mock：如何测试是否调用了 super()

Question

我有以下结构：

class A(Object):
    def method(self):
        return 'a'

class B(A):
   def __init__(self, test):
       self.test = test

   def method(self):
        if self.test:
           return super(A, self).method(self)
        else:
           return 'b'

我想要做的是编写一个测试用例，它会测试如果 self.test 为真，则调用 super 函数并调用类 A 的方法函数。

我怎么能做到这一点？ 我应该嘲笑什么？

进阶问题：当 A 类和 B 类在一个单独的模块中，并且它们具有相同的名称时，该怎么办？ 因此，不是类 BI 会写： class A(module1.A): 这会改变模拟吗？

Answer 1

正如@MartijnPieters所指出的，测试超级调用通常是对实现细节的测试，而不是对合同的测试。 尽管如此，测试特定的实现仍然是可取的 - 或者父类可能有一个需要调用super的合同。 要测试super被调用，请使用@mgilson提到的模拟 。 答案详细说明：

import mock, unittest

class TestB(unittest.TestCase):

    @mock.patch("A.method")
    def test_super_method(self, mock_super):
        B(True).method()
        self.assertTrue(mock_super.called)

    @mock.patch("A.method")
    def test_super_method(self, mock_super):
        B(False).method()
        self.assertFalse(mock_super.called)

您需要在补丁中指定完整的命名空间。 例如， @mock.patch("module.submodule.A.method") 。 这可以用于A中的任何方法，包括__init__ ; 语法完全一样。

Answer 2

为了建立@Malina 的出色答案，如果您无法修补模块 A 上的方法（例如，A 位于第三方库中、讨厌的遗留系统的一部分、在运行时动态创建等），您仍然可以对此进行测试通过在类 B 上显式创建另一个方法来测试功能。

修订B类

import A

class B(A):
   def __init__(self, test):
       self.test = test

   def method(self):
        if self.test:
           self._delegate_to_A()  # Move the super call to a separate method
        else:
           return 'b'
   
   def _delegate_to_A(self)
       return super(A, self).method(self)

修订B类测试代码

import mock, unittest

class TestB(unittest.TestCase):

    # Patch B's `delegate` method instead of A
    @mock.patch("B._delegate_to_A")
    def test_super_method(self, mock_super):
        B(True).method()
        self.assertTrue(mock_super.called)

    @mock.patch("B._delegate_to_A")
    def test_super_method(self, mock_super):
        B(False).method()
        self.assertFalse(mock_super.called)