word-challenge/words/tests.py

# -*- coding: utf-8
from datetime import timedelta
from django.contrib.auth.models import User
from django.core.exceptions import ValidationError
from django.utils import timezone
from django.utils.dateparse import parse_duration
from django.utils.translation import activate
from django.test import TestCase, override_settings

from .models import Word, Draw, Work

class WordTest(TestCase):
    def setUp(self):
        user = User.objects.create_user(username='test', password='test')
        self.word1 = Word.objects.create(language='en-us',
                                         word='color',
                                         added_by=user)

    def test_word_str(self):
        self.assertEquals('color', self.word1.__str__())

@override_settings(DRAW_TIME='1 00:00:00')
class DrawTest(TestCase):
    def setUp(self):
        self.user = User.objects.create_user(username='test', password='test')
        self.word = Word.objects.create(added_by=self.user,
                                        language='en-us',
                                        word='color')

    def test_current_word(self):
        self.assertIsNone(self.user.current_word())

        draw = Draw.objects.create(user=self.user,
                                   word=self.word,
                                   accepted=None)
        self.assertEquals(self.word, self.user.current_word())

        draw.accepted = True
        draw.save()
        self.assertEquals(self.word, self.user.current_word())

    def test_draw_word(self):
        # User has no words yet
        self.assertEquals(self.word, self.user.draw_word())

        # User now has an unaccepted draw
        self.assertEquals(self.word, self.user.draw_word())

        # Accept the last word and make it appear as if it would be 2
        # days ago
        draw = Draw.objects.get(user=self.user, word=self.word)
        draw.accepted = True
        draw.timestamp -= timedelta(days=2)
        draw.save()
        Work.objects.create(draw=draw)

        # Create a second word for further testing
        word2 = Word.objects.create(added_by=self.user)

        # The next word should be different from the previous one
        self.assertEquals(word2, self.user.draw_word())

        # The new word should not be accepted (as it is a new draw)
        draw = Draw.objects.get(user=self.user, word=word2)
        self.assertIsNotNone(draw)
        self.assertIsNone(draw.accepted)

        # Accept the word, make it old again, and create a work for it
        draw.accepted = True
        draw.timestamp -= timedelta(days=2)
        draw.save()
        work = Work.objects.create(draw=draw)

        # As we are out of words now, a new draw should return None
        self.assertIsNone(self.user.draw_word())

        # Now set the last draw to fresh again, and remove the associated work.
        draw.timestamp = timezone.now()
        draw.save()
        work.delete()
        # Also create a new word
        word3 = Word.objects.create(added_by=self.user,
                                    language='en-gb',
                                    word='colour')

        # A next draw should return the same word in this case
        self.assertEquals(word2, self.user.draw_word())

        # Now let’s reject this draw and draw a new one
        draw = Draw.objects.get(user=self.user, word=word2)
        draw.accepted = False
        draw.save()

        # The next draw should be different from the last
        self.assertEquals(word3, self.user.draw_word())

        # Now make the previous one accepted and completed, and reject
        # this last one
        draw.accepted = True
        draw.save()
        Work.objects.create(draw=draw)

        draw = Draw.objects.get(user=self.user, word=word3)
        draw.accepted = False
        draw.save()

        # The next draw must be this last, rejected one (as there are
        # no other options)
        self.assertEquals(word3, self.user.draw_word())

    def test_last_draw(self):
        draw = Draw.objects.create(
            user=self.user,
            word=self.word,
            accepted=True,
            timestamp=timezone.now() - timedelta(days=1))
        Work.objects.create(draw=draw)
        word = Word.objects.create(added_by=self.user)
        draw = Draw.objects.create(user=self.user,
                            word=word,
                            accepted=True)
        Work.objects.create(draw=draw)

        self.assertEquals(word, self.user.last_draw().word)

    def test_draw_per_day(self):
        draw = Draw.objects.create(user=self.user,
                                   word=self.word,
                                   accepted=True)
        Work.objects.create(draw=draw)
        Word.objects.create(added_by=self.user)

        self.assertEquals(self.word, self.user.draw_word())

    def test_draw_successful(self):
        # If there is no work, but we are within the time range
        draw = Draw.objects.create(
            user=self.user,
            word=Word.objects.create(added_by=self.user),
            accepted=True,
            timestamp=timezone.now())
        self.assertIsNone(draw.successful())

        # If there is no work and we are out of time
        draw.timestamp -= timedelta(days=2)
        draw.save()
        self.assertIsNotNone(draw.successful())
        self.assertFalse(draw.successful())

        # If there is work and it was submitted on time
        draw.timestamp = timezone.now() + timedelta(minutes=1)
        draw.save()
        Work.objects.create(draw=draw)
        self.assertTrue(draw.successful())

        # If there is work and it wasn’t submitted on time
        draw.timestamp = timezone.now() - timedelta(days=2)
        draw.save()
        self.assertIsNotNone(draw.successful())
        self.assertFalse(draw.successful())

    def test_failed_successful_words(self):
        self.assertEquals([], self.user.failed_words())

        draw = Draw.objects.create(user=self.user,
                                   word=self.word,
                                   accepted=True,
                                   timestamp=timezone.now() - timedelta(days=2))
        self.assertEquals([draw.word], self.user.failed_words())
        self.assertEquals([], self.user.successful_words())

        draw2 = Draw.objects.create(user=self.user,
                                    word=Word.objects.create(added_by=self.user),
                                   accepted=True,
                                   timestamp=timezone.now() - timedelta(days=3))
        Work.objects.create(draw=draw2,
                            upload_time=timezone.now() - timedelta(days=3))

        self.assertEquals([draw.word], self.user.failed_words())
        self.assertEquals([draw2.word], self.user.successful_words())