from unittest import TestCase from unittest.mock import MagicMock, patch from rest_framework.exceptions import PermissionDenied from openslides.motions.views import MotionViewSet class MotionViewSetCreate(TestCase): """ Tests create view of MotionViewSet. """ def setUp(self): self.request = MagicMock() self.view_instance = MotionViewSet() self.view_instance.request = self.request self.view_instance.format_kwarg = MagicMock() self.view_instance.get_serializer = get_serializer_mock = MagicMock() get_serializer_mock.return_value = self.mock_serializer = MagicMock() @patch('openslides.motions.views.config') def test_simple_create(self, mock_config): self.request.user.has_perm.return_value = True self.view_instance.create(self.request) self.mock_serializer.save.assert_called_with(request_user=self.request.user) @patch('openslides.motions.views.config') def test_user_without_can_create_perm(self, mock_config): self.request.user.has_perm.return_value = False with self.assertRaises(PermissionDenied): self.view_instance.create(self.request) class MotionViewSetUpdate(TestCase): """ Tests update view of MotionViewSet. """ def setUp(self): self.request = MagicMock() self.view_instance = MotionViewSet() self.view_instance.request = self.request self.view_instance.kwargs = MagicMock() self.view_instance.get_object = MagicMock() self.view_instance.get_serializer = get_serializer_mock = MagicMock() get_serializer_mock.return_value = self.mock_serializer = MagicMock() @patch('openslides.motions.views.config') def test_simple_update(self, mock_config): self.request.user.has_perm.return_value = True self.view_instance.update(self.request) self.mock_serializer.save.assert_called_with() @patch('openslides.motions.views.config') def test_user_without_perms(self, mock_config): self.request.user.has_perm.return_value = False with self.assertRaises(PermissionDenied): self.view_instance.update(self.request) class MotionViewSetManageVersion(TestCase): """ Tests views of MotionViewSet to manage versions. """ def setUp(self): self.request = MagicMock() self.view_instance = MotionViewSet() self.view_instance.request = self.request self.view_instance.get_object = get_object_mock = MagicMock() get_object_mock.return_value = self.mock_motion = MagicMock() def test_activate_version(self): self.request.method = 'PUT' self.request.user.has_perm.return_value = True self.view_instance.manage_version(self.request) self.mock_motion.save.assert_called_with(update_fields=['active_version']) def test_delete_version(self): self.request.method = 'DELETE' self.request.user.has_perm.return_value = True self.view_instance.manage_version(self.request) self.mock_motion.versions.get.return_value.delete.assert_called_with()