如何在类的__init__方法中修补实例?

时间:2015-03-20 00:17:00

标签: python unit-testing mocking tdd webob

我正在写一个单元测试。如何在MyValidator类的 init 方法中修补self.conf?在我的unittest中,我想创建一个假的self.conf并获取响应以在self.conf中对每个元素进行断言。

class MyValidator(wsgi.Middleware):

    def __init__(self, app):
        self.app = app
        self.conf = {
            'auth_uri': CONF.someuri
            'admin_domain_name': CONF.somedomainname,
            'admin_user': CONF.someuser,
            'admin_password': CONF.get_admin_password(),
            'domain_name': CONF.somedomainname
        }

对于unittest,我正在考虑......(我知道这是错误的......但是你明白了)

@mock.patch('my_module.MyValidator.__init__.conf')
def setUp(self, mock_config):
    @webob.dec.wsgify()
    def fake_app(req):
        return webob.Response()
    self.request = webob.Request.blank('/')
    mock_config = {
        'auth_uri': 'testuri'
         ....
         ....
    }
    self.middleware = MyValidator(fake_app)

def test_auth_uri(self):
    auth_uri = 'testuri'
    env_auth_uri = self.request.environ.get('auth_uri', None)
    self.assertEqual(auth_uri, env_auth_uri)

如何修补self.conf以获得预期的响应?

1 个答案:

答案 0 :(得分:0)

即使我经常使用嘲弄和修补,我也不认为你的情况需要它。 confMyValidator的公共属性,除了您需要的更改之外,您不需要任何其他内容。

def setUp(self):
    @webob.dec.wsgify()
    def fake_app(req):
        return webob.Response()
    self.request = webob.Request.blank('/')
    self.middleware = MyValidator(fake_app)
    self.middleware.conf = {
        'auth_uri': 'testuri'
         ....
         ....
    }

在这种情况下,patch不能再给你了,因为你对dict访问和一些更小的上下文不感兴趣。如果在某些其他测试中您需要其他值,则可以使用self.middleware.conf.update(...)self.middleware.conf[...]=...而不更改其他测试的行为,因为setUp()以相同的方式为每个测试配置它。

如果confMyValidator的只读属性(设计稍好一些),情况会有所不同。在这种情况下,您需要对其进行修补以进行测试:

class MyValidator(wsgi.Middleware):
    def __init__(self, app):
        self.app = app

    @property
    def conf(self):
        return {
            'auth_uri': CONF.someuri
            'admin_domain_name': CONF.somedomainname,
            'admin_user': CONF.someuser,
            'admin_password': CONF.get_admin_password(),
            'domain_name': CONF.somedomainname
        }

测试类应该在哪里

@mock.patch('my_module.MyValidator.conf', new_callable=PropertyMock)
def setUp(self, mock_conf):
    @webob.dec.wsgify()
    def fake_app(req):
        return webob.Response()
    self.request = webob.Request.blank('/')
    mock_conf.return_value = {
        'auth_uri': 'testuri'
         ....
         ....
    }
    self.middleware = MyValidator(fake_app)

def test_auth_uri(self):
    auth_uri = 'testuri'
    env_auth_uri = self.request.environ.get('auth_uri', None)
    self.assertEqual(auth_uri, env_auth_uri)

补丁__init__()在极少数情况下很有用。例如,当__init__做一些工作并需要无法使用或不会在测试环境中使用的资源时。它的一个例子是当init尝试访问数据库,使用网络资源,启动新线程等等。