def POST(task_owner, task_id, request_payload): if task_owner is None: return { const.KEY_RESP_STATUS: "ERROR", const.KEY_RESP_STATUS_TEXT: "specified TaskOwner object not found" } taskPayloadHelper = TaskPayloadHelper(request_payload) if not taskPayloadHelper.is_valid() or taskPayloadHelper.is_duplicate(): return { const.KEY_RESP_STATUS: "ERROR", const.KEY_RESP_STATUS_TEXT: "payload is invalid or already exists" } if task_id is None: task = Task(owner=task_owner, source=taskPayloadHelper[const.KEY_SOURCE], dest=taskPayloadHelper[const.KEY_DEST], content=taskPayloadHelper[const.KEY_CONTENT], ready_time=taskPayloadHelper.get_ready_datetime()) task.save() task_id = task.id else: task = _get_task_by_id(task_id) task.source = taskPayloadHelper[const.KEY_SOURCE] task.dest = taskPayloadHelper[const.KEY_DEST] task.content = taskPayloadHelper[const.KEY_CONTENT] task.ready_time = taskPayloadHelper.get_ready_datetime() task.save() return { const.KEY_ID: task_id, const.KEY_SOURCE: taskPayloadHelper[const.KEY_SOURCE], const.KEY_DEST: taskPayloadHelper[const.KEY_DEST], const.KEY_CONTENT: taskPayloadHelper[const.KEY_CONTENT], const.KEY_READY_TIME: taskPayloadHelper[const.KEY_READY_TIME] }
def setUp(self): self.taskPayloadHelper = TaskPayloadHelper(TEST_PAYLOAD) self.owner = TaskOwner(email='*****@*****.**', key='example') self.owner.save()