Add unit tests

1 files changed, 18 insertions, 0 deletions

diff --git a/hyperkitty/tests/test_views.py b/hyperkitty/tests/test_views.py
index 649edaa..123989b 100644
--- a/hyperkitty/tests/test_views.py
+++ b/hyperkitty/tests/test_views.py
@@ -23,6 +23,7 @@ from django.test import TestCase
 from django.test.client import Client
 from django.contrib.auth.models import User
 from django.core.urlresolvers import reverse
+import urllib
 
 class AccountViewsTestCase(TestCase):
         
@@ -64,3 +65,20 @@ class AccountViewsTestCase(TestCase):
         
         # @TODO: Try to register a user and verify its working
         
+
+class MessageViewsTestCase(TestCase):
+     def setUp(self):
+        self.client = Client()
+
+     def test_good_vote(self):
+        User.objects.create_user('testuser', 'syst3m.w0rm+test@gmail.com', 'testPass')
+        user = self.client.login(username='testuser', password='testPass')
+        
+        resp = self.client.post(reverse('message_vote', kwargs={'mlist_fqdn': 'list@list.com'}), {'vote': 1, 'messageid': 123, })
+        self.assertEqual(resp.status_code, 200)
+
+     def test_unauth_vote(self):
+        resp = self.client.post(reverse('message_vote', kwargs={'mlist_fqdn': 'list@list.com'}), {'vote': 1, 'messageid': 123, })
+        url = "%s?next=%s" % (reverse('user_login'), urllib.quote(reverse('message_vote', kwargs={'mlist_fqdn': 'list@list.com'})))
+        self.assertRedirects(resp, url)
+