120 lines
4.6 KiB
Python
120 lines
4.6 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()
|