Refactor: Poll method for check if has result

1 files changed, 23 insertions, 0 deletions

diff --git a/webapp/polls/tests/poll_tests.py b/webapp/polls/tests/poll_tests.py
index 86e2f60..a3f11c5 100644
--- a/webapp/polls/tests/poll_tests.py
+++ b/webapp/polls/tests/poll_tests.py
@@ -378,6 +378,29 @@ class PollTests(MongoTestCase):
         poll_result = poll.get_result()
         self.assertTrue(hasattr(poll_result, "to_csv"))
 
+    def test_has_result_with_no_results(self):
+        poll = Poll(data={'name': "poll"})
+        poll_id = poll.save()
+        poll = Poll.get(poll_id)
+
+        self.assertFalse(poll.has_result())
+
+    def test_has_result_with_at_least_one_result(self):
+        poll = Poll(data={'name': "poll"})
+        poll_id = poll.save()
+        poll = Poll.get(poll_id)
+
+        temp_dir = '/tmp'
+        uploaded_file = tempfile.NamedTemporaryFile(dir=temp_dir,
+                                                    suffix='.poll_result')
+        uploaded_file_path = uploaded_file.name
+        uploaded_filename = os.path.basename(uploaded_file_path)
+
+        poll.add_result_files([(uploaded_file_path, uploaded_filename)])
+        stub = lambda path: os.path.basename(path)
+
+        self.assertTrue(poll.has_result(as_instance_of=stub))
+
 
 class PollFormTests(MongoTestCase):