python装饰器测试_测试Python装饰器？

简单地说：from nose.tools import assert_equal

from mock import Mock

class TestLoginRequired(object):

def test_no_user(self):

func = Mock()

decorated_func = login_required(func)

request = prepare_request_without_user()

response = decorated_func(request)

assert not func.called

# assert response is redirect

def test_bad_user(self):

func = Mock()

decorated_func = login_required(func)

request = prepare_request_with_non_authenticated_user()

response = decorated_func(request)

assert not func.called

# assert response is redirect

def test_ok(self):

func = Mock(return_value='my response')

decorated_func = login_required(func)

request = prepare_request_with_ok_user()

response = decorated_func(request)

func.assert_called_with(request)

assert_equal(response, 'my response')

mock库在这里有帮助。