def __handle_finish_tasks(self):
tasks = []
for item in self.commands[1:]:
task = Task()
task.index = item
task.finish_time = task.get_current_time()
tasks.append(task)
return BehaviorEnum.FINISH_TASKS, tasks
def __handle_delete_tasks(self):
params = sorted(self.commands[1:], reverse=True)
tasks = []
for item in params:
task = Task()
task.index = item
tasks.append(task)
return BehaviorEnum.DELETE_TASKS, tasks
def __handle_update_tasks(self):
params = self.commands[1:]
ids = params[::2]
contents = params[1::2]
if len(ids) != len(contents):
return BehaviorEnum.ERROR_COMMAND, 'Error Command: [ Update params error]'
tasks = []
for i in range(len(ids)):
task = Task()
task.index = ids[i]
task.content = contents[i]
tasks.append(task)
return BehaviorEnum.UPDATE_TASKS, tasks
def __handle_task(self):
task = Task()
task.index = self.commands[1]
command3 = self.commands[2]
if command3 == 'a' or command3 == 'add':
return self.__handle_add_task_steps(task)
if command3 == 'd' or command3 == 'delete':
return self.__handle_delete_task_steps(task)
if command3 == 'u' or command3 == 'update':
return self.__handle_update_task_steps(task)
if command3 == 'f' or command3 == 'finish':
return self.__handle_finish_task_steps(task)
return BehaviorEnum.ERROR_COMMAND, 'Error Command: [ \'' + command3 + '\']'
def test_task(self):
user_token = 'eyJhbGciOiJIUzI1NiJ9.' \
'eyJ1c2VySWQiOiJVMTQ2OTUxOTY1MTY3MDIwMDMyMCJ9.' \
'm-pLYhQJgHM4l_d5LzyiM7jjMNDKij49GTJRSRSCIss'
# Add task
task = Task()
task.content = 'Task'
added_task = self.api.add_task(task, user_token)
self.assertEqual(task.content, added_task.content)
# Update task content
task.content = 'new Task'
task.task_id = added_task.task_id
updated_task = self.api.update_task(task, user_token)
self.assertEqual(task.content, updated_task.content)
# Update task status and finish time
task.content = None
task.status = '1'
task.finish_time = '2022-01-01 10:00:00'
finished_task = self.api.update_task(task, user_token)
self.assertEqual(task.status, finished_task.status)
self.assertEqual(task.finish_time, finished_task.finish_time)
# Delete task
deleted_task = self.api.delete_task(task, user_token)
self.assertIsNone(deleted_task)
# Find tasks
selected_tasks = self.api.find_tasks(user_token)
self.assertIsNotNone(selected_tasks)
def test_manage_task_step(self):
task_step = TaskStep()
task_step.task_step_id = TaskStep.get_random_task_step_id()
task_step.task_id = Task.get_random_task_id()
task_step.create_time = TaskStep.get_current_time()
task_step.content = 'This is a task step'
self.task_step_dao.insert_task_step(task_step)
new_task_step = self.task_step_dao.find_task_step_by_task_step_id(
task_step.task_step_id)
self.assertEqual(task_step.content, new_task_step.content)
self.assertIsNotNone(new_task_step.create_time)
task_step.content = 'This is a new task step'
self.task_step_dao.update_task_step_content(task_step)
new_task_step = self.task_step_dao.find_task_step_by_task_step_id(
task_step.task_step_id)
self.assertEqual(task_step.content, new_task_step.content)
task_step.status = TaskStep.Status.FINISH.value
task_step.finish_time = task_step.get_current_time()
self.task_step_dao.update_task_step_status_and_finish_time(task_step)
new_task_step = self.task_step_dao.find_task_step_by_task_step_id(
task_step.task_step_id)
self.assertEqual(task_step.status, new_task_step.status)
self.assertIsNotNone(new_task_step.finish_time)
self.task_step_dao.delete_task_step(task_step)
new_task_step = self.task_step_dao.find_task_step_by_task_step_id(
task_step.task_step_id)
self.assertIsNone(new_task_step)
def __is_today(create_time):
today_time = Task.get_current_time()
today_date_list = today_time.split(" ")[0].split("-")
create_date_list = create_time.split(" ")[0].split("-")
is_year_equal = int(today_date_list[0]) == int(create_date_list[0])
is_month_equal = int(today_date_list[1]) == int(create_date_list[1])
is_day_equal = int(today_date_list[2]) == int(create_date_list[2])
return is_year_equal and is_month_equal and is_day_equal
def convert_to_task(task_json):
task = Task()
task.task_id = task_json['taskId']
task.status = task_json['status']
task.content = task_json['content']
task.create_time = task_json['createTime']
task.finish_time = task_json['finishTime']
return task
def __trans_to_task(task_tuple):
if task_tuple is None:
return None
task = Task()
task.task_id = task_tuple[0]
task.status = task_tuple[1]
task.content = task_tuple[2]
task.create_time = task_tuple[3]
task.deadline_time = task_tuple[4]
task.finish_time = task_tuple[5]
return task
def __handle_add_tasks(self):
tasks = []
for item in self.commands[1:]:
task = Task()
task.task_id = task.get_random_task_id()
task.create_time = task.get_current_time()
task.content = item
tasks.append(task)
return BehaviorEnum.ADD_TASKS, tasks
def test_manage_task(self):
task = Task()
task.task_id = task.get_random_task_id()
task.create_time = task.get_current_time()
task.content = 'This is a task'
self.task_dao.insert_task(task)
new_task = self.task_dao.find_task_by_task_id(task.task_id)
self.assertEqual(task.content, new_task.content)
self.assertIsNotNone(new_task.create_time)
task.content = 'This is a new task'
self.task_dao.update_task_content(task)
new_task = self.task_dao.find_task_by_task_id(task.task_id)
self.assertEqual(task.content, new_task.content)
task.status = Task.Status.FINISH.value
task.finish_time = task.get_current_time()
self.task_dao.update_task_status_and_finish_time(task)
new_task = self.task_dao.find_task_by_task_id(task.task_id)
self.assertEqual(task.status, new_task.status)
self.assertIsNotNone(new_task.finish_time)
self.task_dao.delete_task(task.task_id)
new_task = self.task_dao.find_task_by_task_id(task.task_id)
self.assertIsNone(new_task)