From 62c6403cc35a3381da33c47d951b58367e7dfff7 Mon Sep 17 00:00:00 2001
From: liumengyi <liumengyi@easycorp.ltd>
Date: Thu, 1 Feb 2024 14:08:20 +0800
Subject: [PATCH] * Optimize code for unit test of task.

---
 module/task/test/task.class.php | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/module/task/test/task.class.php b/module/task/test/task.class.php
index 54c58cbb69..fd865c61a7 100755
--- a/module/task/test/task.class.php
+++ b/module/task/test/task.class.php
@@ -363,6 +363,7 @@ class taskTest
         $task->id           = $taskID;
         $task->status       = $status;
         $task->lastEditedBy = $this->objectModel->app->user->account;
+        $task->project      = 10 + $taskID;
 
         $postData = new stdclass();
         $postData->team         = $team;
@@ -994,9 +995,7 @@ class taskTest
     public function addTaskEffortTest($data)
     {
         $data->date = date("Y-m-d");
-        $this->objectModel->addTaskEffort($data);
-
-        $objectID = $this->objectModel->dao->lastInsertID();
+        $objectID = $this->objectModel->addTaskEffort($data);
         $object   = $this->objectModel->getEffortByID($objectID);
 
         if(dao::isError())
@@ -1555,8 +1554,9 @@ class taskTest
     public function manageTaskTeamTest(int $taskID, string $taskStatus, string $mode, array $teamData): array|false
     {
         $task = new stdclass();
-        $task->id     = $taskID;
-        $task->status = $taskStatus;
+        $task->id      = $taskID;
+        $task->status  = $taskStatus;
+        $task->project = 10 + $taskID;
 
         $teamInfo = new stdclass();
         foreach($teamData as $key => $value) $teamInfo->{$key} = $value;