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()


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()