from unittest import TestCase
from unittest.mock import MagicMock, patch

from openslides.agenda.models import Item


class ItemTitle(TestCase):
    def test_get_title_without_item_no(self):
        item = Item(title='test_title')
        self.assertEqual(
            item.get_title(),
            'test_title')

    @patch('openslides.agenda.models.Item.item_no', '5')
    def test_get_title_with_item_no(self):
        item = Item(title='test_title')
        self.assertEqual(
            item.get_title(),
            '5 test_title')

    @patch('openslides.agenda.models.Item.content_object')
    def test_get_title_from_related(self, content_object):
        item = Item(title='test_title')
        content_object.get_agenda_title.return_value = 'related_title'

        self.assertEqual(
            item.get_title(),
            'related_title')

    @patch('openslides.agenda.models.Item.content_object')
    def test_get_title_invalid_related(self, content_object):
        item = Item(title='test_title')
        content_object.get_agenda_title.return_value = 'related_title'
        del content_object.get_agenda_title

        with self.assertRaises(NotImplementedError):
            item.get_title()

    def test_title_supplement_without_related(self):
        item = Item()
        self.assertEqual(
            item.get_title_supplement(),
            '')

    @patch('openslides.agenda.models.Item.content_object')
    def test_title_supplement_with_related(self, content_object):
        item = Item()
        content_object.get_agenda_title_supplement.return_value = 'related_title_supplement'

        self.assertEqual(
            item.get_title_supplement(),
            'related_title_supplement')

    @patch('openslides.agenda.models.Item.content_object')
    def test_title_supplement_invalid_related(self, content_object):
        item = Item()
        del content_object.get_agenda_title_supplement

        with self.assertRaises(NotImplementedError):
            item.get_title_supplement()


@patch('openslides.agenda.models.Item.objects.rebuild')
@patch('openslides.agenda.models.Item.get_children')
class ItemDelete(TestCase):
    def test_delete_with_children_is_true(self, get_children, rebuild):
        item = Item()

        with patch('builtins.super') as mock_super:
            item.delete(with_children=True)

        self.assertFalse(get_children.called)
        rebuild.assert_called_once_with()
        mock_super().delete.assert_called_once_with()

    def test_delete_with_children_is_false(self, get_children, rebuild):
        parent = Item()
        item = Item()
        item.parent = parent
        child_item = MagicMock()
        get_children.return_value = [child_item]

        with patch('builtins.super') as mock_super:
            item.delete(with_children=False)

        child_item.move_to.assert_called_once_with(item.parent)
        child_item.save_assert_called_once_with()
        rebuild.assert_called_once_with()
        mock_super().delete.assert_called_once_with()