使用类装饰器，如何在不重新定义类的情况下覆盖方法？

Question

对于使用App Engine testbed 的单元测试（使用unittest 模块），我需要setUp 和tearDown 方法来分别激活和停用测试台（稍微简化）：

class SomeTest(unittest.TestCase):

  def setUp(self):
    self.testbed = testbed.Testbed()
    self.testbed.activate()

  def tearDown(self):
    self.testbed.deactivate()

  def testSomething(self):
    ...

这很快就会成为写作的负担。我可以编写一个基类TestCaseWithTestbed，但是每次我在一个测试用例中需要一个自定义setUp 时，我都必须记住调用超类方法。

我认为用类装饰器来解决这个问题会更优雅。所以我想写：

@WithTestbed
class SomeTest(unittest.TestCase):

  def testSomething(self):
    ...

应用此装饰器后，测试平台应该会神奇地被激活。那么...如何实现WithTestbed 装饰器？我目前有以下：

def WithTestbed(cls):
  class ClsWithTestbed(cls):

    def setUp(self):
      self.testbed = testbed.Testbed()
      self.testbed.activate()
      cls.setUp(self)

    def tearDown(self):
      cls.tearDown(self)
      self.testbed.deactivate()

  return ClsWithTestbed

这适用于简单的情况，但有一些严重的问题：

• 测试类的名称变为ClsWithTestbed，并显示在测试输出中。
• 调用super(SomeTestClass, self).setUp() 的具体测试类最终会无限递归，因为SomeTestClass 现在等于WithTestbed。

我对 Python 的运行时类型操作有点模糊。那么，如何以正确的方式做到这一点？

Answer 1

这样的事情会起作用：

def WithTestbed(cls):
    cls._post_testbed_setUp = getattr(cls, 'setUp', lambda self : None)
    cls._post_testbed_tearDown = getattr(cls, 'tearDown', lambda self : None)

    def setUp(self):
        self.testbed = testbed.Testbed()
        self.testbed.activate()
        self._post_testbed_setUp()

    def tearDown(self):
        self.testbed.deactivate()
        self._post_testbed_tearDown()

    cls.setUp = setUp
    cls.tearDown = tearDown
    return cls

@WithTestbed
class SomeTest(object):
    ...

Answer 2

这是一种简单的方法，可以使用子类而不是装饰器来完成您的要求：

class TestCaseWithTestBed(unittest.TestCase):

  def setUp(self):
    self.testbed = testbed.Testbed()
    self.testbed.activate()
    self.mySetUp()

  def tearDown(self):
    self.myTearDown()
    self.testbed.deactivate()

  def mySetUp(self): pass
  def myTearDown(self): pass

class SomeTest(TestCaseWithTestBed):
  def mySetUp(self):
    "Insert custom setup here"

您所要做的就是在测试用例中定义mySetUp 和myTearDown，而不是setUp 和tearDown。

Answer 3

这似乎可以解决问题：

def WithTestbed(cls):
  def DoNothing(self):
    pass

  orig_setUp = getattr(cls, 'setUp', DoNothing)
  orig_tearDown = getattr(cls, 'tearDown', DoNothing)

  def setUp(self):
    self.testbed = testbed.Testbed()
    self.testbed.activate()
    orig_setUp(self)
  def tearDown(self):
    orig_tearDown(self)
    self.testbed.deactivate()

  cls.setUp = setUp
  cls.tearDown = tearDown
  return cls

有人发现这种方法有什么问题吗？