단위 테스트를위한 데코레이터 생략하기

Question

데코레이터 @auth은 기본적으로 데이터베이스를 검사하여 사용자가 주어진 REST 호출에 액세스 할 수 있는지 확인합니다. 이 호출에 대한 단위 테스트를 작성하고 싶습니다. 나의 초기 아이디어는 단순히 장식자를 아무 것도하지 않는 패스로 넘겨주는 것이었다. (내 초기 아이디어는 실패 했으므로 항상 기능을 수행 할 수 있도록 @auth의 일부 기능을 monkeypatch 할 수 있습니다. 그러나 장식자를 완전히 무시할 수 있다면 여전히 궁금합니다.)

나는 달성하기를 희망합니다.

example.py

# example.py 
from __future__ import print_function 

def sample_decorator(func): 
    def decorated(*args, **kwargs): 
     print("Start Calculation") 
     ans = func(*args, **kwargs) + 3 
     print(ans) 
     print("Finished") 
     return ans 
    return decorated 

@sample_decorator 
def add(a, b): 
    return a + b 

test_example.py

# test_example.py 
from __future__ import print_function 
import pytest 

import example 

def test_add_with_decorator(): 
    assert example.add(1, 1) == 5 

def testadd_with_monkeypatch_out_decorator(monkeypatch): 
    monkeypatch.setattr(example, 'sample_decorator', lambda func: func) 
    assert example.add(1, 1) == 2 # this fails, but is the behaviour I want 

이러한 목표를 달성하기 위해 몇 가지 정직 방법이 있나요?

def raw_add... 

add = sample_decorator(raw_add) 

assert example.raw_add... 

Answer 1

데코레이터은, 랩 된 기능에 대한 액세스 권한을 부여 포장 기능에 대한 속성을 설정할 수 있습니다

Answer 2

는 그냥 정의가 별도로 보관.

모듈이로드 될 때 함수를 래핑

def wrap_foo(func): 
def decorated(*args, **kwargs): 
    func(*args, **kwargs) 
decorated.__wrapped_fn = func 
return decorated 

@wrap_foo 
def foo(): 
pass 

# Wrapped 
foo() 

# Unwrapped 
foo.__wrapped_fn()