从TestCase派生一个类会引发两个错误

Question

我有一些基本的设置/拆卸代码，我想在一大堆单元测试中重用它们。因此，我想到了创建一些派生类以避免在每个测试类中重复代码的好主意。

这样，我收到了两个奇怪的错误。第一，我无法解决。这是无法解决的一个：

AttributeError: 'TestDesktopRootController' object has no attribute '_testMethodName'

这是我的基类：

import unittest
import twill
import cherrypy
from cherrypy._cpwsgi import CPWSGIApp


class BaseControllerTest(unittest.TestCase):

    def __init__(self):
        self.controller = None

    def setUp(self):
        app = cherrypy.Application(self.controller)

        wsgi = CPWSGIApp(app)

        twill.add_wsgi_intercept('localhost', 8080, lambda : wsgi)

    def tearDown(self):
        twill.remove_wsgi_intercept('localhost', 8080)

这是我的派生类：

import twill
from base_controller_test import BaseControllerTest

class TestMyController(BaseControllerTest):

    def __init__(self, args):
        self.controller = MyController()
        BaseControllerTest.__init__(self)

    def test_root(self):
        script = "find 'Contacts'"
        twill.execute_string(script, initial_url='http://localhost:8080/')

另一个奇怪的错误是：

TypeError: __init__() takes exactly 1 argument (2 given)

解决方案是__init__在派生类的函数中添加单词“ args” 。有什么办法可以避免这种情况？

记住，我在这一方面有两个错误。

Answer 1

这是因为您__init__()不正确地重写了。几乎可以肯定，您根本不想覆盖__init__()。你应该做一切setUp()。我已经使用unittest了10多年了，我认为自己从未被超越__init__()。

但是，如果确实需要重写__init__()，请记住，您不控制构造函数的调用位置，而是框架为您调用。因此，您必须提供一个可以调用的签名。从源代码（unittest/case.py），该签名为：

def __init__(self, methodName='runTest'):

执行此操作的安全方法是接受任何参数，然后将它们传递给基类。这是一个有效的实现：

class BaseTest(unittest.TestCase):
    def __init__(self, *args, **kwargs):
        unittest.TestCase.__init__(self, *args, **kwargs)

    def setUp(self):
        print "Base.setUp()"

    def tearDown(self):
        print "Base.tearDown()"


class TestSomething(BaseTest):
    def __init__(self, *args, **kwargs):
        BaseTest.__init__(self, *args, **kwargs)
        self.controller = object()

    def test_silly(self):
        self.assertTrue(1+1 == 2)