Я пишу некоторые модульные тесты для проекта Django, и мне было интересно, возможно ли его (или необходимо?) проверить некоторые из декораторов, которые я написал для него.
Вот пример декоратора, который я написал:
class login_required(object):
def __init__(self, f):
self.f = f
def __call__(self, *args):
request = args[0]
if request.user and request.user.is_authenticated():
return self.f(*args)
return redirect('/login')
Просто:
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.
Декоратор, подобный этому, может быть протестирован просто благодаря утиной печати. Просто поставьте макет объекта на функцию вызов, которая, кажется, удерживает и действует как запрос, и посмотрите, получаете ли вы ожидаемое поведение.
Когда нужно использовать модульные тесты, вполне понятно, я бы сказал. В приведенном примере содержится такой базовый код, который можно сказать, что это необязательно. Но опять же, стоимость тестирования такого класса одинаково низкая.
class TestCaseExample(TestCase):
def test_decorator(self):
request = HttpRequest()
# Set the required properties of your request
function = lambda x: x
decorator = login_required(function)
response = decorator(request)
self.assertRedirects(response)
В общем, подход, который я использовал, следующий: