fbf7d0e43d
* django templates * widgets * views * mppt * main_menu * projector 1.x api Sorted all imports Add a ending slash to each url with a redirect view
139 lines
4.6 KiB
Python
139 lines
4.6 KiB
Python
from unittest import TestCase
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from django.core.exceptions import PermissionDenied
|
|
|
|
from openslides.utils import views
|
|
|
|
|
|
class PermissionMixinTest(TestCase):
|
|
def test_check_permission_non_required_permission(self):
|
|
view = views.PermissionMixin()
|
|
view.required_permission = None
|
|
request = MagicMock()
|
|
|
|
self.assertTrue(view.check_permission(request))
|
|
|
|
def test_check_permission_with_required_permission(self):
|
|
view = views.PermissionMixin()
|
|
view.required_permission = 'required_permission'
|
|
request = MagicMock()
|
|
|
|
view.check_permission(request)
|
|
|
|
request.user.has_perm.assert_called_once_with('required_permission')
|
|
|
|
@patch('builtins.super')
|
|
def test_dispatch_with_perm(self, mock_super):
|
|
view = views.PermissionMixin()
|
|
view.check_permission = MagicMock(return_value=True)
|
|
request = MagicMock()
|
|
|
|
view.dispatch(request)
|
|
|
|
mock_super().dispatch.called_once_with(request)
|
|
|
|
@patch('openslides.utils.views.settings')
|
|
@patch('openslides.utils.views.HttpResponseRedirect')
|
|
@patch('builtins.super')
|
|
def test_dispatch_without_perm_logged_out(self, mock_super, mock_response, mock_settings):
|
|
view = views.PermissionMixin()
|
|
view.check_permission = MagicMock(return_value=False)
|
|
request = MagicMock()
|
|
request.user.is_authenticated.return_value = False
|
|
request.get_full_path.return_value = '/requested/path/'
|
|
mock_settings.LOGIN_URL = 'my_login_url'
|
|
|
|
value = view.dispatch(request)
|
|
|
|
mock_response.assert_called_once_with('my_login_url?next=/requested/path/')
|
|
self.assertEqual(value, mock_response())
|
|
|
|
@patch('openslides.utils.views.settings')
|
|
@patch('openslides.utils.views.HttpResponseRedirect')
|
|
@patch('builtins.super')
|
|
def test_dispatch_without_perm_logged_in(self, mock_super, mock_response, mock_settings):
|
|
view = views.PermissionMixin()
|
|
view.check_permission = MagicMock(return_value=False)
|
|
request = MagicMock()
|
|
request.user.is_authenticated.return_value = True
|
|
|
|
with self.assertRaises(PermissionDenied):
|
|
view.dispatch(request)
|
|
|
|
|
|
@patch('builtins.super')
|
|
class SingleObjectMixinTest(TestCase):
|
|
def test_get_object_cache(self, mock_super):
|
|
"""
|
|
Test that the method get_object caches his result.
|
|
|
|
Tests that get_object from the django view is only called once, even if
|
|
get_object on our class is called twice.
|
|
"""
|
|
view = views.SingleObjectMixin()
|
|
|
|
view.get_object()
|
|
view.get_object()
|
|
|
|
mock_super().get_object.assert_called_once_with()
|
|
|
|
def test_dispatch_with_existin_object(self, mock_super):
|
|
view = views.SingleObjectMixin()
|
|
view.object = 'old_object'
|
|
view.get_object = MagicMock()
|
|
|
|
view.dispatch()
|
|
|
|
mock_super().dispatch.assert_called_with()
|
|
self.assertEqual(
|
|
view.object,
|
|
'old_object',
|
|
"view.object should not be changed")
|
|
self.assertFalse(
|
|
view.get_object.called,
|
|
"view.get_object() should not be called")
|
|
|
|
def test_dispatch_without_existin_object(self, mock_super):
|
|
view = views.SingleObjectMixin()
|
|
view.get_object = MagicMock(return_value='new_object')
|
|
|
|
view.dispatch()
|
|
|
|
mock_super().dispatch.assert_called_with()
|
|
self.assertEqual(
|
|
view.object,
|
|
'new_object',
|
|
"view.object should be changed")
|
|
self.assertTrue(
|
|
view.get_object.called,
|
|
"view.get_object() should be called")
|
|
|
|
|
|
class TestAPIView(TestCase):
|
|
def test_class_creation(self):
|
|
"""
|
|
Tests that the APIView has all relevant methods
|
|
"""
|
|
http_methods = set(('get', 'post', 'put', 'patch', 'delete', 'head', 'options', 'trace'))
|
|
|
|
self.assertTrue(
|
|
http_methods.issubset(views.APIView.__dict__),
|
|
"All http methods should be defined in the APIView")
|
|
self.assertFalse(
|
|
hasattr(views.APIView, 'method_call'),
|
|
"The APIView should not have the method 'method_call'")
|
|
|
|
|
|
class TestCSRFMixin(TestCase):
|
|
@patch('builtins.super')
|
|
def test_as_view(self, mock_super):
|
|
"""
|
|
Tests, that ensure_csrf_cookie is called.
|
|
"""
|
|
mock_super().as_view.return_value = 'super_view'
|
|
with patch('openslides.utils.views.ensure_csrf_cookie') as ensure_csrf_cookie:
|
|
views.CSRFMixin.as_view()
|
|
|
|
ensure_csrf_cookie.assert_called_once_with('super_view')
|