def test_basic_business_hours(self): """ Test input to BusinessHours model which deal with choices and timefield """ test_bus_hours1 = BusinessHours(day = 4, open_time = datetime.time(9,30), close_time = datetime.time(16,00)) self.assertEqual(test_bus_hours1.get_day_display(), "Friday") self.assertEqual(test_bus_hours1.open_time.hour, 9) self.assertEqual(test_bus_hours1.close_time.minute, 0)
def test_contain_curr_time_bh(self): """ Test function contain_curr_time in BusinessHours model """ curr_date_time = datetime.datetime(2014, 4, 9, 15, 55, 52) test_bus_hours1 = BusinessHours(day = curr_date_time.weekday(), open_time = datetime.time(9,30), close_time = datetime.time(16,00)) test_bus_hours2 = BusinessHours(day = curr_date_time.weekday(), open_time = datetime.time(9,30), close_time = datetime.time(15,00)) test_bus_hours3 = BusinessHours(day = 0, open_time = datetime.time(9,30), close_time = datetime.time(21,00)) self.assertTrue(test_bus_hours1.contain_curr_time(curr_date_time)) self.assertFalse(test_bus_hours2.contain_curr_time(curr_date_time)) self.assertFalse(test_bus_hours3.contain_curr_time(curr_date_time))