场景
测试一些方法时, 可能会遇到该方法需要鉴权的情况(如login_required之类), 需要想办法绕过装饰器.
解决方案
- 为装饰器加开关. 在装饰器中通过读取某个配置判断是否开启该装饰器;
- mock掉装饰器.
前者操作起来方便, 但是在遇到不同装饰器都需要mock时, 为每个装饰器添加开关有些工作量;
而后者也有弊端, 即此次测试中, 所有测试都基于该装饰器不生效的情况下测试. 因此对于如权限等的测试, 需要添加其他测试手段(如接口测试)来完成.
本文讲述的是第二个解决方案的具体实现.
技术点
Mock装饰器与普通方法的不同之处在于, 装饰器在方法声明时就已经产生影响, 而不像普通方法被调用前mock即可, 因此需要在测试执行的最开始, 还没初始化服务端代码时就mock装饰器.
写一个自己的装饰器
from functools import wraps
def mock_decorator(*args, **kwargs):
def decorator(f):
@wraps(f)
def decorated_function(*args, **kwargs):
return f(*args, **kwargs)
return decorated_function
return decorator
mock原来装饰器
在测试模块的__init__.py下执行Mock操作, 即可在运行测试前先替换原来的装饰器.
from unittest import mock
mock.patch('decoretor.path.to.mock',mock_decorator).start()
参考
[1] Can I patch a Python decorator before it wraps a function?, Stack Overflow
