108 lines
4.5 KiB
Python
108 lines
4.5 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.inform_changed_data')
|
|
@patch('openslides.agenda.views.has_perm')
|
|
@patch('openslides.agenda.views.Speaker')
|
|
def test_add_oneself_as_speaker(self, mock_speaker, mock_has_perm, mock_icd):
|
|
self.request.method = 'POST'
|
|
self.request.user = 1
|
|
mock_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.inform_changed_data')
|
|
@patch('openslides.agenda.views.has_perm')
|
|
@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, mock_has_perm, mock_icd):
|
|
self.request.method = 'POST'
|
|
self.request.user = 1
|
|
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()
|
|
mock_has_perm.return_value = True
|
|
|
|
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.inform_changed_data')
|
|
@patch('openslides.agenda.views.has_perm')
|
|
@patch('openslides.agenda.views.Speaker')
|
|
def test_remove_someone_else(self, mock_speaker, mock_has_perm, mock_inform_changed_data):
|
|
self.request.method = 'DELETE'
|
|
self.request.user = 1
|
|
self.request.data = {'speaker': '1'}
|
|
mock_has_perm.return_value = True
|
|
|
|
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(skip_autoupdate=True)
|
|
mock_inform_changed_data.assert_called_with(self.mock_item)
|
|
|
|
|
|
class ItemViewSetSpeak(TestCase):
|
|
"""
|
|
Tests views of ItemViewSet to begin and end speech.
|
|
"""
|
|
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()
|
|
|
|
def test_begin_speech(self):
|
|
self.request.method = 'PUT'
|
|
self.request.user.has_perm.return_value = True
|
|
self.request.data = {}
|
|
self.mock_item.get_next_speaker.return_value = mock_next_speaker = MagicMock()
|
|
self.view_instance.speak(self.request)
|
|
mock_next_speaker.begin_speech.assert_called_with()
|
|
|
|
@patch('openslides.agenda.views.Speaker')
|
|
def test_begin_speech_specific_speaker(self, mock_speaker):
|
|
self.request.method = 'PUT'
|
|
self.request.user.has_perm.return_value = True
|
|
self.request.data = {'speaker': '1'}
|
|
mock_speaker.objects.get.return_value = mock_next_speaker = MagicMock()
|
|
self.view_instance.speak(self.request)
|
|
mock_next_speaker.begin_speech.assert_called_with()
|
|
|
|
@patch('openslides.agenda.views.Speaker')
|
|
def test_end_speech(self, mock_speaker):
|
|
self.request.method = 'DELETE'
|
|
self.request.user.has_perm.return_value = True
|
|
mock_speaker.objects.filter.return_value.exclude.return_value.get.return_value = mock_speaker = MagicMock()
|
|
self.view_instance.speak(self.request)
|
|
mock_speaker.end_speech.assert_called_with()
|