Python unittest -- 在我的 setUpClass 方法中使用父类方法？

Question

我正在编写单元测试并具有以下结构：

class TestSuite(unittest.TestCase, HelperClass):
  @classmethod
  def setUpClass(cls):
    # I want to use methods from HelperClass here, but get errors
    # I have tried cls.method_name() and self.method_name()

  def setUp(self):
    self.method_name() # methods from HelperClass work here
...

正如我的评论中所指出的，我想在我的 setUpClass 方法中使用 HelperClass 中的一些验证方法，但是如果我尝试使用 cls.method_name() 调用，我会得到TypeError: method_name() missing 1 required positional argument: 'self' ，但如果我使用 self.method_name()，我会得到NameError: name 'self' is not defined 。

这可能是非常基本的东西，但我不确定我应该用来找到答案的正确搜索词。 不幸的是，有关 setUpClass 的 unittest 文档也没有涉及。

Answer 1

问题是，在setUpClass ，你根本就没有的一个实例TestSuite上，你可以调用定义的实例方法HelperClass 。 测试运行器按照以下方式做一些事情

TestSuite.setUpClass()

t = TestSuite(...)

t.setUp()
t.test_foo()
t.tearDown()

t.setUp()
t.test_bar()
t.tearDown()

在您创建将用于调用类中定义的每个测试的单个实例之前， setUpClass被调用一次，没有TestSuite的实例。 setUp和tearDown方法分别在每次测试之前和之后调用。

您可能想要的是覆盖TestSuite.__init__ ，每个实例调用一次，而不是在调用每个测试方法之前。

def __init__(self, *args, **kwargs):
    super().__init__(*args, **kwargs)
    self.method_name()

Answer 2

我不知道这是否是正确的解决方案，但我找到了一种最终非常明显的方法。

class TestSuite(unittest.TestCase, HelperClass):
  @classmethod
  def setUpClass(cls):
    hc = HelperClass()
    hc.method_name()

  def setUp(self):
    self.method_name() # methods from HelperClass work here
...

回想起来，我应该马上看到这个，哈哈。 想我会分享给有人也在寻找有一天。