在單元測試中模擬python裝飾器

Question

我想測試裝飾的函數。當裝飾器已經被應用時，有沒有辦法模擬一個裝飾器和測試函數？

import mock 


def decorator(func): 
    def wrapper(*args, **kwargs): 
     return 1 
    return wrapper 


def mocked(func): 
    def wrapper(*args, **kwargs): 
     return 2 
    return wrapper 


@decorator 
def f(): 
    return 0 


with mock.patch('test.decorator') as d: 
    d.side_effect = mocked 
    assert f() == 2 # error 

Answer 1

沒有一個簡單的解決方案。

這是一個類似的問題：How to strip decorators from a function in python

您可以修改原來的代碼只是爲了測試，或使用這樣的庫：https://pypi.python.org/pypi/undecorated爲了寫一個輔助函數從原包裝轉換到測試包裝：

from undecorated import undecorated 
mocked(undecorated(f))()