From 9f99ef8a38a78830d99d92f95ebc35debc8b1603 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Norman=20J=C3=A4ckel?= <contact@normanjaeckel.de>
Date: Mon, 12 May 2014 17:56:07 +0200
Subject: [PATCH] Fixed error when creating an assignment poll. Fixed #1203.

---
 openslides/assignment/models.py | 34 ++++++++++++++++++---------------
 tests/assignment/test_views.py  | 30 ++++++++++++++++++++++++++++-
 2 files changed, 48 insertions(+), 16 deletions(-)

diff --git a/openslides/assignment/models.py b/openslides/assignment/models.py
index 06360cdf3..ca855777d 100644
--- a/openslides/assignment/models.py
+++ b/openslides/assignment/models.py
@@ -205,9 +205,24 @@ class Assignment(SlideMixin, AbsoluteUrlMixin, models.Model):
         Creates an new poll for the assignment and adds all candidates to all
         lists of speakers of related agenda items.
         """
+        if config['assignment_poll_vote_values'] == 'votes':
+            yesnoabstain = False
+        elif config['assignment_poll_vote_values'] == 'yesnoabstain':
+            yesnoabstain = True
+        else:
+            # config['assignment_poll_vote_values'] == 'auto'
+            # candidates <= available posts -> yes/no/abstain
+            if len(self.candidates) <= (self.posts - len(self.elected)):
+                yesnoabstain = True
+            else:
+                yesnoabstain = False
+
         poll = AssignmentPoll.objects.create(
-            assignment=self, description=self.poll_description_default)
+            assignment=self,
+            description=self.poll_description_default,
+            yesnoabstain=yesnoabstain)
         poll.set_options([{'candidate': person} for person in self.candidates])
+
         items = Item.objects.filter(content_type=ContentType.objects.get_for_model(Assignment), object_id=self.pk)
         for item in items:
             someone_added = None
@@ -220,6 +235,7 @@ class Assignment(SlideMixin, AbsoluteUrlMixin, models.Model):
                     pass
             if someone_added is not None:
                 someone_added.check_and_update_projector()
+
         return poll
 
     def vote_results(self, only_published):
@@ -281,11 +297,11 @@ class AssignmentPoll(SlideMixin, RelatedModelMixin, CollectDefaultVotesMixin,
                      PublishPollMixin, AbsoluteUrlMixin, BasePoll):
 
     slide_callback_name = 'assignmentpoll'
-    """Name of the callback for the slide-system."""
+    """Name of the callback for the slide system."""
 
     option_class = AssignmentOption
     assignment = models.ForeignKey(Assignment, related_name='poll_set')
-    yesnoabstain = models.NullBooleanField()
+    yesnoabstain = models.BooleanField()
     description = models.CharField(
         max_length=79, null=True, blank=True,
         verbose_name=ugettext_lazy("Comment on the ballot paper"))
@@ -314,18 +330,6 @@ class AssignmentPoll(SlideMixin, RelatedModelMixin, CollectDefaultVotesMixin,
         return self.assignment
 
     def get_vote_values(self):
-        if self.yesnoabstain is None:
-            if config['assignment_poll_vote_values'] == 'votes':
-                self.yesnoabstain = False
-            elif config['assignment_poll_vote_values'] == 'yesnoabstain':
-                self.yesnoabstain = True
-            else:
-                # candidates <= available posts -> yes/no/abstain
-                if len(self.assignment.candidates) <= (self.assignment.posts - len(self.assignment.elected)):
-                    self.yesnoabstain = True
-                else:
-                    self.yesnoabstain = False
-            self.save()
         if self.yesnoabstain:
             return [ugettext_noop('Yes'), ugettext_noop('No'), ugettext_noop('Abstain')]
         else:
diff --git a/tests/assignment/test_views.py b/tests/assignment/test_views.py
index 48295682b..2b2358b44 100644
--- a/tests/assignment/test_views.py
+++ b/tests/assignment/test_views.py
@@ -2,7 +2,8 @@
 
 from django.test.client import Client
 
-from openslides.assignment.models import Assignment
+from openslides.assignment.models import Assignment, AssignmentPoll
+from openslides.config.api import config
 from openslides.participant.models import Group, User
 from openslides.utils.test import TestCase
 
@@ -81,3 +82,30 @@ class TestAssignmentDetailView(AssignmentViewTestCase):
         response = self.staff_client.get('/assignment/1/')
         self.assertContains(response, 'No candidates available.')
         self.assertContains(response, 'Blocked Candidates')
+
+
+class TestAssignmentPollCreateView(TestCase):
+    """
+    Tests the creation of assignment polls.
+    """
+    def test_assignment_add_candidate(self):
+        admin = User.objects.get(pk=1)
+        self.assignment = Assignment.objects.create(
+            name='test_assignment_oiL2heerookiegeirai0',
+            posts=1)
+        self.assignment.run(admin, admin)
+        self.assertEqual(len(Assignment.objects.get(pk=self.assignment.pk).candidates), 1)
+
+    def test_assignment_poll_creation(self):
+        self.test_assignment_add_candidate()
+        self.assignment.set_status('vot')
+        admin_client = Client()
+        admin_client.login(username='admin', password='admin')
+        self.assertFalse(AssignmentPoll.objects.exists())
+        self.assertEqual(config['assignment_poll_vote_values'], 'auto')
+        response = admin_client.get('/assignment/1/gen_poll/')
+        self.assertRedirects(response, '/assignment/1/')
+        poll = AssignmentPoll.objects.get()
+        self.assertEqual(poll.assignment, self.assignment)
+        self.assertEqual(poll.assignmentoption_set.count(), 1)
+        self.assertTrue(poll.yesnoabstain)