From b2b8546e3c1ead11ab54efa35057cd0c2dca3b7c Mon Sep 17 00:00:00 2001 From: Tom Price Date: Fri, 18 Mar 2016 16:41:13 +0000 Subject: [PATCH] Add test for events_in_bounds --- RIGS/test_models.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/RIGS/test_models.py b/RIGS/test_models.py index a46410a2..a7896cbe 100644 --- a/RIGS/test_models.py +++ b/RIGS/test_models.py @@ -239,6 +239,32 @@ class EventTestCase(TestCase): event.end_time = time(23, 00) self.assertEqual(event.latest_time, self.create_datetime(2016, 1, 2, 23, 00)) + def test_in_bounds(self): + manager = models.Event.objects + events = [ + manager.create(name="TE IB0", start_date='2016-01-02'), # yes no + manager.create(name="TE IB1", start_date='2015-12-31', end_date='2016-01-04'), + + # basic checks + manager.create(name='TE IB2', start_date='2016-01-02', end_date='2016-01-04'), + manager.create(name='TE IB3', start_date='2015-12-31', end_date='2016-01-03'), + manager.create(name='TE IB4', start_date='2016-01-04', access_at='2016-01-03'), + manager.create(name='TE IB5', start_date='2016-01-04', meet_at='2016-01-02'), + + # negative check + manager.create(name='TE IB6', start_date='2015-12-31', end_date='2016-01-01'), + ] + + in_bounds = manager.events_in_bounds(datetime(2016, 1, 2), datetime(2016, 1, 3)) + self.assertIn(events[0], in_bounds) + self.assertIn(events[1], in_bounds) + self.assertIn(events[2], in_bounds) + self.assertIn(events[3], in_bounds) + self.assertIn(events[4], in_bounds) + self.assertIn(events[5], in_bounds) + + self.assertNotIn(events[6], in_bounds) + def create_datetime(self, year, month, day, hour, min): tz = pytz.timezone(settings.TIME_ZONE) return tz.localize(datetime(year, month, day, hour, min))