def test_basic_menu_category(self): """ Test input to MenuCategory model which deal with choices and timefield """ test_name = "dinner" test_description = "Best Dinner" test_menu_category = MenuCategory(name = test_name, description = test_description, start_day = 0, end_day = 6, start_time = datetime.time(16,00), end_time = datetime.time(22,00)) self.assertEqual(test_menu_category.name, test_name) self.assertEqual(test_menu_category.get_end_day_display(), "Sunday") self.assertEqual(test_menu_category.start_time.hour, 16)
def test_contain_curr_time_mc(self): """ Test function contain_curr_time in MenuCategory model """ test_menu_category = MenuCategory(name = "dinner", description = "Best Dinner", start_day = 0, end_day = 5, start_time = datetime.time(16,00), end_time = datetime.time(22,00)) curr_date_time1 = datetime.datetime(2014, 4, 9, 16, 55, 52) self.assertTrue(test_menu_category.contain_curr_time(curr_date_time1)) curr_date_time2 = datetime.datetime(2014, 4, 6, 16, 55, 52) self.assertFalse(test_menu_category.contain_curr_time(curr_date_time2)) curr_date_time3 = datetime.datetime(2014, 4, 9, 15, 55, 52) self.assertFalse(test_menu_category.contain_curr_time(curr_date_time3))