First impression of poll results

1 files changed, 40 insertions, 0 deletions

diff --git a/webapp/polls/tests/result_tests.py b/webapp/polls/tests/result_tests.py
new file mode 100644
index 0000000..e06862d
--- /dev/null
+++ b/webapp/polls/tests/result_tests.py
@@ -0,0 +1,40 @@
+# -*- encoding: utf-8 -*-
+from polls.models import Poll, PollResult
+
+from utils.test import MongoTestCase
+
+
+class PollTests(MongoTestCase):
+
+    def test_init(self):
+        data = {'name': "poll #1"}
+        poll = Poll(data=data)
+        poll_id = poll.save()
+
+        data = {'poll_id': poll_id}
+        poll_result = PollResult([data])
+
+        self.assertEqual(data, poll_result._data)
+
+    def test_validate(self):
+        pass
+
+    def test_to_python(self):
+        data = {'poll_id': 'ID'}
+        poll_result = PollResult([data])
+
+        self.assertTrue('poll_id' in poll_result.to_python().keys())
+
+    def test_save(self):
+        data = {'poll_id': "ID"}
+        poll_result = PollResult([data])
+        poll_result.save()
+
+        self.assertEqual(1, self.db.poll_results.count())
+
+    def test_get(self):
+        data = {'poll_id': 'ID'}
+        poll_result = PollResult([data])
+        poll_result_id = poll_result.save()
+
+        self.assertIsNotNone(PollResult.get(poll_result_id))