From 2ee7594858dd8786d3933704bd009d7f6e16da72 Mon Sep 17 00:00:00 2001
From: zhujinyong <vimrus.com@gmail.com>
Date: Tue, 23 Mar 2021 08:45:02 +0800
Subject: [PATCH] * Fix menu.

---
 module/product/control.php   | 2 ++
 module/testsuite/control.php | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/module/product/control.php b/module/product/control.php
index 865e7be3b5..258b92f10c 100644
--- a/module/product/control.php
+++ b/module/product/control.php
@@ -322,6 +322,8 @@ class product extends control
             $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'locate' => $locate));
         }
 
+        if($this->app->openApp == 'program') $this->loadModel('program')->setMenu($programID);
+
         $this->loadModel('user');
         $poUsers = $this->user->getPairs('nodeleted|pofirst|noclosed',  '', $this->config->maxCount);
         if(!empty($this->config->user->moreLink)) $this->config->moreLinks["PO"] = $this->config->user->moreLink;
diff --git a/module/testsuite/control.php b/module/testsuite/control.php
index 9e889b875d..1f41073fe0 100644
--- a/module/testsuite/control.php
+++ b/module/testsuite/control.php
@@ -169,7 +169,7 @@ class testsuite extends control
         $productID = $suite->product;
 
         $this->view->products = $this->products = $this->loadModel('product')->getProductPairsByProject($this->session->project);
-        $this->testsuite->setMenu($this->products, $productID);
+        $this->loadModel('qa')->setMenu($this->products, $productID);
 
         /* Save session. */
         $this->session->set('caseList', $this->app->getURI(true));