Mercurial > public > sg101

annotate user_photos/tests/test_views.py @ 1154:4da4e32b314c

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Do not restrict oEmbed support to just video. This will allow us to embed stuff from SoundClound and ReverbNation.
author Brian Neal <bgneal@gmail.com>
date Tue, 27 Dec 2016 10:21:37 -0600
parents 7138883966b3
children
rev   line source
bgneal@972 1 """
bgneal@972 2 Tests for the views in the user_photos application.
bgneal@972 3
bgneal@972 4 """
bgneal@972 5 import json
bgneal@972 6
bgneal@972 7 import mock
bgneal@972 8 from django.test import TestCase
bgneal@972 9 from django.test.utils import override_settings
bgneal@972 10 from django.contrib.auth.models import User
bgneal@972 11 from django.core.urlresolvers import reverse
bgneal@972 12 from django.http import JsonResponse
bgneal@972 13
bgneal@972 14
bgneal@972 15 class HotlinkImageTestCase(TestCase):
bgneal@972 16
bgneal@972 17 def setUp(self):
bgneal@972 18 self.username = 'test_user'
bgneal@972 19 self.pw = 'password'
bgneal@972 20 self.user = User.objects.create_user(self.username, '', self.pw)
bgneal@972 21 self.user.save()
bgneal@972 22 self.assertTrue(self.client.login(username=self.username, password=self.pw))
bgneal@972 23 self.view_url = reverse('user_photos-hotlink')
bgneal@972 24
bgneal@972 25 def testUnauthenticatedUser(self):
bgneal@972 26 self.client.logout()
bgneal@972 27 response = self.client.post(self.view_url,
bgneal@972 28 {'url': 'http://example.com/test.jpg'},
bgneal@972 29 HTTP_X_REQUESTED_WITH='XMLHttpRequest')
bgneal@972 30 self.assertIsInstance(response, JsonResponse)
bgneal@972 31 self.assertEqual(response.status_code, 403)
bgneal@972 32
bgneal@972 33 def testWrongMethod(self):
bgneal@972 34 response = self.client.get(self.view_url)
bgneal@972 35 self.assertIsInstance(response, JsonResponse)
bgneal@972 36 self.assertEqual(response.status_code, 405)
bgneal@972 37
bgneal@972 38 def testNotAjax(self):
bgneal@972 39 response = self.client.post(self.view_url,
bgneal@972 40 {'url': 'http://example.com/test.jpg'})
bgneal@972 41 self.assertIsInstance(response, JsonResponse)
bgneal@972 42 self.assertEqual(response.status_code, 405)
bgneal@972 43
bgneal@972 44 @override_settings(USER_PHOTOS_ENABLED=False)
bgneal@972 45 def testServiceDisabled(self):
bgneal@972 46 response = self.client.post(self.view_url,
bgneal@972 47 {'url': 'http://example.com/test.jpg'},
bgneal@972 48 HTTP_X_REQUESTED_WITH='XMLHttpRequest')
bgneal@972 49 self.assertIsInstance(response, JsonResponse)
bgneal@972 50 self.assertEqual(response.status_code, 403)
bgneal@972 51
bgneal@972 52 @mock.patch('user_photos.views.HotLinkImageForm')
bgneal@972 53 def testFormInvalid(self, form_mock):
bgneal@972 54 form_mock.return_value.is_valid.return_value = False
bgneal@972 55 form_mock.return_value.non_field_errors.return_value = mock.Mock()
bgneal@972 56 form_mock.return_value.non_field_errors.return_value.as_text.return_value = 'bad stuff'
bgneal@972 57 form_mock.return_value.errors.return_value = {}
bgneal@972 58
bgneal@972 59 response = self.client.post(self.view_url,
bgneal@972 60 {'url': 'http://example.com/test.jpg'},
bgneal@972 61 HTTP_X_REQUESTED_WITH='XMLHttpRequest')
bgneal@972 62 self.assertIsInstance(response, JsonResponse)
bgneal@972 63 self.assertEqual(response.status_code, 400)
bgneal@972 64
bgneal@972 65 @mock.patch('user_photos.views.HotLinkImageForm')
bgneal@972 66 def testHappyPath(self, form_mock):
bgneal@972 67 form_mock.return_value.is_valid.return_value = True
bgneal@972 68 form_mock.return_value.save.return_value = 'https://example.com/a.jpg'
bgneal@972 69
bgneal@972 70 response = self.client.post(self.view_url,
bgneal@972 71 {'url': 'http://example.com/test.jpg'},
bgneal@972 72 HTTP_X_REQUESTED_WITH='XMLHttpRequest')
bgneal@972 73 self.assertIsInstance(response, JsonResponse)
bgneal@972 74 self.assertEqual(response.status_code, 200)
bgneal@972 75 expected = {"url": "https://example.com/a.jpg", "error_msg": ""}
bgneal@972 76 actual = json.loads(response.content)
bgneal@972 77 self.assertEqual(actual, expected)
bgneal@972 78
bgneal@972 79 @mock.patch('user_photos.views.HotLinkImageForm')
bgneal@972 80 def testFormSaveRaises(self, form_mock):
bgneal@972 81 form_mock.return_value.is_valid.return_value = True
bgneal@972 82 form_mock.return_value.save.side_effect = ValueError
bgneal@972 83
bgneal@972 84 response = self.client.post(self.view_url,
bgneal@972 85 {'url': 'http://example.com/test.jpg'},
bgneal@972 86 HTTP_X_REQUESTED_WITH='XMLHttpRequest')
bgneal@972 87 self.assertIsInstance(response, JsonResponse)
bgneal@972 88 self.assertEqual(response.status_code, 500)