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"))
