Pollster asignation in poll form

1 files changed, 5 insertions, 1 deletions

diff --git a/webapp/polls/tests/poll_tests.py b/webapp/polls/tests/poll_tests.py
index 2bf04ad..27f2e2e 100644
--- a/webapp/polls/tests/poll_tests.py
+++ b/webapp/polls/tests/poll_tests.py
@@ -3,6 +3,7 @@ from bson import ObjectId
 
 from polls.models import Poll
 from polls.forms import PollAddForm
+from pollster.models import Pollster
 
 from utils.test import MongoTestCase
 
@@ -10,12 +11,15 @@ from utils.test import MongoTestCase
 class PollTests(MongoTestCase):
 
     def test_poll_init(self):
+        pollster = Pollster.create(username="test", password="test")
+
         name = 'nombre de encuesta'
-        data = {'name': name}
+        data = {'name': name, 'pollster': pollster.id}
 
         poll = Poll(data=data)
         self.assertIsNone(poll.id)
         self.assertEqual(name, poll.name)
+        self.assertEqual(pollster.id, poll.pollster)
 
         id = '513a0634421aa932884daa5c'
         data = {'id': id, 'name': name}