From b786e4a428bc09eabc85a6597ead057e2e0c3680 Mon Sep 17 00:00:00 2001
From: caoyanyi <caoyanyi@easycorp.ltd>
Date: Fri, 13 Jan 2023 13:32:54 +0800
Subject: [PATCH] * Add test case.

---
 test/class/execution.class.php           | 16 ++++++
 test/model/execution/filltasksintree.php | 68 ++++++++++++++++++++++++
 2 files changed, 84 insertions(+)
 create mode 100644 test/model/execution/filltasksintree.php

diff --git a/test/class/execution.class.php b/test/class/execution.class.php
index f141853ed3..6879534898 100644
--- a/test/class/execution.class.php
+++ b/test/class/execution.class.php
@@ -11,6 +11,7 @@ class executionTest
     {
         global $tester;
         $this->objectModel = $tester->loadModel('execution');
+        $this->treeModel   = $tester->loadModel('tree');
     }
 
     /**
@@ -2669,4 +2670,19 @@ class executionTest
         $tasks = $this->objectModel->getSearchTasks("execution = $executionID", null, 'id_desc');
         return $this->objectModel->formatTasksForTree($tasks);
     }
+
+    /**
+     * Test fill tasks in tree.
+     *
+     * @param  int    $executionID
+     * @access public
+     * @return object
+     */
+    public function fillTasksInTreeTest($executionID)
+    {
+        $fullTrees = $this->treeModel->getTaskStructure($executionID, 0);
+        if(empty($fullTrees)) return '0';
+
+        return $this->objectModel->fillTasksInTree((object)$fullTrees[0], $executionID);
+    }
 }
diff --git a/test/model/execution/filltasksintree.php b/test/model/execution/filltasksintree.php
new file mode 100644
index 0000000000..6309c140dd
--- /dev/null
+++ b/test/model/execution/filltasksintree.php
@@ -0,0 +1,68 @@
+#!/usr/bin/env php
+<?php
+include dirname(dirname(dirname(__FILE__))) . '/lib/init.php';
+include dirname(dirname(dirname(__FILE__))) . '/class/execution.class.php';
+
+$execution = zdTable('project');
+$execution->id->range('1-5');
+$execution->name->range('项目1,项目2,迭代1,迭代2,迭代3');
+$execution->type->range('project{2},sprint,stage,kanban');
+$execution->status->range('doing');
+$execution->parent->range('0,0,1,1,2');
+$execution->project->range('0,0,1,1,2');
+$execution->grade->range('2{2},1{3}');
+$execution->openedVersion->range('18.0');
+$execution->path->range('1,2,`1,3`,`1,4`,`2,5`')->prefix(',')->postfix(',');
+$execution->begin->range('20230102 000000:0')->type('timestamp')->format('YY/MM/DD');
+$execution->end->range('20230212 000000:0')->type('timestamp')->format('YY/MM/DD');
+$execution->gen(5);
+
+$task = zdTable('task');
+$task->id->range('1-10');
+$task->name->range('1-10')->prefix('任务');
+$task->execution->range('3-5');
+$task->type->range('test,devel');
+$task->status->range('wait,doing');
+$task->estimate->range('1-10');
+$task->left->range('1-10');
+$task->consumed->range('1-10');
+$task->gen(10);
+
+$product = zdTable('product');
+$product->id->range('1-3');
+$product->name->range('1-3')->prefix('产品');
+$product->code->range('1-3')->prefix('product');
+$product->type->range('normal');
+$product->status->range('normal');
+$product->gen(3);
+
+$product = zdTable('module');
+$product->id->range('1-10');
+$product->name->range('1-10')->prefix('模块');
+$product->root->range('1-3');
+$product->parent->range('0,1{9}');
+$product->type->range('task');
+$product->gen(10);
+
+su('admin');
+
+/**
+
+title=测试executionModel->fillTasksInTree();
+cid=1
+pid=1
+
+敏捷执行查询 >> 产品1
+瀑布执行查询 >> 产品1
+看板执行查询 >> 产品1
+错误执行查询 >> 0
+
+*/
+
+$executionIDList = array('0', '3', '4', '5');
+
+$execution = new executionTest();
+r($execution->fillTasksInTreeTest($executionIDList[1])) && p('name') && e('产品1'); // 敏捷执行查询
+r($execution->fillTasksInTreeTest($executionIDList[2])) && p('name') && e('产品1'); // 瀑布执行查询
+r($execution->fillTasksInTreeTest($executionIDList[3])) && p('name') && e('产品1'); // 看板执行查询
+r($execution->fillTasksInTreeTest($executionIDList[0])) && p() && e('0');           // 错误执行查询