diff --git a/test/class/program.class.php b/test/class/program.class.php
index eea5e51baf..f223f5bce5 100644
--- a/test/class/program.class.php
+++ b/test/class/program.class.php
@@ -199,4 +199,16 @@ class programTest
     {
         return $this->program->getInvolvedPrograms($account);
     }
+
+    /*
+     * Get Tree menu.
+     *
+     * @param  string $programID
+     * @access public
+     * @return array
+     */
+    public function getTreeMenuTest($programID)
+    {
+        return $this->program->getTreeMenu($programID);
+    }
 }
diff --git a/test/model/program/gettreemenu.php b/test/model/program/gettreemenu.php
index f73d77e4e4..e9a4504fc4 100755
--- a/test/model/program/gettreemenu.php
+++ b/test/model/program/gettreemenu.php
@@ -1,6 +1,19 @@
 #!/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-5');
+$program->name->range('项目集1,项目集2');
+$program->type->range('program');
+$program->status->range('wait');
+$program->parent->range('0,0,1,1,2');
+$program->path->range('1,2,`1,3`,`1,4`,`2,5`')->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(5);
 
 /**
 
@@ -8,12 +21,10 @@ title=测试 programModel::getTreeMenu();
 cid=1
 pid=1
 
-查看返回的字符个数 >> 789
+查看返回的字符个数 >> 396
 
 */
+$programTester = new programTest();
+$programs1 = $programTester->getTreeMenuTest(1);
 
-global $tester;
-$tester->loadModel('program');
-$programs1 = $tester->program->getTreeMenu(1);
-
-r(strlen($programs1)) && p() && e('789'); // 查看返回的字符个数
\ No newline at end of file
+r(strlen($programs1)) && p() && e('396'); // 查看返回的字符个数