From 60fd9bcfa45e61213be66584a0cf5661c95165c1 Mon Sep 17 00:00:00 2001
From: caoyanyi <caoyanyi@easycorp.ltd>
Date: Fri, 6 Jan 2023 10:21:53 +0800
Subject: [PATCH] * Add test case.

---
 test/class/program.class.php            | 15 +++++++++++
 test/model/program/buildoperatemenu.php | 33 +++++++++++++++++++++++++
 2 files changed, 48 insertions(+)
 create mode 100644 test/model/program/buildoperatemenu.php

diff --git a/test/class/program.class.php b/test/class/program.class.php
index 3078f1ed6a..08a42f30e5 100644
--- a/test/class/program.class.php
+++ b/test/class/program.class.php
@@ -387,4 +387,19 @@ class programTest
     {
         return $this->program->saveState($programID, $programs);
     }
+
+    /**
+     * Test build operate menu.
+     *
+     * @param  int    $programID
+     * @access public
+     * @return void
+     */
+    public function buildOperateMenuTest($programID = 0)
+    {
+        $program = $this->program->getByID($programID);
+        if(empty($program)) return '0';
+
+        return $this->program->buildOperateMenu($program);
+    }
 }
diff --git a/test/model/program/buildoperatemenu.php b/test/model/program/buildoperatemenu.php
new file mode 100644
index 0000000000..a7fcc437cb
--- /dev/null
+++ b/test/model/program/buildoperatemenu.php
@@ -0,0 +1,33 @@
+#!/usr/bin/env php
+<?php
+include dirname(dirname(dirname(__FILE__))) . '/lib/init.php';
+include dirname(dirname(dirname(__FILE__))) . '/class/program.class.php';
+
+$program = zdTable('project');
+$program->id->range('1,2');
+$program->name->range('父项目集1,父项目集2');
+$program->type->range('program');
+$program->budget->range('900000,899900');
+$program->path->range('1,2')->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(2);
+
+su('admin');
+/**
+
+title=测试 programModel::saveState();
+cid=1
+pid=1
+
+项目集不存在 >> 0
+项目集存在   >> div
+
+*/
+
+$programTester = new programTest();
+$menus1        = $programTester->buildOperateMenuTest(0);
+$menus2        = $programTester->buildOperateMenuTest(1);
+
+r($menus1) && p() && e('0');                 // 项目集不存在
+r(substr($menus2, 1, 3)) && p() && e('div'); // 项目集存在