2017-10-06 166 views
1

我试图模拟django.core.urlresolvers.resolve函数,但它似乎并不工作。我用自定义函数进行了测试,它的工作原理类似于魅力,但模拟忽略了完全解决了Django单元测试:不能模拟django.core.urlresolvers.resolve

代码:

test_something.py:

class SomethingTestCase(TestCase) 
    def setUp(self): 
     self.middleware = SomeMiddleware() 
     self.request = Mock() 
     self.request.session = {} 

    @patch('django.core.urlresolvers.resolve', side_effect=lambda: None) 
    def test_something(self): 
     self.assertEqual(self.middleware.process_request(self.request), None) 

middleware.py

class SomeMiddleware(object): 
    def process_request(self, request): 
     app = resolve(request.path).app_name 
     print('App name: {}'.format(app)) 

这将导致以下错误:

====================================================================== 
ERROR: test_process_request_unauthenticated (something.tests.unit.test_middleware.SomethingTestCase) 
---------------------------------------------------------------------- 
Traceback (most recent call last): 
    File "/home/user/virtualenv/something/local/lib/python2.7/site-packages/mock/mock.py", line 1305, in patched 
    return func(*args, **keywargs) 
    File "/home/user/projects/something/working/something/tests/unit/test_middleware.py", line 23, in test_process_request_unauthenticated 
    self.assertEqual(self.middleware.process_request(self.request), None) 
    File "/home/user/projects/something/working/something/middleware.py", line 14, in process_request 
    app = resolve(request.path).app_name 
    File "/home/user/virtualenv/something/local/lib/python2.7/site-packages/django/core/urlresolvers.py", line 534, in resolve 
    return get_resolver(urlconf).resolve(path) 
    File "/home/user/virtualenv/something/local/lib/python2.7/site-packages/django/core/urlresolvers.py", line 405, in resolve 
    raise Resolver404({'path': path}) 
Resolver404: {u'path': u"<Mock name='mock.path' id='140678271233168'>"} 

我的目标是使解决函数返回的东西,我可以得到应用程序名称。

为什么mock.patch无法覆盖解析函数?

回答

1

首先,您正在修补它在错误的位置。您应该将其修补到正在使用的文件中,而不是修补程序定义的位置,因为在修补程序运行之前,导入的目标代码通常都不是。其次,如果嘲讽的作品会在NoneType object has no attribute app_name的行上产生一些错误。

@patch(<path_to_middleware.py>, side_effect=lambda: type('mock', (object,), {'app_name': 1})): 
def test_something(self): 
    self.assertEqual(self.middleware.process_request(self.request), None) 
+0

哦,这是完全合理的。谢谢! – user2937998