from openslides.agenda.models import ListOfSpeakers, Speaker from openslides.topics.models import Topic from openslides.users.models import User from openslides.utils.exceptions import OpenSlidesError from openslides.utils.test import TestCase class ListOfSpeakerModelTests(TestCase): def setUp(self): self.list_of_speakers_1 = Topic.objects.create( title="list_of_speakers_1" ).list_of_speakers self.list_of_speakers_2 = Topic.objects.create( title="list_of_speakers_2" ).list_of_speakers self.speaker1 = User.objects.create(username="user1") self.speaker2 = User.objects.create(username="user2") def test_append_speaker(self): # Append speaker1 to the list of list_of_speakers_1 speaker1_los1 = Speaker.objects.add(self.speaker1, self.list_of_speakers_1) self.assertTrue( Speaker.objects.filter( user=self.speaker1, list_of_speakers=self.list_of_speakers_1 ).exists() ) # Append speaker1 to the list of list_of_speakers_2 speaker1_los2 = Speaker.objects.add(self.speaker1, self.list_of_speakers_2) self.assertTrue( Speaker.objects.filter( user=self.speaker1, list_of_speakers=self.list_of_speakers_2 ).exists() ) # Append speaker2 to the list of list_of_speakers_1 speaker2_los1 = Speaker.objects.add(self.speaker2, self.list_of_speakers_1) self.assertTrue( Speaker.objects.filter( user=self.speaker2, list_of_speakers=self.list_of_speakers_1 ).exists() ) # Try to append speaker 1 again to the list of list_of_speakers_1 with self.assertRaises(OpenSlidesError): Speaker.objects.add(self.speaker1, self.list_of_speakers_1) # Check time and weight for object in (speaker1_los1, speaker2_los1, speaker1_los2): self.assertIsNone(object.begin_time) self.assertIsNone(object.end_time) self.assertEqual(speaker1_los1.weight, 1) self.assertEqual(speaker1_los2.weight, 1) self.assertEqual(speaker2_los1.weight, 2) def test_open_close_list_of_speaker(self): self.assertFalse( ListOfSpeakers.objects.get(pk=self.list_of_speakers_1.pk).closed ) self.list_of_speakers_1.closed = True self.list_of_speakers_1.save() self.assertTrue( ListOfSpeakers.objects.get(pk=self.list_of_speakers_1.pk).closed ) def test_speak_and_finish(self): speaker1_los1 = Speaker.objects.add(self.speaker1, self.list_of_speakers_1) self.assertIsNone(speaker1_los1.begin_time) self.assertIsNone(speaker1_los1.end_time) speaker1_los1.begin_speech() self.assertIsNotNone(Speaker.objects.get(pk=speaker1_los1.pk).begin_time) self.assertIsNone(Speaker.objects.get(pk=speaker1_los1.pk).weight) speaker1_los1.end_speech() self.assertIsNotNone(Speaker.objects.get(pk=speaker1_los1.pk).end_time) def test_finish_when_other_speaker_begins(self): speaker1_los1 = Speaker.objects.add(self.speaker1, self.list_of_speakers_1) speaker2_los1 = Speaker.objects.add(self.speaker2, self.list_of_speakers_1) speaker1_los1.begin_speech() self.assertIsNone(speaker1_los1.end_time) self.assertIsNone(speaker2_los1.begin_time) speaker2_los1.begin_speech() self.assertIsNotNone( Speaker.objects.get( user=self.speaker1, list_of_speakers=self.list_of_speakers_1 ).end_time ) self.assertIsNotNone(speaker2_los1.begin_time)