From 970e11bf0a69367a4f91095e9707e4304fb8c856 Mon Sep 17 00:00:00 2001
From: tianshujie <tianshujie@easycorp.ltd>
Date: Mon, 4 Dec 2023 14:04:14 +0800
Subject: [PATCH] + Add doc::getLibsOptionMenu unit test.

---
 module/doc/test/doc.class.php               | 15 ++++++++++++
 module/doc/test/model/getlibsoptionmenu.php | 27 +++++++++++++++++++++
 2 files changed, 42 insertions(+)
 create mode 100644 module/doc/test/model/getlibsoptionmenu.php

diff --git a/module/doc/test/doc.class.php b/module/doc/test/doc.class.php
index e301bac465..33be174cfe 100644
--- a/module/doc/test/doc.class.php
+++ b/module/doc/test/doc.class.php
@@ -1095,4 +1095,19 @@ class docTest
     {
         return $this->objectModel->getApiLibs($appendLib, $objectType, $objectID);
     }
+
+    /**
+     * 获取带库名称的模块数据。
+     * Get option menu for libs.
+     *
+     * @param  array  $libIdList
+     * @param  string $docType   doc|api
+     * @access public
+     * @return array
+     */
+    public function getLibsOptionMenuTest(array $libIdList, string $docType): array
+    {
+        $libs = $this->objectModel->dao->select('id,name')->from(TABLE_DOCLIB)->where('id')->in($libIdList)->fetchPairs();
+        return $this->objectModel->getLibsOptionMenu($libs, $docType);
+    }
 }
diff --git a/module/doc/test/model/getlibsoptionmenu.php b/module/doc/test/model/getlibsoptionmenu.php
new file mode 100644
index 0000000000..9c278dfbe5
--- /dev/null
+++ b/module/doc/test/model/getlibsoptionmenu.php
@@ -0,0 +1,27 @@
+#!/usr/bin/env php
+<?php
+/**
+
+title=测试 docModel->getLibsOptionMenu();
+timeout=0
+cid=1
+
+*/
+
+include dirname(__FILE__, 5) . '/test/lib/init.php';
+include dirname(__FILE__, 2) . '/doc.class.php';
+
+zdTable('module')->config('module')->gen(10);
+zdTable('doclib')->config('doclib')->gen(30);
+zdTable('user')->gen(5);
+su('admin');
+
+$libs[0]  = array();
+$libs[1]  = array(6, 12, 13, 18);
+$libs[2]  = range(1, 5);
+$docTypes = array('doc', 'api');
+
+$docTester = new docTest();
+r($docTester->getLibsOptionMenuTest($libs[0], $docTypes[0])) && p()      && e('0');              // 测试libs为空
+r($docTester->getLibsOptionMenuTest($libs[1], $docTypes[0])) && p('6_0') && e('自定义文档库6/'); // 获取文档库的目录
+r($docTester->getLibsOptionMenuTest($libs[2], $docTypes[1])) && p('1_0') && e('项目接口库1/');   // 获取接口库的目录