From 71c7124fd7633a0e76004405809596e864229b4f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=94=B0=E6=B7=91=E6=9D=B0?= <tianshujie0123@163.com>
Date: Thu, 5 Jan 2023 13:59:55 +0800
Subject: [PATCH] * Add case for program model.

---
 test/class/program.class.php             | 12 ++++++++++++
 test/model/program/getbudgetunitlist.php | 19 ------------------
 test/model/program/getchildren.php       | 25 ++++++++++++++++--------
 3 files changed, 29 insertions(+), 27 deletions(-)
 delete mode 100755 test/model/program/getbudgetunitlist.php

diff --git a/test/class/program.class.php b/test/class/program.class.php
index 4551f246e7..36434cf37d 100644
--- a/test/class/program.class.php
+++ b/test/class/program.class.php
@@ -134,4 +134,16 @@ class programTest
 
         return $budget;
     }
+
+    /**
+     * Get children.
+     *
+     * @param  int    $programID
+     * @access public
+     * @return int
+     */
+    public function getChildrenTest($programID)
+    {
+        return $this->program->getChildren($programID);
+    }
 }
diff --git a/test/model/program/getbudgetunitlist.php b/test/model/program/getbudgetunitlist.php
deleted file mode 100755
index 367aa5677d..0000000000
--- a/test/model/program/getbudgetunitlist.php
+++ /dev/null
@@ -1,19 +0,0 @@
-#!/usr/bin/env php
-<?php
-include dirname(dirname(dirname(__FILE__))) . '/lib/init.php';
-su('admin');
-
-/**
-
-title=测试 programModel::getBudgetUnitList();
-cid=1
-pid=1
-
-获取货币类型列表 >> 人民币;美元
-
-*/
-
-global $tester;
-$tester->loadModel('project');
-
-r($tester->loadModel('program')->getBudgetUnitList()) && p('CNY;USD') && e('人民币;美元'); //获取货币类型列表
\ No newline at end of file
diff --git a/test/model/program/getchildren.php b/test/model/program/getchildren.php
index 61f6bb3ef8..d22267f467 100755
--- a/test/model/program/getchildren.php
+++ b/test/model/program/getchildren.php
@@ -1,23 +1,32 @@
 #!/usr/bin/env php
 <?php
 include dirname(dirname(dirname(__FILE__))) . '/lib/init.php';
+include dirname(dirname(dirname(__FILE__))) . '/class/program.class.php';
 su('admin');
 
+$program = zdTable('project');
+$program->id->range('1-3');
+$program->name->range('父项目集1,子项目集1,子项目集2');
+$program->parent->range('0,1,1');
+$program->type->range('program');
+$program->path->setFields(array(
+    array('field' => 'path1', 'range' => '1', 'prefix' => ',', 'postfix' => ','),
+    array('field' => 'path2', 'range' => '[],2,3', 'postfix' => ',')
+));
+$program->begin->range('20220112 000000:0')->type('timestamp')->format('YY/MM/DD');
+$program->end->range('20220212 000000:0')->type('timestamp')->format('YY/MM/DD');
+$program->gen(3);
+
 /**
 
 title=测试 programModel:: getChildren();
 cid=1
 pid=1
 
-通过id查找id=1的子项目集个数 >> 9
-通过id查找id=22的子项目集个数 >> 7
-通过id查找id=221的子项目集个数 >> 0
+通过id查找id=1的子项目集个数 >> 2
 
 */
 
-global $tester;
-$tester->loadModel('program');
+$programTester = new programTest();
 
-r($tester->program->getChildren(1))   && p() && e('9'); // 通过id查找id=1的子项目集个数
-r($tester->program->getChildren(22))  && p() && e('7'); // 通过id查找id=22的子项目集个数
-r($tester->program->getChildren(221)) && p() && e('0'); // 通过id查找id=221的子项目集个数
\ No newline at end of file
+r($programTester->getChildrenTest(1)) && p() && e('2'); // 通过id查找id=1的子项目集个数