Mercurial > public > sg101
view bandmap/tests/test_views.py @ 1037:7e0c3cbd3cda
Fix bad select_related call.
In Django 1.8, select_related now throws an error if you give it an invalid
field. This was happening. Fix that query.
Also noticed an extra query generated in the display_post template. Fixed.
author | Brian Neal <bgneal@gmail.com> |
---|---|
date | Tue, 29 Dec 2015 22:21:42 -0600 |
parents | 5103edd3acc4 |
children |
line wrap: on
line source
""" Unit tests for the bandmap application views. """ import datetime import json from django.test import TestCase from django.core.urlresolvers import reverse from django.contrib.auth.models import User from bandmap.models import BandEntry class NotLoggedInTestCase(TestCase): """Tests for a non-authenticated user. """ LOGIN_URL = reverse('accounts-login') def test_map(self): url = reverse('bandmap-map') response = self.client.get(url) self.assertEqual(response.status_code, 200) def test_get_add(self): url = reverse('bandmap-add') response = self.client.get(url, follow=True) self.assertRedirects(response, self.LOGIN_URL + '?next=' + url) def test_post_add(self): url = reverse('bandmap-add') response = self.client.post(url, follow=True) self.assertRedirects(response, self.LOGIN_URL + '?next=' + url) class BasicTestCase(TestCase): """Tests for an authenticated user.""" def setUp(self): self.user = User.objects.create_user( username='pj', email='pj@example.com', password='top_secret') self.client.login(username='pj', password='top_secret') def test_map(self): url = reverse('bandmap-map') response = self.client.get(url) self.assertEqual(response.status_code, 200) def test_get_add(self): url = reverse('bandmap-add') response = self.client.get(url, follow=True) self.assertEqual(response.status_code, 200) def test_post_add(self): url = reverse('bandmap-add') post_data = { 'name': 'Test Band', 'url': 'http://example.com/', 'location': 'Here & There', 'note': 'They suck', 'is_active': 'on', 'lat': '42.0', 'lon': '-23.5', } now = datetime.datetime.now() delta_t = datetime.timedelta(seconds=1) response = self.client.post(url, data=post_data, follow=True) self.assertRedirects(response, url) self.assertEqual(BandEntry.objects.all().count(), 1) entry = BandEntry.objects.get(pk=1) self.assertEqual(entry.name, post_data['name']) self.assertEqual(entry.user, self.user) self.assertEqual(entry.user, self.user) self.assertTrue(now - entry.date_submitted < delta_t) self.assertIsNone(entry.date_approved) self.assertEqual(entry.url, post_data['url']) self.assertEqual(entry.location, post_data['location']) self.assertAlmostEqual(entry.lat, float(post_data['lat'])) self.assertAlmostEqual(entry.lon, float(post_data['lon'])) self.assertEqual(entry.note, post_data['note']) self.assertTrue(entry.is_active) self.assertFalse(entry.is_approved) class QueryTestCase(TestCase): """Testing the query view function which returns JSON.""" def setUp(self): self.user = User.objects.create_user( username='pj', email='pj@example.com', password='top_secret') now = datetime.datetime.now() kwargs = { 'name': 'Test Band 1', 'user': self.user, 'date_submitted': now, 'date_approved': now, 'url': 'http://example.com', 'location': 'Somewhere', 'lat': 42.0, 'lon': -30.0, 'note': 'Some text here', 'is_active': True, 'is_approved': True, } BandEntry.objects.create(**kwargs) kwargs['name'] = 'Test Band 2' kwargs['is_active'] = False BandEntry.objects.create(**kwargs) kwargs['name'] = 'Test Band 3' kwargs['is_active'] = True kwargs['is_approved'] = False BandEntry.objects.create(**kwargs) def test_query(self): url = reverse('bandmap-query') response = self.client.get(url) self.assertEqual(response.status_code, 200) result = json.loads(response.content) self.assertEqual(len(result), 2) self.assertEqual(result[0]['name'], 'Test Band 1') self.assertEqual(result[1]['name'], 'Test Band 2') def test_query_all(self): url = reverse('bandmap-query') response = self.client.get(url, {'show': 'all'}) self.assertEqual(response.status_code, 200) result = json.loads(response.content) self.assertEqual(len(result), 2) self.assertEqual(result[0]['name'], 'Test Band 1') self.assertEqual(result[1]['name'], 'Test Band 2') def test_query_active(self): url = reverse('bandmap-query') response = self.client.get(url, {'show': 'active'}) self.assertEqual(response.status_code, 200) result = json.loads(response.content) self.assertEqual(len(result), 1) self.assertEqual(result[0]['name'], 'Test Band 1') def test_query_inactive(self): url = reverse('bandmap-query') response = self.client.get(url, {'show': 'inactive'}) self.assertEqual(response.status_code, 200) result = json.loads(response.content) self.assertEqual(len(result), 1) self.assertEqual(result[0]['name'], 'Test Band 2')