55 lines
2.4 KiB
Python
55 lines
2.4 KiB
Python
|
from unittest import TestCase
|
||
|
|
from unittest.mock import MagicMock, patch
|
||
|
|
|
||
|
|
from openslides.agenda.views import ItemViewSet
|
||
|
|
|
||
|
|
|
||
|
|
class ItemViewSetManageSpeaker(TestCase):
|
||
|
|
"""
|
||
|
|
Tests views of ItemViewSet to manage speakers.
|
||
|
|
"""
|
||
|
|
def setUp(self):
|
||
|
|
self.request = MagicMock()
|
||
|
|
self.view_instance = ItemViewSet()
|
||
|
|
self.view_instance.request = self.request
|
||
|
|
self.view_instance.get_object = get_object_mock = MagicMock()
|
||
|
|
get_object_mock.return_value = self.mock_item = MagicMock()
|
||
|
|
|
||
|
|
@patch('openslides.agenda.views.Speaker')
|
||
|
|
def test_add_oneself_as_speaker(self, mock_speaker):
|
||
|
|
self.request.method = 'POST'
|
||
|
|
self.request.user.has_perm.return_value = True
|
||
|
|
self.request.data = {}
|
||
|
|
self.mock_item.speaker_list_closed = False
|
||
|
|
self.view_instance.manage_speaker(self.request)
|
||
|
|
mock_speaker.objects.add.assert_called_with(self.request.user, self.mock_item)
|
||
|
|
|
||
|
|
@patch('openslides.agenda.views.get_user_model')
|
||
|
|
@patch('openslides.agenda.views.Speaker')
|
||
|
|
def test_add_someone_else_as_speaker(self, mock_speaker, mock_get_user_model):
|
||
|
|
self.request.method = 'POST'
|
||
|
|
self.request.user.has_perm.return_value = True
|
||
|
|
self.request.data = {'user': '2'} # It is assumed that the request user has pk!=2.
|
||
|
|
mock_get_user_model.return_value = MockUser = MagicMock()
|
||
|
|
MockUser.objects.get.return_value = mock_user = MagicMock()
|
||
|
|
self.view_instance.manage_speaker(self.request)
|
||
|
|
MockUser.objects.get.assert_called_with(pk=2)
|
||
|
|
mock_speaker.objects.add.assert_called_with(mock_user, self.mock_item)
|
||
|
|
|
||
|
|
@patch('openslides.agenda.views.Speaker')
|
||
|
|
def test_remove_oneself(self, mock_speaker):
|
||
|
|
self.request.method = 'DELETE'
|
||
|
|
self.request.data = {}
|
||
|
|
self.view_instance.manage_speaker(self.request)
|
||
|
|
mock_queryset = mock_speaker.objects.filter.return_value.exclude.return_value
|
||
|
|
mock_queryset.get.return_value.delete.assert_called_with()
|
||
|
|
|
||
|
|
@patch('openslides.agenda.views.Speaker')
|
||
|
|
def test_remove_someone_else(self, mock_speaker):
|
||
|
|
self.request.method = 'DELETE'
|
||
|
|
self.request.user.has_perm.return_value = True
|
||
|
|
self.request.data = {'speaker': '1'}
|
||
|
|
self.view_instance.manage_speaker(self.request)
|
||
|
|
mock_speaker.objects.get.assert_called_with(pk='1')
|
||
|
|
mock_speaker.objects.get.return_value.delete.assert_called_with()
|