Mercurial > public > sg101

annotate bio/tests/test_views.py @ 887:9a15f7c27526

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Actually save model object upon change. This commit was tested on the comments model. Additional logging added. Added check for Markdown image references. Added TODOs after observing behavior on comments.
author Brian Neal <bgneal@gmail.com>
date Tue, 03 Feb 2015 21:09:44 -0600
parents 9e803323a0d0
children
rev   line source
bgneal@463 1 """
bgneal@463 2 View tests for the bio application.
bgneal@463 3
bgneal@463 4 """
bgneal@767 5 from io import BytesIO
bgneal@767 6 import os
bgneal@767 7
bgneal@463 8 from django.contrib.auth.models import User
bgneal@463 9 from django.test import TestCase
bgneal@463 10 from django.core.urlresolvers import reverse, NoReverseMatch
bgneal@768 11 from django.core import mail
bgneal@463 12
bgneal@767 13 from PIL import Image
bgneal@767 14
bgneal@768 15 from bio.models import UserProfile, UserProfileFlag
bgneal@767 16
bgneal@463 17
bgneal@463 18 class MemberSearchTest(TestCase):
bgneal@463 19
bgneal@463 20 USERNAME = u'John'
bgneal@463 21
bgneal@463 22 def setUp(self):
bgneal@463 23 user = User.objects.create_user(self.USERNAME, '', 'password')
bgneal@463 24 user.save()
bgneal@463 25
bgneal@463 26 self.username = 'test_user'
bgneal@463 27 self.pw = 'password'
bgneal@463 28 self.user = User.objects.create_user(self.username, '', self.pw)
bgneal@463 29 self.user.save()
bgneal@463 30 self.assertTrue(self.client.login(username=self.username,
bgneal@463 31 password=self.pw))
bgneal@463 32
bgneal@463 33 def tearDown(self):
bgneal@463 34 self.client.logout()
bgneal@463 35
bgneal@463 36 def testValidName(self):
bgneal@463 37 """
bgneal@463 38 Test a valid username.
bgneal@463 39 """
bgneal@463 40
bgneal@769 41 response = self.client.get(reverse('bio-member_search'),
bgneal@463 42 {'username': self.USERNAME},
bgneal@463 43 follow=True)
bgneal@463 44
bgneal@463 45 self.assertEqual(len(response.redirect_chain), 1)
bgneal@463 46 if response.redirect_chain:
bgneal@463 47 self.assertEqual(response.redirect_chain[0][0],
bgneal@463 48 'http://testserver' + reverse('bio-view_profile',
bgneal@463 49 kwargs={'username': self.USERNAME}))
bgneal@463 50 self.assertEqual(response.redirect_chain[0][1], 302)
bgneal@463 51
bgneal@463 52 self.assertEqual(response.status_code, 200)
bgneal@463 53
bgneal@463 54 def testInvalidName(self):
bgneal@463 55 """
bgneal@463 56 Test a invalid username.
bgneal@463 57 """
bgneal@463 58
bgneal@769 59 response = self.client.get(reverse('bio-member_search'),
bgneal@463 60 {'username': self.USERNAME + '!'})
bgneal@463 61
bgneal@463 62 self.assertEqual(response.status_code, 200)
bgneal@463 63 self.assertContains(response, "That username does not exist.")
bgneal@463 64
bgneal@463 65 def testTrailingSpace(self):
bgneal@463 66 """
bgneal@463 67 Test a username with a trailing space.
bgneal@463 68 """
bgneal@463 69
bgneal@463 70 try:
bgneal@769 71 response = self.client.get(reverse('bio-member_search'),
bgneal@463 72 {'username': self.USERNAME + ' '},
bgneal@463 73 follow=True)
bgneal@463 74 except NoReverseMatch:
bgneal@463 75 self.fail('bit by a MySQL bug?')
bgneal@463 76
bgneal@463 77 self.assertEqual(len(response.redirect_chain), 1)
bgneal@463 78 if response.redirect_chain:
bgneal@463 79 self.assertEqual(response.redirect_chain[0][0],
bgneal@463 80 'http://testserver' + reverse('bio-view_profile',
bgneal@463 81 kwargs={'username': self.USERNAME}))
bgneal@463 82 self.assertEqual(response.redirect_chain[0][1], 302)
bgneal@463 83
bgneal@463 84 self.assertEqual(response.status_code, 200)
bgneal@765 85
bgneal@765 86
bgneal@765 87 class AnonMemberListTest(TestCase):
bgneal@765 88
bgneal@765 89 def _test_login_required(self, list_by):
bgneal@765 90
bgneal@765 91 member_list_url = reverse('bio-member_list', kwargs={'type': list_by})
bgneal@765 92 response = self.client.get(member_list_url, follow=True)
bgneal@765 93 login_url = reverse('accounts-login') + '?next=' + member_list_url
bgneal@765 94 self.assertRedirects(response, login_url)
bgneal@765 95
bgneal@765 96 def test_login_required(self):
bgneal@765 97 self._test_login_required('user')
bgneal@765 98 self._test_login_required('date')
bgneal@765 99
bgneal@765 100
bgneal@765 101 class MemberListTest(TestCase):
bgneal@765 102
bgneal@765 103 def setUp(self):
bgneal@765 104 u1 = User.objects.create_user('paul', '', 'pw')
bgneal@765 105 u1.save()
bgneal@765 106 u2 = User.objects.create_user('eddie', '', 'pw')
bgneal@765 107 u2.save()
bgneal@765 108
bgneal@765 109 self.user = u1
bgneal@765 110 self.client.login(username=u1.username, password='pw')
bgneal@765 111
bgneal@765 112 def _test_get(self, list_by, page=None, code=200):
bgneal@765 113
bgneal@765 114 member_list_url = reverse('bio-member_list', kwargs={'type': list_by})
bgneal@765 115 if page is not None:
bgneal@765 116 member_list_url += '?page=%d' % page
bgneal@765 117 response = self.client.get(member_list_url)
bgneal@765 118 self.assertEqual(response.status_code, code)
bgneal@765 119
bgneal@765 120 def test_get(self):
bgneal@765 121 self._test_get('user')
bgneal@765 122 self._test_get('date')
bgneal@765 123 self._test_get('user', page=1)
bgneal@765 124 self._test_get('date', page=1)
bgneal@765 125
bgneal@765 126 def test_invalid_page(self):
bgneal@765 127 self._test_get('user', page=99, code=404)
bgneal@765 128 self._test_get('date', page=99, code=404)
bgneal@765 129
bgneal@765 130
bgneal@766 131 class ProfileTest(TestCase):
bgneal@765 132
bgneal@765 133 def setUp(self):
bgneal@765 134 u1 = User.objects.create_user('paul', '', 'pw')
bgneal@765 135 u1.save()
bgneal@765 136 u2 = User.objects.create_user('eddie', '', 'pw')
bgneal@765 137 u2.save()
bgneal@765 138
bgneal@767 139 def tearDown(self):
bgneal@767 140 profile = UserProfile.objects.get(user__username='paul')
bgneal@767 141 if profile.avatar:
bgneal@767 142 os.remove(profile.avatar.path)
bgneal@767 143
bgneal@766 144 def test_anon_my_profile(self):
bgneal@765 145 my_url = reverse('bio-me')
bgneal@765 146 response = self.client.get(my_url)
bgneal@765 147 login_url = reverse('accounts-login') + '?next=' + my_url
bgneal@765 148 self.assertRedirects(response, login_url)
bgneal@765 149
bgneal@766 150 def test_authenticated_my_profile(self):
bgneal@765 151 self.client.login(username='paul', password='pw')
bgneal@765 152 response = self.client.get(reverse('bio-me'))
bgneal@765 153 self.assertEqual(response.status_code, 200)
bgneal@766 154
bgneal@766 155 def test_anon_other_profile(self):
bgneal@766 156 url = reverse('bio-view_profile', kwargs={'username': 'eddie'})
bgneal@766 157 response = self.client.get(url)
bgneal@766 158 login_url = reverse('accounts-login') + '?next=' + url
bgneal@766 159 self.assertRedirects(response, login_url)
bgneal@766 160
bgneal@766 161 def test_authenticated_other_profile(self):
bgneal@766 162 self.client.login(username='paul', password='pw')
bgneal@766 163 url = reverse('bio-view_profile', kwargs={'username': 'eddie'})
bgneal@766 164 response = self.client.get(url)
bgneal@766 165 self.assertEqual(response.status_code, 200)
bgneal@766 166
bgneal@766 167 def test_view_profile_as_me_redirects(self):
bgneal@766 168 self.client.login(username='paul', password='pw')
bgneal@766 169 url = reverse('bio-view_profile', kwargs={'username': 'paul'})
bgneal@766 170 response = self.client.get(url)
bgneal@766 171 self.assertRedirects(response, reverse('bio-me'))
bgneal@766 172
bgneal@766 173 def test_edit_anon(self):
bgneal@766 174 url = reverse('bio-edit_profile')
bgneal@766 175 response = self.client.get(url)
bgneal@766 176 login_url = reverse('accounts-login') + '?next=' + url
bgneal@766 177 self.assertRedirects(response, login_url)
bgneal@766 178
bgneal@766 179 def test_edit_profile(self):
bgneal@766 180 self.client.login(username='paul', password='pw')
bgneal@766 181 url = reverse('bio-edit_profile')
bgneal@766 182 response = self.client.get(url)
bgneal@766 183 self.assertEqual(response.status_code, 200)
bgneal@766 184
bgneal@766 185 # Save some new data. We don't exhaustively test every field.
bgneal@766 186 response = self.client.post(url, {
bgneal@766 187 'first_name': 'Paul',
bgneal@766 188 'last_name': 'Johnson',
bgneal@766 189 'email': 'pjmoto@example.com',
bgneal@766 190 'occupation': 'Surf Guitar Pioneer',
bgneal@766 191 'profile_text': 'Wrote Mr. Moto at age 15.',
bgneal@766 192 'submit_button': 'Save',
bgneal@766 193 })
bgneal@766 194 self.assertRedirects(response, reverse('bio-me'))
bgneal@766 195
bgneal@766 196 u = User.objects.get(username='paul')
bgneal@766 197 self.assertEqual(u.first_name, 'Paul')
bgneal@766 198 self.assertEqual(u.last_name, 'Johnson')
bgneal@766 199 self.assertEqual(u.email, 'pjmoto@example.com')
bgneal@789 200 profile = u.profile
bgneal@766 201 self.assertEqual(profile.occupation, 'Surf Guitar Pioneer')
bgneal@766 202 self.assertEqual(profile.profile_text, 'Wrote Mr. Moto at age 15.')
bgneal@767 203
bgneal@767 204 def test_avatar_change(self):
bgneal@767 205 self.client.login(username='paul', password='pw')
bgneal@767 206 url = reverse('bio-change_avatar')
bgneal@767 207 response = self.client.get(url)
bgneal@767 208 self.assertEqual(response.status_code, 200)
bgneal@767 209
bgneal@767 210 # Create an in-memory image for a dummy avatar
bgneal@767 211 file_obj = BytesIO()
bgneal@767 212 image = Image.new('RGBA', size=(100, 100))
bgneal@767 213 image.save(file_obj, 'png')
bgneal@767 214 file_obj.name = 'avatar.png'
bgneal@767 215 file_obj.seek(0)
bgneal@767 216
bgneal@767 217 response = self.client.post(url, {'avatar_file': file_obj}, follow=True)
bgneal@767 218 self.assertRedirects(response, reverse('bio-me'))
bgneal@767 219
bgneal@768 220 profile = UserProfile.objects.get(user__username='paul')
bgneal@768 221 self.assertTrue(profile.avatar.path)
bgneal@768 222 self.assertTrue(profile.avatar.name)
bgneal@768 223 self.assertTrue(os.path.exists(profile.avatar.path))
bgneal@768 224
bgneal@768 225 def test_flag_profile_get(self):
bgneal@768 226 paul = User.objects.get(username='paul')
bgneal@768 227 eddie = User.objects.get(username='eddie')
bgneal@768 228 url = reverse('bio-flag_profile', args=[eddie.pk])
bgneal@768 229
bgneal@768 230 # not logged in, wrong request method:
bgneal@768 231 response = self.client.get(url)
bgneal@768 232 self.assertEqual(response.status_code, 405)
bgneal@768 233
bgneal@768 234 # logged in, wrong request method
bgneal@768 235 self.client.login(username='paul', password='pw')
bgneal@768 236 response = self.client.get(url)
bgneal@768 237 self.assertEqual(response.status_code, 405)
bgneal@768 238
bgneal@768 239 # logged out, right request method
bgneal@768 240 self.client.logout()
bgneal@768 241 response = self.client.post(url)
bgneal@768 242 self.assertEqual(response.status_code, 200)
bgneal@768 243 self.assertEqual(UserProfileFlag.objects.all().count(), 0)
bgneal@768 244
bgneal@768 245 # logged in, right request method
bgneal@768 246 self.client.login(username='paul', password='pw')
bgneal@768 247 response = self.client.post(url)
bgneal@768 248 self.assertEqual(response.status_code, 200)
bgneal@768 249 self.assertEqual(UserProfileFlag.objects.all().count(), 1)
bgneal@768 250
bgneal@768 251 flag = UserProfileFlag.objects.first()
bgneal@768 252 self.assertEqual(flag.user, paul)
bgneal@789 253 self.assertEqual(flag.profile, eddie.profile)
bgneal@768 254
bgneal@768 255 self.assertEqual(len(mail.outbox), 1)
bgneal@768 256 self.assertTrue('A Profile Has Been Flagged' in mail.outbox[0].subject)