From 1d14db165a88d0d1051bc792e90e775dade735dc Mon Sep 17 00:00:00 2001 From: Tom Price Date: Tue, 31 Mar 2015 18:44:45 +0100 Subject: [PATCH] Add events and check count/rig_count --- RIGS/test_models.py | 45 ++++++++++++++++++++++++++++++++++++++++++++- 1 file changed, 44 insertions(+), 1 deletion(-) diff --git a/RIGS/test_models.py b/RIGS/test_models.py index dfc0824f..b0ea0d76 100644 --- a/RIGS/test_models.py +++ b/RIGS/test_models.py @@ -1,5 +1,6 @@ from django.test import TestCase from RIGS import models +from datetime import date, timedelta class VatRateTestCase(TestCase): def setUp(self): @@ -14,4 +15,46 @@ class VatRateTestCase(TestCase): def test_percent_correct(self): r = models.VatRate.objects.get(rate=0.20) - self.assertEqual(r.as_percent, 20) \ No newline at end of file + self.assertEqual(r.as_percent, 20) + +class EventTestCase(TestCase): + def setUp(self): + self.vatrate = models.VatRate.objects.create(start_at='2014-03-01',rate=0.20,comment='test1') + self.person = models.Person.objects.create(name="TE P1") + self.organisation = models.Organisation.objects.create(name="TE O1") + self.venue = models.Venue.objects.create(name="TE V1") + self.profile = models.Profile.objects.create(username="testuser1", email="1@test.com") + + # produce 7 normal events + models.Event.objects.create(name="TE E1", start_date=date.today() + timedelta(days=6), description="start future no end") + models.Event.objects.create(name="TE E2", start_date=date.today(), description="start today no end") + models.Event.objects.create(name="TE E3", start_date=date.today(), end_date=date.today(), description="start today with end") + models.Event.objects.create(name="TE E4", start_date='2014-03-20', description="start past no end") + models.Event.objects.create(name="TE E5", start_date='2014-03-20', end_date='2014-03-21', description="start past with end") + models.Event.objects.create(name="TE E6", start_date=date.today()-timedelta(days=2), end_date=date.today()+timedelta(days=2), description="start past, end future") + models.Event.objects.create(name="TE E7", start_date=date.today()+timedelta(days=2), end_date=date.today()+timedelta(days=2), description="start + end in future") + + # 2 cancelled + models.Event.objects.create(name="TE E8", start_date=date.today()+timedelta(days=2), end_date=date.today()+timedelta(days=2), status=models.Event.CANCELLED, description="cancelled in future") + models.Event.objects.create(name="TE E9", start_date=date.today()-timedelta(days=1), end_date=date.today()+timedelta(days=2), status=models.Event.CANCELLED, description="cancelled and started") + + # 5 dry hire + models.Event.objects.create(name="TE E10", start_date=date.today(), dry_hire=True, description="dryhire today") + models.Event.objects.create(name="TE E11", start_date=date.today(), dry_hire=True, checked_in_by=self.profile, description="dryhire today, checked in") + models.Event.objects.create(name="TE E12", start_date=date.today()-timedelta(days=1), dry_hire=True, description="dryhire past") + models.Event.objects.create(name="TE E13", start_date=date.today()-timedelta(days=1), dry_hire=True, checked_in_by=self.profile, description="dryhire past checked in") + models.Event.objects.create(name="TE E14", start_date=date.today(), dry_hire=True, status=models.Event.CANCELLED, description="dryhire today cancelled") + + # 4 non rig + models.Event.objects.create(name="TE E15", start_date=date.today(), is_rig=False, description="non rig today") + models.Event.objects.create(name="TE E16", start_date=date.today()+timedelta(days=1), is_rig=False, description="non rig tomorrow") + models.Event.objects.create(name="TE E17", start_date=date.today()-timedelta(days=1), is_rig=False, description="non rig yesterday") + models.Event.objects.create(name="TE E18", start_date=date.today(), is_rig=False, status=models.Event.CANCELLED, description="non rig today cancelled") + + def test_count(self): + # Santiy check we have the expected events created + self.assertEqual(models.Event.objects.count(), 18, "Incorrect number of events, check setup") + + def test_rig_count(self): + # by my count this is 7 + self.assertEqual(models.Event.objects.rig_count(), 7) \ No newline at end of file