From f8226baaa19f477fd7d3101fe2f7e0abe33742cb Mon Sep 17 00:00:00 2001
From: tianshujie <tianshujie@easycorp.ltd>
Date: Sat, 25 Nov 2023 10:54:20 +0800
Subject: [PATCH] + Add custom::getMainMenu unit test.

---
 module/custom/test/custom.class.php      | 13 +++++++++++++
 module/custom/test/model/getmainmenu.php | 21 +++++++++++++++++++++
 2 files changed, 34 insertions(+)
 create mode 100644 module/custom/test/model/getmainmenu.php

diff --git a/module/custom/test/custom.class.php b/module/custom/test/custom.class.php
index 4d82a25c5f..ddda209281 100644
--- a/module/custom/test/custom.class.php
+++ b/module/custom/test/custom.class.php
@@ -729,4 +729,17 @@ class customTest
     {
         return customModel::getModuleMenu($module);
     }
+
+    /**
+     * 获取主菜单数据。
+     * Get main menu data.
+     *
+     * @static
+     * @access public
+     * @return array
+     */
+    public static function getMainMenuTest(): array
+    {
+        return customModel::getMainMenu();
+    }
 }
diff --git a/module/custom/test/model/getmainmenu.php b/module/custom/test/model/getmainmenu.php
new file mode 100644
index 0000000000..0236acb6b8
--- /dev/null
+++ b/module/custom/test/model/getmainmenu.php
@@ -0,0 +1,21 @@
+#!/usr/bin/env php
+<?php
+/**
+
+title=测试 customModel->getMainMenu();
+timeout=0
+cid=1
+
+*/
+
+include dirname(__FILE__, 5) . '/test/lib/init.php';
+include dirname(__FILE__, 2) . '/custom.class.php';
+
+zdTable('lang')->config('lang')->gen(10);
+zdTable('user')->gen(5);
+su('admin');
+
+$customTester = new customTest();
+r($customTester->getMainMenuTest()) && p() && e('0'); // 获取主菜单
+
+zdTable('lang')->gen(0);