如何使用python mock来删除方法装饰器的行为,以便我不依赖它们来测试当前方法?
import utils
class Router(object):
@utils.with_user
@utils.formatted_response('resources', with_pagination=True)
def get_resources(self, user_id=None, offset=None, limit=None):
# do stuff
pass
我已经尝试过了:
# @patch('utils.with_tenant')
# @patch.object(utils, 'with_tenant')
def test_stub_decorator(self):
# patch('utils.with_tenant')
# patch.object(utils, 'with_tenant')
这些似乎都不起作用!有什么想法吗?
答案 0 :(得分:1)
对于模拟装饰器,您需要提前模拟它们(在加载该模块之前):
from mock import patch
# mock the retry decorator before any module loads it
patch('utils.with_user', lambda x: x).start()
import utils # or any module which imports utils