def test_register(self): """ Verifies that the register function registers a Task """ task_manager = TaskManager() task_manager.register('TestTask', TestTask) self.assert_(task_manager.registry.has_key('TestTask'), 'Registry does not contain key for task')
def test_listtasks(self): """ Tests list tasks function to verify it returns all the tasks that it should """ task_manager = TaskManager() task_manager.register('TestTask', TestTask) task_manager.register('TestContainerTask', TestContainerTask) task_manager.register('TestParallelTask', TestParallelTask) tasks = task_manager.list_tasks() self.assertEqual(len(tasks), 3, "There should be 3 registered tasks") for task in self.tasks: self.assert_(tasks.has_key(task), 'Task is missing from list tasks') recorded_time = self.completion[task] list_time = tasks[task]['last_run'] list_time = list_time.strftime('%Y-%m-%d %H:%M:%S') if list_time else None self.assertEqual(recorded_time, list_time, "Completion times for task don't match: %s != %s" % (recorded_time, list_time))