def test_validate_task_type(self, _): """Test validate start task type.""" tgt = task_group_task.TaskGroupTask() self.assertRaises(ValueError, tgt.validate_task_type, "task_type", "helloh") self.assertEqual("menu", tgt.validate_task_type("task_type", "menu"))
def test_validate_date(self): t = task_group_task.TaskGroupTask() self.assertEqual(date(2002, 4, 16), t.validate_date(date(2, 4, 16))) self.assertEqual(date(2014, 7, 23), t.validate_date(datetime(2014, 7, 23, 22, 5, 7))) self.assertEqual(date(2014, 7, 23), t.validate_date(datetime(2014, 7, 23, 0, 0, 0)))
def test_validate_end_date_decorator(self): t = task_group_task.TaskGroupTask() t.end_date = date(15, 4, 17) self.assertEqual(date(2015, 4, 17), t.end_date) t.start_date = date(2015, 4, 17) self.assertRaises(ValueError, t.validate_end_date, "end_date", date(2014, 2, 5)) t.end_date = date(2015, 2, 17) self.assertEqual(date(2015, 2, 17), t.end_date)
def test_show_view_end_date( # pylint: disable=too-many-arguments self, unit, repeat_every, repeat_multiplier, date, expected, _): """Calculate view end date.""" task_group = all_models.TaskGroup(workflow=all_models.Workflow()) task_group.workflow.unit = unit task_group.workflow.repeat_every = repeat_every task_group.workflow.repeat_multiplier = repeat_multiplier task = task_group_task.TaskGroupTask() task.task_group = task_group task.start_date = date - datetime.timedelta(1) task.end_date = date self.assertEqual(expected, task.view_end_date)
def test_validate_start_date_decorator(self): t = task_group_task.TaskGroupTask() t.start_date = date(16, 4, 21) self.assertEqual(date(2016, 4, 21), t.start_date) t.end_date = date(2016, 4, 21) t.start_date = date(2015, 2, 25) self.assertEqual(date(2015, 2, 25), t.start_date) t.start_date = date(2015, 6, 17) self.assertEqual(date(2015, 6, 17), t.start_date)
def test_validate_start_date(self, _): """Test on validation start date decorator.""" tgt = task_group_task.TaskGroupTask() tgt.start_date = datetime.date(16, 4, 21) self.assertEqual(datetime.date(2016, 4, 21), tgt.start_date) tgt.end_date = datetime.date(2016, 4, 21) tgt.start_date = datetime.date(2015, 2, 25) self.assertEqual(datetime.date(2015, 2, 25), tgt.start_date) tgt.start_date = datetime.date(2015, 6, 17) self.assertEqual(datetime.date(2015, 6, 17), tgt.start_date)
def test_validate_date(self, _): """Test validate date.""" tgt = task_group_task.TaskGroupTask() self.assertEqual( datetime.date(2002, 4, 16), tgt.validate_date('start_date', datetime.date(2, 4, 16))) self.assertEqual( datetime.date(2014, 7, 23), tgt.validate_date('start_date', datetime.datetime(2014, 7, 23, 22, 5, 7))) self.assertEqual( datetime.date(2014, 7, 23), tgt.validate_date('start_date', datetime.datetime(2014, 7, 23, 0, 0, 0)))
def test_validate_task_type(self): t = task_group_task.TaskGroupTask() self.assertRaises(ValueError, t.validate_task_type, "task_type", "helloh") self.assertEqual("menu", t.validate_task_type("task_type", "menu"))