duck-booking-tool/booking/tests.py

# -*- coding: utf-8 -*-
from django.conf import settings
from django.contrib.auth.models import User
from django.core.urlresolvers import reverse
from django.test import TestCase, Client
from django.utils import timezone

import datetime

from .ducklevel import level_to_up_minutes, level_to_down_minutes, minutes_to_level
from .templatetags import booking_tags
from .models import Duck, Competence, DuckCompetence, Species, Location, Booking

class FrontTest(TestCase):
    def setUp(self):
        self.client = Client()

    def test_index_page(self):
        response = self.client.get('/')
        self.assertEqual(response.status_code, 200)

    def test_vocabulary_page(self):
        response = self.client.get('/vocabulary.html')
        self.assertEqual(response.status_code, 200)

    def test_terms_page(self):
        response = self.client.get('/terms.html')
        self.assertEqual(response.status_code, 200)

    def test_disclaimer_page(self):
        response = self.client.get('/disclaimer.html')
        self.assertEqual(response.status_code, 200)

class DuckCompLevelTest(TestCase):
    def setUp(self):
        user = User.objects.create_user(username='test', password='test')

        species = Species(name='test species')
        species.save()

        location = Location(name='test location')
        location.save()

        duck = Duck(
                species=species,
                location=location,
                donated_by=user)
        duck.save()

        comp = Competence(
                name='testing',
                added_by=user)
        comp.save()

        self.duckcomp = DuckCompetence(
            duck = duck,
            comp = comp,
            up_minutes = 0,
            down_minutes =0)

    def test_sane_max(self):
        self.assertGreater(
            settings.MAX_DUCK_LEVEL, 0,
            msg = "MAX_DUCK_LEVEL must be greater than zero!")

    def test_max_minutes(self):
        """Test if level can not go above settings.MAX_DUCK_LEVEL)"""

        max_up_minutes = level_to_up_minutes(settings.MAX_DUCK_LEVEL)
        double_minutes = level_to_up_minutes(settings.MAX_DUCK_LEVEL * 2)
        max_down_minutes = level_to_down_minutes(settings.MAX_DUCK_LEVEL)

        level = minutes_to_level(max_up_minutes, 0)
        self.assertEqual(level, settings.MAX_DUCK_LEVEL)

        level = minutes_to_level(max_up_minutes + 1, 0)
        self.assertEqual(level, settings.MAX_DUCK_LEVEL)

        level = minutes_to_level(double_minutes, 0)
        self.assertEqual(level, settings.MAX_DUCK_LEVEL)

        level = minutes_to_level(0, max_down_minutes)
        self.assertEqual(level, settings.MAX_DUCK_LEVEL)

        level = minutes_to_level(0, max_down_minutes + 1)
        self.assertEqual(level, settings.MAX_DUCK_LEVEL)

    def test_conversions(self):
        for i in range(1, settings.MAX_DUCK_LEVEL):
            up_minutes = level_to_up_minutes(i)
            down_minutes = level_to_down_minutes(i)

            up_level = minutes_to_level(up_minutes, 0)
            down_level = minutes_to_level(0, down_minutes)

            self.assertEqual(up_level, i, msg = "Test failed for value %d" % i)
            self.assertEqual(
                down_level, i,
                msg = "Test failed for value %d" % i)

    def test_level_to_minutes(self):
        self.assertEqual(level_to_up_minutes(0), 0)
        self.assertEqual(level_to_up_minutes(1), 20)
        self.assertEqual(level_to_up_minutes(2), 200)
        self.assertEqual(level_to_up_minutes(3), 2000)
        self.assertEqual(level_to_up_minutes(4), 20000)
        self.assertEqual(level_to_up_minutes(5), 200000)

        self.assertEqual(level_to_down_minutes(0), 0)
        self.assertEqual(level_to_down_minutes(1), 200)
        self.assertEqual(level_to_down_minutes(2), 2000)
        self.assertEqual(level_to_down_minutes(3), 20000)
        self.assertEqual(level_to_down_minutes(4), 200000)
        self.assertEqual(level_to_down_minutes(5), 2000000)

    def test_no_comp(self):
        self.duckcomp.up_minutes = 0
        self.duckcomp.down_minutes = 0
        self.assertEquals(self.duckcomp.level(), 0)

    def test_comp_levels(self):
        self.duckcomp.down_minutes = 0

        for lvl in range(1, settings.MAX_DUCK_LEVEL):
            minutes = level_to_up_minutes(lvl)
            self.duckcomp.up_minutes = minutes
            self.assertEqual(self.duckcomp.level(), lvl)

    def test_high_minutes(self):
        self.duckcomp.up_minutes = level_to_up_minutes(settings.MAX_DUCK_LEVEL)
        self.duckcomp.down_minutes = level_to_down_minutes(settings.MAX_DUCK_LEVEL)
        self.assertEqual(self.duckcomp.level(), settings.MAX_DUCK_LEVEL)

class DuckAgeTest(TestCase):
    def test_duck_is_from_the_future(self):
        future_duck = Duck(donated_at = timezone.now() + datetime.timedelta(days = 2))
        self.assertEqual(future_duck.age(), -1)

    def test_duck_age_formatter(self):
        self.assertEqual(booking_tags.age_format("aoeu"), "aoeu")
        self.assertEqual(booking_tags.age_format(0), "a few moments")
        self.assertEqual(booking_tags.age_format(1), "1 second")
        self.assertEqual(booking_tags.age_format(2), "2 seconds")
        self.assertEqual(booking_tags.age_format(60), "1 minute")
        self.assertEqual(booking_tags.age_format(61), "1 minute 1 second")
        self.assertEqual(booking_tags.age_format(62), "1 minute 2 seconds")
        self.assertEqual(booking_tags.age_format(120), "2 minutes")
        self.assertEqual(booking_tags.age_format(3600), "1 hour")
        self.assertEqual(booking_tags.age_format(3601), "1 hour 1 second")
        self.assertEqual(booking_tags.age_format(3660), "1 hour 1 minute")
        self.assertEqual(booking_tags.age_format(3720), "1 hour 2 minutes")
        self.assertEqual(booking_tags.age_format(7200), "2 hours")
        self.assertEqual(booking_tags.age_format(86400), "1 day")
        self.assertEqual(booking_tags.age_format(86401), "1 day 1 second")
        self.assertEqual(booking_tags.age_format(86460), "1 day 1 minute")
        self.assertEqual(booking_tags.age_format(90000), "1 day 1 hour")
        self.assertEqual(booking_tags.age_format(93600), "1 day 2 hours")
        self.assertEqual(booking_tags.age_format(172800), "2 days")
        self.assertEqual(booking_tags.age_format(2592000), "1 month")
        self.assertEqual(booking_tags.age_format(2592001), "1 month 1 second")
        self.assertEqual(booking_tags.age_format(2592060), "1 month 1 minute")
        self.assertEqual(booking_tags.age_format(2595600), "1 month 1 hour")
        self.assertEqual(booking_tags.age_format(2678400), "1 month 1 day")
        self.assertEqual(booking_tags.age_format(2764800), "1 month 2 days")
        self.assertEqual(booking_tags.age_format(5184000), "2 months")
        self.assertEqual(booking_tags.age_format(31536000), "1 year")
        self.assertEqual(booking_tags.age_format(31536001), "1 year 1 second")
        self.assertEqual(booking_tags.age_format(31536060), "1 year 1 minute")
        self.assertEqual(booking_tags.age_format(31539600), "1 year 1 hour")
        self.assertEqual(booking_tags.age_format(31622400), "1 year 1 day")
        self.assertEqual(booking_tags.age_format(34128000), "1 year 1 month")
        self.assertEqual(booking_tags.age_format(36720000), "1 year 2 months")
        self.assertEqual(booking_tags.age_format(63072000), "2 years")

class BookingTimeTest(TestCase):
    duck1 = None
    duck2 = None

    def setUp(self):
        user = User()
        user.save()

        species = Species(name = 'duck')
        species.save()

        location = Location(name = 'start')
        location.save()

        self.duck1 = Duck(species = species, location = location, donated_by = user)
        self.duck1.save()

        competence = Competence(name = 'test', added_by = user)
        competence.save()

        now = timezone.now()
        booking = Booking(duck = self.duck1, start_ts = now - datetime.timedelta(days = 2), end_ts = now - datetime.timedelta(days = 1), user = user, comp_req = competence)
        booking.save()

        self.duck2 = Duck(species = species, location = location, donated_by = user)
        self.duck2.save()

        booking = Booking(duck = self.duck2, start_ts = now - datetime.timedelta(days = 3), end_ts = now - datetime.timedelta(days = 2), user = user, comp_req = competence)
        booking.save()

        booking = Booking(duck = self.duck2, start_ts = now - datetime.timedelta(days = 2), end_ts = now - datetime.timedelta(days = 1), user = user, comp_req = competence)
        booking.save()

    def test_total_booking_time(self):
        self.assertEqual(Booking.total_booking_time(), 259200)

    def test_duck_booking_time(self):
        self.assertEqual(Booking.duck_booking_time(self.duck1), 86400)
        self.assertEqual(Booking.duck_booking_time(self.duck2), 172800)

    def test_dpx(self):
        self.assertEqual(self.duck1.dpx(), 1/3)
        self.assertEqual(self.duck2.dpx(), 2/3)

class TestListing(TestCase):
    def setUp(self):
        self.client = Client()

        species = Species()
        species.save()

        loc = Location()
        loc.save()

        user = User()
        user.save()

        self.duck = Duck(species = species, location = loc, donated_by = user)
        self.duck.save()

    def test_front_page(self):
        response = self.client.get('/')
        self.assertEqual(response.status_code, 200)

        self.assertEqual(len(response.context['duck_list']), 1)

class SimilarCompTest(TestCase):
    def setUp(self):
        admin = User(username = 'admin')
        admin.save()

        competence_list = (
            'Creativity',
            'Administration',
            'Perl',
            'Python',
            'TCSH',
        )

        for c in competence_list:
            comp = Competence(name = c, added_by = admin)
            comp.save()

    def test_good_similar_competences(self):
        l = Competence.get_similar_comps('perl')
        self.assertEquals(len(l), 1)

        l = Competence.get_similar_comps('pzthon')
        self.assertEquals(len(l), 1)

        l = Competence.get_similar_comps(u'kreativitás')
        self.assertEqual(len(l), 1)

    def test_bad_similar_competence(self):
        l = Competence.get_similar_comps('development')
        self.assertEqual(len(l), 0)

class BookingTest(TestCase):
    def setUp(self):
        spec = Species(name = 'test')
        spec.save()

        loc = Location(name = 'test')
        loc.save()

        user = User(username = 'test')
        user.save()

        self.booked_duck = Duck(species = spec, location = loc, donated_by = user)
        self.booked_duck.save()

        comp = Competence(name = 'test', added_by = user)
        comp.save()

        booking = Booking(duck = self.booked_duck, user = user, comp_req = comp)
        booking.save()

        self.unbooked_duck = Duck(species = spec, location = loc, donated_by = user)
        self.unbooked_duck.save()

    def test_booked_duck(self):
        self.assertNotEqual(self.booked_duck.booked_by(), None)

    def test_unbooked_duck(self):
        self.assertEqual(self.unbooked_duck.booked_by(), None)