From ca015c4f94b0a3031f9ff5b4296f9a8b5a9684d9 Mon Sep 17 00:00:00 2001 From: Tom Price Date: Tue, 31 Mar 2015 19:05:09 +0100 Subject: [PATCH] Add current event events test: FAILING. --- RIGS/test_models.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/RIGS/test_models.py b/RIGS/test_models.py index b0ea0d76..dff35b45 100644 --- a/RIGS/test_models.py +++ b/RIGS/test_models.py @@ -26,7 +26,7 @@ class EventTestCase(TestCase): 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 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") @@ -57,4 +57,13 @@ class EventTestCase(TestCase): 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 + self.assertEqual(models.Event.objects.rig_count(), 7) + + def test_current_events(self): + # by my count 7 + 4 + 1 + current_events = models.Event.objects.current_events() + # for event in current_events: + # print event + self.assertEqual(len(current_events), 7+4+1) + self.assertIn(models.Event.objects.get(name="TE E12"), current_events) + \ No newline at end of file