diff bio/tests/test_views.py @ 768:1b1a12abde3b

More bio tests. Change Celery to be eager during testing.
author Brian Neal <bgneal@gmail.com>
date Tue, 21 Jan 2014 19:42:09 -0600
parents a4ae99359286
children cd3343abca9d
line wrap: on
line diff
--- a/bio/tests/test_views.py	Mon Jan 20 20:04:25 2014 -0600
+++ b/bio/tests/test_views.py	Tue Jan 21 19:42:09 2014 -0600
@@ -8,10 +8,11 @@
 from django.contrib.auth.models import User
 from django.test import TestCase
 from django.core.urlresolvers import reverse, NoReverseMatch
+from django.core import mail
 
 from PIL import Image
 
-from bio.models import UserProfile
+from bio.models import UserProfile, UserProfileFlag
 
 
 class MemberSearchTest(TestCase):
@@ -216,3 +217,40 @@
         response = self.client.post(url, {'avatar_file': file_obj}, follow=True)
         self.assertRedirects(response, reverse('bio-me'))
 
+        profile = UserProfile.objects.get(user__username='paul')
+        self.assertTrue(profile.avatar.path)
+        self.assertTrue(profile.avatar.name)
+        self.assertTrue(os.path.exists(profile.avatar.path))
+
+    def test_flag_profile_get(self):
+        paul = User.objects.get(username='paul')
+        eddie = User.objects.get(username='eddie')
+        url = reverse('bio-flag_profile', args=[eddie.pk])
+
+        # not logged in, wrong request method:
+        response = self.client.get(url)
+        self.assertEqual(response.status_code, 405)
+
+        # logged in, wrong request method
+        self.client.login(username='paul', password='pw')
+        response = self.client.get(url)
+        self.assertEqual(response.status_code, 405)
+
+        # logged out, right request method
+        self.client.logout()
+        response = self.client.post(url)
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(UserProfileFlag.objects.all().count(), 0)
+
+        # logged in, right request method
+        self.client.login(username='paul', password='pw')
+        response = self.client.post(url)
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(UserProfileFlag.objects.all().count(), 1)
+
+        flag = UserProfileFlag.objects.first()
+        self.assertEqual(flag.user, paul)
+        self.assertEqual(flag.profile, eddie.get_profile())
+
+        self.assertEqual(len(mail.outbox), 1)
+        self.assertTrue('A Profile Has Been Flagged' in mail.outbox[0].subject)