view contests/tests/test_views.py @ 796:5977b43499f7

Modified contests to have multiple winners.
author Brian Neal <bgneal@gmail.com>
date Fri, 04 Jul 2014 20:30:14 -0500
parents 8789299c75b1
children
line wrap: on
line source
"""
View tests for the contests application.

"""
import datetime
import json

from django.test import TestCase
from django.contrib.auth.models import User
from django.core.urlresolvers import reverse

from contests.models import Contest


class NoConstestsTestCase(TestCase):

    def test_no_contests(self):
        response = self.client.get(reverse('contests-index'))
        self.assertEqual(response.status_code, 200)

        url = reverse('contests-contest', kwargs={'slug': 'test'})
        response = self.client.get(url)
        self.assertEqual(response.status_code, 404)


class ConstestsTestCase(TestCase):

    def setUp(self):
        now = datetime.datetime.now()
        start = now - datetime.timedelta(days=7)
        end = start - datetime.timedelta(days=3)

        user = User.objects.create_user('test_user', '', 'password')
        user.save()

        c = Contest(title='test',
                slug='test',
                description='test',
                is_public=True,
                creation_date=start,
                end_date=end,
                win_date=end + datetime.timedelta(days=1),
                num_winners=1)
        c.save()
        c.contestants.add(user)
        c.winners.add(user)
        self.contest_id = c.id

    def test_contests(self):
        response = self.client.get(reverse('contests-index'))
        self.assertEqual(response.status_code, 200)

        url = reverse('contests-contest', kwargs={'slug': 'test'})
        response = self.client.get(url)
        self.assertEqual(response.status_code, 200)


class ContestEntryTestCase(TestCase):

    def setUp(self):
        self.username = 'test_user'
        self.pw = 'password'
        self.user = User.objects.create_user(self.username, '', self.pw)
        self.user.save()
        self.assertTrue(self.client.login(username=self.username,
            password=self.pw))

        now = datetime.datetime.now()
        start = now - datetime.timedelta(days=7)
        end = now + datetime.timedelta(days=3)

        c = Contest(title='test',
                slug='test',
                description='test',
                is_public=True,
                creation_date=start,
                end_date=end,
                num_winners=1)
        c.save()
        self.contest_id = c.id

    def test_entry_toggle(self):
        response = self.client.post(reverse('contests-enter'),
                {'contest_id': self.contest_id},
                HTTP_X_REQUESTED_WITH='XMLHttpRequest')
        self.assertEqual(response.status_code, 200)

        c = json.loads(response.content)
        self.assertTrue(c['entered'])

        contest = Contest.objects.get(pk=self.contest_id)
        self.assertTrue(self.user in contest.contestants.all())

        response = self.client.post(reverse('contests-enter'),
                {'contest_id': self.contest_id},
                HTTP_X_REQUESTED_WITH='XMLHttpRequest')
        self.assertEqual(response.status_code, 200)

        c = json.loads(response.content)
        self.assertFalse(c['entered'])

        contest = Contest.objects.get(pk=self.contest_id)
        self.assertFalse(self.user in contest.contestants.all())


class NoPublicConstestsTestCase(TestCase):

    def setUp(self):
        now = datetime.datetime.now()
        start = now - datetime.timedelta(days=7)
        end = start - datetime.timedelta(days=3)

        c = Contest(title='test',
                slug='test',
                description='test',
                is_public=False,
                creation_date=start,
                end_date=end)
        c.save()

    def test_contests(self):
        response = self.client.get(reverse('contests-index'))
        self.assertEqual(response.status_code, 200)

        url = reverse('contests-contest', kwargs={'slug': 'test'})
        response = self.client.get(url)
        self.assertEqual(response.status_code, 404)