From 682d97e9f52414f2aabd0d757603513f739eb008 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 11:20:15 +0800
Subject: [PATCH] * Add test of program model.

---
 test/class/program.class.php                | 11 +++++-----
 test/model/program/createdefaultprogram.php | 11 ++++------
 test/model/program/createstakeholder.php    | 23 +++++++++++++--------
 3 files changed, 24 insertions(+), 21 deletions(-)

diff --git a/test/class/program.class.php b/test/class/program.class.php
index dc67fe51eb..b581982738 100644
--- a/test/class/program.class.php
+++ b/test/class/program.class.php
@@ -36,15 +36,16 @@ class programTest
     }
 
     /**
-     * createStakeholder
+     * Create stakeholder.
      *
-     * @param  int    mixed $programID
+     * @param  int    $programID
+     * @param  array  $accounts
      * @access public
      * @return void
      */
-    public function createStakeholder($programID)
+    public function createStakeholderTest($programID, $accounts = array())
     {
-        $_POST['accounts'] = array('dev1', 'dev2');
+        $_POST['accounts'] = $accounts;
         $stakeHolder = $this->program->createStakeholder($programID);
 
         return $this->program->getStakeholdersByPrograms($programID);
@@ -60,7 +61,7 @@ class programTest
     {
         $programID = $this->program->createDefaultProgram();
 
-        return $programID;
+        return $programID > 0;
     }
 
     /**
diff --git a/test/model/program/createdefaultprogram.php b/test/model/program/createdefaultprogram.php
index 136076867e..14270abcf7 100644
--- a/test/model/program/createdefaultprogram.php
+++ b/test/model/program/createdefaultprogram.php
@@ -2,7 +2,6 @@
 <?php
 include dirname(dirname(dirname(__FILE__))) . '/lib/init.php';
 include dirname(dirname(dirname(__FILE__))) . '/class/program.class.php';
-$db->switchDB();
 
 /**
 
@@ -10,14 +9,12 @@ title=测试 programModel::createDefaultProgram();
 cid=1
 pid=1
 
-创建默认项目集并返回ID >> 751
+创建默认项目集 >> 1
 
 */
 
-$test = new programTest();
+$programTester = new programTest();
 
-$result = $test->createDefaultProgramTest();
+$result = $programTester->createDefaultProgramTest();
 
-r($result) && p('') && e('751'); // 创建默认项目集并返回ID
-
-$db->restoreDB();
\ No newline at end of file
+r($result) && p('') && e('1'); // 创建默认项目集
diff --git a/test/model/program/createstakeholder.php b/test/model/program/createstakeholder.php
index 26d773a962..115e695ee9 100755
--- a/test/model/program/createstakeholder.php
+++ b/test/model/program/createstakeholder.php
@@ -2,7 +2,15 @@
 <?php
 include dirname(dirname(dirname(__FILE__))) . '/lib/init.php';
 include dirname(dirname(dirname(__FILE__))) . '/class/program.class.php';
-$db->switchDB();
+
+$program = zdTable('project');
+$program->id->range('1');
+$program->name->range('父项目集1');
+$program->type->range('program');
+$program->path->range('1')->prefix(',')->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(1);
 
 /**
 
@@ -11,17 +19,14 @@ cid=1
 pid=1
 
 创建id=1的项目集的干系人并查看数量。 >> 2
-创建id=1的项目集的干系人dev1,dev2并查看Account。 >> dev2;dev1
+创建id=1的项目集的干系人dev1,dev2并查看Account。 >> dev1;dev2
 
 */
 
-global $tester;
-$tester->loadModel('program');
+$programTester = new programTest();
 
-$_POST['accounts'] = array('dev1', 'dev2');
-$tester->program->createStakeholder(1);
-$result = $tester->program->getStakeholdersByPrograms(1);
+$accounts = array('dev1', 'dev2');
+$result   = $programTester->createStakeholderTest(1, $accounts);
 
 r(count($result)) && p('')                    && e('2');         // 创建id=1的项目集的干系人并查看数量。
-r($result)        && p('0:account;1:account') && e('dev2;dev1'); // 创建id=1的项目集的干系人dev1,dev2并查看Account。
-$db->restoreDB();
\ No newline at end of file
+r($result)        && p('0:account;1:account') && e('dev1;dev2'); // 创建id=1的项目集的干系人dev1,dev2并查看Account。