mirror of
https://github.com/jazzband/django-axes.git
synced 2026-03-16 22:30:23 +00:00
fa7f35dda5
Use mocks and test new backends, handlers and middleware on an API call level, aiming for a 100% coverage on behaviour. Also add tests for old decorators which are not covered after moving the default authentication checks from them to the authentication backends, middleware and signal handlers. Fixes #323 Signed-off-by: Aleksi Häkli <aleksi.hakli@iki.fi>
40 lines
1.8 KiB
Python
40 lines
1.8 KiB
Python
from unittest.mock import MagicMock, patch
|
|
|
|
from django.http import HttpResponse
|
|
from django.test import TestCase
|
|
|
|
from axes.decorators import axes_dispatch, axes_form_invalid
|
|
|
|
|
|
class DecoratorTestCase(TestCase):
|
|
SUCCESS_RESPONSE = HttpResponse(status=200, content='Dispatched')
|
|
LOCKOUT_RESPONSE = HttpResponse(status=403, content='Locked out')
|
|
|
|
def setUp(self):
|
|
self.request = MagicMock()
|
|
self.cls = MagicMock(return_value=self.request)
|
|
self.func = MagicMock(return_value=self.SUCCESS_RESPONSE)
|
|
|
|
@patch('axes.decorators.is_already_locked', return_value=True)
|
|
@patch('axes.decorators.get_lockout_response', return_value=LOCKOUT_RESPONSE)
|
|
def test_axes_dispatch_locks_out(self, _, __):
|
|
response = axes_dispatch(self.func)(self.request)
|
|
self.assertEqual(response.content, self.LOCKOUT_RESPONSE.content)
|
|
|
|
@patch('axes.decorators.is_already_locked', return_value=False)
|
|
@patch('axes.decorators.get_lockout_response', return_value=LOCKOUT_RESPONSE)
|
|
def test_axes_dispatch_dispatches(self, _, __):
|
|
response = axes_dispatch(self.func)(self.request)
|
|
self.assertEqual(response.content, self.SUCCESS_RESPONSE.content)
|
|
|
|
@patch('axes.decorators.is_already_locked', return_value=True)
|
|
@patch('axes.decorators.get_lockout_response', return_value=LOCKOUT_RESPONSE)
|
|
def test_axes_form_invalid_locks_out(self, _, __):
|
|
response = axes_form_invalid(self.func)(self.cls)
|
|
self.assertEqual(response.content, self.LOCKOUT_RESPONSE.content)
|
|
|
|
@patch('axes.decorators.is_already_locked', return_value=False)
|
|
@patch('axes.decorators.get_lockout_response', return_value=LOCKOUT_RESPONSE)
|
|
def test_axes_form_invalid_dispatches(self, _, __):
|
|
response = axes_form_invalid(self.func)(self.cls)
|
|
self.assertEqual(response.content, self.SUCCESS_RESPONSE.content)
|