# -*- 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.objects.create(name='test') loc = Location.objects.create(name='test') self.user = User.objects.create_user(username='test') self.booked_duck = Duck.objects.create(species=spec, location=loc, donated_by=self.user) self.comp = Competence.objects.create(name='test', added_by=self.user) booking = Booking.objects.create(duck=self.booked_duck, user=self.user, comp_req=self.comp) self.unbooked_duck = Duck.objects.create(species=spec, location=loc, donated_by=self.user) 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) def test_multiple_booking(self): Booking.objects.create(duck=self.booked_duck, user=self.user, comp_req=self.comp) with self.assertRaises(RuntimeError): self.booked_duck.booked_by()