A researcher can associate a poll for many pollsters.

1 files changed, 44 insertions, 7 deletions

diff --git a/webapp/polls/tests/poll_tests.py b/webapp/polls/tests/poll_tests.py
index 17fa215..bb4b5a6 100644
--- a/webapp/polls/tests/poll_tests.py
+++ b/webapp/polls/tests/poll_tests.py
@@ -14,12 +14,12 @@ class PollTests(MongoTestCase):
         pollster = Pollster.create(username="test", password="test")
 
         name = 'nombre de encuesta'
-        data = {'name': name, 'pollster': pollster.id}
+        data = {'name': name, 'pollsters': pollster.id}
 
         poll = Poll(data=data)
         self.assertIsNone(poll.id)
         self.assertEqual(name, poll.name)
-        self.assertEqual(pollster.id, poll.pollster)
+        self.assertEqual(pollster.id, poll.pollsters)
 
         id = '513a0634421aa932884daa5c'
         data = {'id': id, 'name': name}
@@ -128,16 +128,53 @@ class PollTests(MongoTestCase):
 
         pollster = Pollster.create(username="test", password="test")
 
-        ids = [str(poll1_id), str(poll2_id)]
+        poll_ids = [str(poll1_id), str(poll2_id)]
 
-        Poll.pollster_assignment(str(pollster.id), ids)
+        Poll.pollster_assignment(str(pollster.id), poll_ids)
 
         poll1 = Poll.get(poll1_id)
         poll2 = Poll.get(poll2_id)
         poll3 = Poll.get(poll3_id)
-        self.assertEqual(pollster.id, poll1.get_pollster().id)
-        self.assertEqual(pollster.id, poll2.get_pollster().id)
-        self.assertIsNone(poll3.get_pollster())
+
+        self.assertTrue(pollster in poll1.get_pollsters())
+        self.assertTrue(pollster in poll2.get_pollsters())
+        self.assertEqual(0, len(poll3.get_pollsters()))
+
+    def test_two_pollsters_into_one_poll(self):
+
+        poll = Poll(data={'name': "poll"})
+        poll_id = poll.save()
+
+        pollster_1 = Pollster.create(username="test1", password="test1")
+        pollster_2 = Pollster.create(username="test2", password="test2")
+
+        Poll.pollster_assignment(str(pollster_1.id), [poll_id])
+        Poll.pollster_assignment(str(pollster_2.id), [poll_id])
+
+        pollster_1 = Pollster.get(pollster_1.id)
+        pollster_2 = Pollster.get(pollster_2.id)
+
+        poll = Poll.get(poll_id)
+
+        self.assertTrue(pollster_1 in poll.get_pollsters())
+        self.assertTrue(pollster_2 in poll.get_pollsters())
+
+    def test_pollster_assignment_to_poll_dont_repeat_pollster(self):
+
+        poll = Poll(data={'name': "poll"})
+        poll_id = poll.save()
+
+        pollster = Pollster.create(username="test", password="test")
+
+        Poll.pollster_assignment(str(pollster.id), [poll_id])
+        poll = Poll.get(poll_id)
+
+        self.assertEqual(1, len(poll.get_pollsters()))
+
+        Poll.pollster_assignment(str(pollster.id), [str(poll.id)])
+        poll = Poll.get(poll_id)
+
+        self.assertEqual(1, len(poll.get_pollsters()))
 
     # TODO: Mock file system methods
     def test_clone(self):