test try_except_deco

test_try_except_decorator.py

from unittest.mock import MagicMock
from try_except_decorator import try_except

def test_valid():
    """
    例外が起きなかった場合にデコレートされた関数の戻り値が返ること
    """
    mock_func = MagicMock(return_value='correct')  # こんな感じで定義時に与えることもできます
    result = try_except(exreturn_value='excepted')(mock_func)()  # 3回の呼び出し
    assert result == 'correct', "デコレートされた関数の戻り値が返る"

def test_invalid():
    """
    例外が起きた時にデコレータに渡した引数が返ること
    """
    mock_func = MagicMock(return_value='correct')
    mock_func.side_effect = Exception()  # 例外が起きるようにする
    result = try_except(exreturn_value='excepted')(mock_func)()
    assert result == 'excepted', "デコレータに渡した引数が返る"