使用类装饰器，如何重写方法而不重新定义类？

Question

单元测试（使用unittest模块）使用App Engine testbed，我需要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): 
    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 

有谁看到使用这种方法的任何问题？

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): 
    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): 
    ...