view bandmap/tests/test_views.py @ 1157:e4f2d6a4b401

Rework S3 connection logic for latest versions of Python 2.7. Had to make these changes for Ubuntu 16.04. Seems backward compatible with production.
author Brian Neal <bgneal@gmail.com>
date Thu, 19 Jan 2017 18:35:53 -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')