From 5de3fa5cdd93ff2d18f10c75f2a44abaac62eaff Mon Sep 17 00:00:00 2001
From: tianshujie <tianshujie@easycorp.ltd>
Date: Thu, 23 Nov 2023 11:06:59 +0800
Subject: [PATCH] + Add custom::hasScrumAuditplanData unit test.

---
 module/custom/test/custom.class.php           | 20 +++++++++++
 .../test/model/hasscrumauditplandata.php      | 36 +++++++++++++++++++
 2 files changed, 56 insertions(+)
 create mode 100644 module/custom/test/model/hasscrumauditplandata.php

diff --git a/module/custom/test/custom.class.php b/module/custom/test/custom.class.php
index 12eb009ed6..b6a9b64a2e 100644
--- a/module/custom/test/custom.class.php
+++ b/module/custom/test/custom.class.php
@@ -513,4 +513,24 @@ class customTest
         if(dao::isError()) return dao::getError();
         return $count;
     }
+
+    /**
+     * 检查系统中是否有审计数据。
+     * Check if there is auditplan data in the system.
+     *
+     * @param  string    $edition
+     * @access public
+     * @return int|array
+     */
+    public function hasScrumAuditplanDataTest(string $edition): int|array
+    {
+        $oldEdition = $this->objectModel->config->edition;
+
+        $this->objectModel->config->edition = $edition;
+        $count = $this->objectModel->hasScrumAuditplanData();
+
+        $this->objectModel->config->edition = $oldEdition;
+        if(dao::isError()) return dao::getError();
+        return $count;
+    }
 }
diff --git a/module/custom/test/model/hasscrumauditplandata.php b/module/custom/test/model/hasscrumauditplandata.php
new file mode 100644
index 0000000000..8645e7a92e
--- /dev/null
+++ b/module/custom/test/model/hasscrumauditplandata.php
@@ -0,0 +1,36 @@
+#!/usr/bin/env php
+<?php
+/**
+
+title=测试 customModel->hasScrumAuditplanData();
+timeout=0
+cid=1
+
+*/
+
+include dirname(__FILE__, 5) . '/test/lib/init.php';
+include dirname(__FILE__, 2) . '/custom.class.php';
+
+$projectTable = zdTable('project');
+$projectTable->id->range('1-5');
+$projectTable->model->range('scrum');
+$projectTable->gen(5);
+
+zdTable('auditplan')->gen(0);
+zdTable('user')->gen(5);
+su('admin');
+
+$editionList = array('open', 'ipd', 'max');
+
+$customTester = new customTest();
+r($customTester->hasScrumAuditplanDataTest($editionList[0])) && p() && e('0'); // 测试开源版中无审计数据
+r($customTester->hasScrumAuditplanDataTest($editionList[1])) && p() && e('0'); // 测试ipd版中无审计数据
+r($customTester->hasScrumAuditplanDataTest($editionList[2])) && p() && e('0'); // 测试旗舰版中无审计数据
+
+$auditplanTable = zdTable('auditplan');
+$auditplanTable->deleted->range('0');
+$auditplanTable->project->range('1-5');
+$auditplanTable->gen(5);
+r($customTester->hasScrumAuditplanDataTest($editionList[0])) && p() && e('0'); // 测试开源版中有审计数据
+r($customTester->hasScrumAuditplanDataTest($editionList[1])) && p() && e('5'); // 测试ipd版中有审计数据
+r($customTester->hasScrumAuditplanDataTest($editionList[2])) && p() && e('5'); // 测试旗舰版中有审计数据