From fa7d6fcff2f06c78f3df86679d12374b7974b534 Mon Sep 17 00:00:00 2001 From: lids <1713278948@qq.com> Date: Wed, 29 Apr 2026 15:21:22 +0800 Subject: [PATCH] =?UTF-8?q?AI=20=E8=87=AA=E5=8A=A8=E7=94=9F=E6=88=90?= =?UTF-8?q?=E6=B5=8B=E8=AF=95=E7=94=A8=E4=BE=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/CMakeLists.txt | 8 +- tests/test_alert_manager.cpp | 345 ++++++++++++----------------------- 2 files changed, 121 insertions(+), 232 deletions(-) diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt index 59052db..d247ec0 100644 --- a/tests/CMakeLists.txt +++ b/tests/CMakeLists.txt @@ -1,5 +1,5 @@ cmake_minimum_required(VERSION 3.10.0) -project(CppGenerate_test) +project(test_CppGenerate) include(FetchContent) if (MSVC) add_compile_options(/utf-8) @@ -18,8 +18,8 @@ include_directories(${CMAKE_CURRENT_SOURCE_DIR}/../include) include(CTest) enable_testing() -add_executable(CppGenerate_test test_alert_manager.cpp ../src/alert_manager.cpp) +add_executable(test_CppGenerate test_alert_manager.cpp ../src/alert_manager.cpp) -target_link_libraries(CppGenerate_test gtest gmock gtest_main) +target_link_libraries(test_CppGenerate gtest gmock gtest_main) include(GoogleTest) -gtest_discover_tests(CppGenerate_test) \ No newline at end of file +gtest_discover_tests(test_CppGenerate) \ No newline at end of file diff --git a/tests/test_alert_manager.cpp b/tests/test_alert_manager.cpp index 5bc247a..f19a179 100644 --- a/tests/test_alert_manager.cpp +++ b/tests/test_alert_manager.cpp @@ -1,267 +1,156 @@ #include "gtest/gtest.h" #include "alert_manager.hpp" +#include #include -#include -// Mock for time to ensure deterministic tests -static uint32_t mock_timestamp = 1640995200; // 2022-01-01 00:00:00 UTC - -class MockTime { +// Mock class to capture output for testing +class MockAlertManager : public AlertManager { public: - static uint32_t getTimestamp() { return mock_timestamp; } - static void setTimestamp(uint32_t t) { mock_timestamp = t; } + std::vector captured_output; + + void print(const std::string& msg) override { + captured_output.push_back(msg); + } + + // Override to prevent actual console output + void triggerAlert(AlertType type, uint32_t timestamp, const std::string& description) override { + AlertInfo alert; + alert.type = type; + alert.status = AlertStatus::ACTIVE; + alert.trigger_time = timestamp; + alert.description = description; + alerts_[type] = alert; + addToHistory(alert); + } + + void updateAlertStatus(AlertType type, AlertStatus new_status, uint32_t timestamp) override { + auto it = alerts_.find(type); + if (it == alerts_.end()) return; + AlertInfo& alert = it->second; + alert.status = new_status; + alert.acknowledge_time = (new_status == AlertStatus::ACKNOWLEDGED) ? timestamp : alert.acknowledge_time; + alert.resolve_time = (new_status == AlertStatus::RESOLVED) ? timestamp : alert.resolve_time; + } }; -// Helper function to compare AlertInfo -bool operator==(const AlertInfo& a, const AlertInfo& b) { - return a.type == b.type && - a.status == b.status && - a.trigger_time == b.trigger_time && - a.acknowledge_time == b.acknowledge_time && - a.resolve_time == b.resolve_time && - a.description == b.description; -} - -// Test fixture +// Test fixture for AlertManager class AlertManagerTest : public ::testing::Test { protected: - AlertManager manager; + std::unique_ptr alert_manager_; void SetUp() override { - mock_timestamp = 1640995200; - manager.initialize(); + alert_manager_ = std::make_unique(); } void TearDown() override { - manager.clearHistory(); + alert_manager_->captured_output.clear(); } }; -TEST_F(AlertManagerTest, testAlertManagerDefaultInitialization) { - EXPECT_FLOAT_EQ(manager.getAltitudeThresholds().first, 100.0f); - EXPECT_FLOAT_EQ(manager.getAltitudeThresholds().second, -50.0f); - EXPECT_TRUE(manager.getActiveAlerts().empty()); - EXPECT_TRUE(manager.getAlertHistory(10).empty()); +// Test case: 正常输入测试 - 检查高度上限预警触发 +TEST_F(AlertManagerTest, testCheckAltitudeAlertNormalUpperTrigger) { + alert_manager_->setAltitudeThresholds(100.0f, -50.0f); + alert_manager_->checkAltitudeAlert(105.0f, 1234567890); + + EXPECT_TRUE(alert_manager_->hasActiveAlerts()); + EXPECT_EQ(alert_manager_->getActiveAlerts().size(), 1); + EXPECT_EQ(alert_manager_->getActiveAlerts()[0].type, AlertType::ALTITUDE_UPPER); + EXPECT_EQ(alert_manager_->getActiveAlerts()[0].status, AlertStatus::ACTIVE); } -TEST_F(AlertManagerTest, testInitialize) { - // No direct output assertion, but verify initialization logic - // This is tested implicitly via constructor - EXPECT_NO_FATAL_FAILURE(manager.initialize()); +// Test case: 正常输入测试 - 检查高度下限预警触发 +TEST_F(AlertManagerTest, testCheckAltitudeAlertNormalLowerTrigger) { + alert_manager_->setAltitudeThresholds(100.0f, -50.0f); + alert_manager_->checkAltitudeAlert(-55.0f, 1234567890); + + EXPECT_TRUE(alert_manager_->hasActiveAlerts()); + EXPECT_EQ(alert_manager_->getActiveAlerts().size(), 1); + EXPECT_EQ(alert_manager_->getActiveAlerts()[0].type, AlertType::ALTITUDE_LOWER); + EXPECT_EQ(alert_manager_->getActiveAlerts()[0].status, AlertStatus::ACTIVE); } -TEST_F(AlertManagerTest, testCheckAltitudeAlertNormal) { - mock_timestamp = 1640995200; - manager.checkAltitudeAlert(105.0f, mock_timestamp); - auto alerts = manager.getActiveAlerts(); - EXPECT_EQ(alerts.size(), 1); - EXPECT_EQ(alerts[0].type, AlertType::ALTITUDE_UPPER); - EXPECT_EQ(alerts[0].status, AlertStatus::ACTIVE); - EXPECT_EQ(alerts[0].trigger_time, mock_timestamp); +// Test case: 边界值测试 - 高度等于上限阈值，不应触发预警 +TEST_F(AlertManagerTest, testCheckAltitudeAlertBoundaryUpperEqual) { + alert_manager_->setAltitudeThresholds(100.0f, -50.0f); + alert_manager_->checkAltitudeAlert(100.0f, 1234567890); + + EXPECT_FALSE(alert_manager_->hasActiveAlerts()); } -TEST_F(AlertManagerTest, testCheckAltitudeAlertBelowLower) { - mock_timestamp = 1640995200; - manager.checkAltitudeAlert(-55.0f, mock_timestamp); - auto alerts = manager.getActiveAlerts(); - EXPECT_EQ(alerts.size(), 1); - EXPECT_EQ(alerts[0].type, AlertType::ALTITUDE_LOWER); - EXPECT_EQ(alerts[0].status, AlertStatus::ACTIVE); +// Test case: 边界值测试 - 高度等于下限阈值，不应触发预警 +TEST_F(AlertManagerTest, testCheckAltitudeAlertBoundaryLowerEqual) { + alert_manager_->setAltitudeThresholds(100.0f, -50.0f); + alert_manager_->checkAltitudeAlert(-50.0f, 1234567890); + + EXPECT_FALSE(alert_manager_->hasActiveAlerts()); } -TEST_F(AlertManagerTest, testCheckAltitudeAlertResetWhenNormal) { - mock_timestamp = 1640995200; - manager.checkAltitudeAlert(105.0f, mock_timestamp); // Trigger upper - mock_timestamp = 1640995201; - manager.checkAltitudeAlert(90.0f, mock_timestamp); // Reset - auto alerts = manager.getActiveAlerts(); - EXPECT_TRUE(alerts.empty()); +// Test case: 异常输入测试 - 上限阈值小于下限阈值，应输出错误提示 +TEST_F(AlertManagerTest, testSetAltitudeThresholdsInvalidInput) { + alert_manager_->setAltitudeThresholds(-10.0f, 10.0f); + + EXPECT_EQ(alert_manager_->upper_threshold_, 100.0f); // Should remain unchanged + EXPECT_EQ(alert_manager_->lower_threshold_, -50.0f); + EXPECT_TRUE(alert_manager_->captured_output.empty()); // No error message printed } -TEST_F(AlertManagerTest, testSetAltitudeThresholdsValid) { - manager.setAltitudeThresholds(200.0f, -100.0f); - auto thresholds = manager.getAltitudeThresholds(); - EXPECT_FLOAT_EQ(thresholds.first, 200.0f); - EXPECT_FLOAT_EQ(thresholds.second, -100.0f); +// Test case: 特殊场景测试 - 高度从超限恢复到正常范围，预警应被解决 +TEST_F(AlertManagerTest, testCheckAltitudeAlertRecoveryFromUpper) { + alert_manager_->setAltitudeThresholds(100.0f, -50.0f); + alert_manager_->checkAltitudeAlert(105.0f, 1234567890); // Trigger upper alert + EXPECT_TRUE(alert_manager_->hasActiveAlerts()); + + alert_manager_->checkAltitudeAlert(95.0f, 1234567891); // Recovery + EXPECT_FALSE(alert_manager_->hasActiveAlerts()); + EXPECT_EQ(alert_manager_->getActiveAlerts().size(), 0); } -TEST_F(AlertManagerTest, testSetAltitudeThresholdsInvalid) { - EXPECT_NO_FATAL_FAILURE(manager.setAltitudeThresholds(50.0f, 100.0f)); - // Should print error, but no crash - EXPECT_FLOAT_EQ(manager.getAltitudeThresholds().first, 100.0f); - EXPECT_FLOAT_EQ(manager.getAltitudeThresholds().second, -50.0f); +// Test case: 特殊场景测试 - 高度从超限恢复到正常范围，下限预警应被解决 +TEST_F(AlertManagerTest, testCheckAltitudeAlertRecoveryFromLower) { + alert_manager_->setAltitudeThresholds(100.0f, -50.0f); + alert_manager_->checkAltitudeAlert(-55.0f, 1234567890); // Trigger lower alert + EXPECT_TRUE(alert_manager_->hasActiveAlerts()); + + alert_manager_->checkAltitudeAlert(-45.0f, 1234567891); // Recovery + EXPECT_FALSE(alert_manager_->hasActiveAlerts()); + EXPECT_EQ(alert_manager_->getActiveAlerts().size(), 0); } -TEST_F(AlertManagerTest, testTriggerSensorFailure) { - mock_timestamp = 1640995200; - manager.triggerSensorFailure(mock_timestamp); - auto alerts = manager.getActiveAlerts(); - EXPECT_EQ(alerts.size(), 1); - EXPECT_EQ(alerts[0].type, AlertType::SENSOR_FAILURE); - EXPECT_EQ(alerts[0].status, AlertStatus::ACTIVE); - EXPECT_EQ(alerts[0].trigger_time, mock_timestamp); +// Test case: 正常输入测试 - 设置合法阈值，检查是否更新成功 +TEST_F(AlertManagerTest, testSetAltitudeThresholdsValidInput) { + alert_manager_->setAltitudeThresholds(200.0f, -100.0f); + + EXPECT_EQ(alert_manager_->upper_threshold_, 200.0f); + EXPECT_EQ(alert_manager_->lower_threshold_, -100.0f); + EXPECT_TRUE(alert_manager_->hasActiveAlerts()); } -TEST_F(AlertManagerTest, testTriggerLowBattery) { - mock_timestamp = 1640995200; - manager.triggerLowBattery(mock_timestamp, 15.0f); - auto alerts = manager.getActiveAlerts(); - EXPECT_EQ(alerts.size(), 1); - EXPECT_EQ(alerts[0].type, AlertType::LOW_BATTERY); - EXPECT_EQ(alerts[0].status, AlertStatus::ACTIVE); - EXPECT_NE(alerts[0].description.find("15.0%"), std::string::npos); +// Test case: 特殊场景测试 - 多次调用 checkAltitudeAlert，验证状态一致性 +TEST_F(AlertManagerTest, testCheckAltitudeAlertMultipleCalls) { + alert_manager_->setAltitudeThresholds(100.0f, -50.0f); + + alert_manager_->checkAltitudeAlert(105.0f, 1234567890); // Trigger + EXPECT_TRUE(alert_manager_->hasActiveAlerts()); + + alert_manager_->checkAltitudeAlert(105.0f, 1234567891); // Still above + EXPECT_TRUE(alert_manager_->hasActiveAlerts()); + + alert_manager_->checkAltitudeAlert(95.0f, 1234567892); // Recover + EXPECT_FALSE(alert_manager_->hasActiveAlerts()); } -TEST_F(AlertManagerTest, testTriggerCommunicationError) { - mock_timestamp = 1640995200; - manager.triggerCommunicationError(mock_timestamp); - auto alerts = manager.getActiveAlerts(); - EXPECT_EQ(alerts.size(), 1); - EXPECT_EQ(alerts[0].type, AlertType::COMMUNICATION_ERROR); - EXPECT_EQ(alerts[0].status, AlertStatus::ACTIVE); +// Test case: 边界值测试 - 阈值为极小值和极大值 +TEST_F(AlertManagerTest, testSetAltitudeThresholdsExtremeValues) { + alert_manager_->setAltitudeThresholds(FLT_MAX, FLT_MIN); + + EXPECT_EQ(alert_manager_->upper_threshold_, FLT_MAX); + EXPECT_EQ(alert_manager_->lower_threshold_, FLT_MIN); } -TEST_F(AlertManagerTest, testAcknowledgeAllAlerts) { - mock_timestamp = 1640995200; - manager.triggerSensorFailure(mock_timestamp); - manager.triggerLowBattery(mock_timestamp + 1, 10.0f); - mock_timestamp = 1640995202; - manager.acknowledgeAllAlerts(mock_timestamp); - auto alerts = manager.getActiveAlerts(); - EXPECT_TRUE(alerts.empty()); - EXPECT_EQ(alerts.size(), 0); -} +// Test case: 异常输入测试 - 空值输入（使用零值） +TEST_F(AlertManagerTest, testSetAltitudeThresholdsZeroValues) { + alert_manager_->setAltitudeThresholds(0.0f, 0.0f); -TEST_F(AlertManagerTest, testAcknowledgeAlert) { - mock_timestamp = 1640995200; - manager.triggerSensorFailure(mock_timestamp); - mock_timestamp = 1640995201; - manager.acknowledgeAlert(AlertType::SENSOR_FAILURE, mock_timestamp); - auto alerts = manager.getActiveAlerts(); - EXPECT_TRUE(alerts.empty()); + EXPECT_EQ(alert_manager_->upper_threshold_, 100.0f); + EXPECT_EQ(alert_manager_->lower_threshold_, -50.0f); } - -TEST_F(AlertManagerTest, testResetAlert) { - mock_timestamp = 1640995200; - manager.triggerSensorFailure(mock_timestamp); - manager.resetAlert(AlertType::SENSOR_FAILURE); - auto alerts = manager.getActiveAlerts(); - EXPECT_TRUE(alerts.empty()); -} - -TEST_F(AlertManagerTest, testResetAllAlerts) { - mock_timestamp = 1640995200; - manager.triggerSensorFailure(mock_timestamp); - manager.triggerLowBattery(mock_timestamp + 1, 10.0f); - manager.resetAllAlerts(); - auto alerts = manager.getActiveAlerts(); - EXPECT_TRUE(alerts.empty()); -} - -TEST_F(AlertManagerTest, testHasActiveAlerts) { - EXPECT_FALSE(manager.hasActiveAlerts()); - mock_timestamp = 1640995200; - manager.triggerSensorFailure(mock_timestamp); - EXPECT_TRUE(manager.hasActiveAlerts()); -} - -TEST_F(AlertManagerTest, testHasUnacknowledgedAlerts) { - EXPECT_FALSE(manager.hasUnacknowledgedAlerts()); - mock_timestamp = 1640995200; - manager.triggerSensorFailure(mock_timestamp); - EXPECT_TRUE(manager.hasUnacknowledgedAlerts()); - manager.acknowledgeAllAlerts(mock_timestamp + 1); - EXPECT_FALSE(manager.hasUnacknowledgedAlerts()); -} - -TEST_F(AlertManagerTest, testGetActiveAlerts) { - mock_timestamp = 1640995200; - manager.triggerSensorFailure(mock_timestamp); - auto alerts = manager.getActiveAlerts(); - EXPECT_EQ(alerts.size(), 1); - EXPECT_EQ(alerts[0].type, AlertType::SENSOR_FAILURE); -} - -TEST_F(AlertManagerTest, testGetAlertHistoryMaxCount) { - mock_timestamp = 1640995200; - for (int i = 0; i < 15; ++i) { - manager.triggerSensorFailure(mock_timestamp + i); - } - auto history = manager.getAlertHistory(10); - EXPECT_EQ(history.size(), 10); -} - -TEST_F(AlertManagerTest, testGetStatusDescriptionNoAlerts) { - std::string desc = manager.getStatusDescription(); - EXPECT_EQ(desc, "系统状态：正常"); -} - -TEST_F(AlertManagerTest, testGetStatusDescriptionWithAlerts) { - mock_timestamp = 1640995200; - manager.triggerSensorFailure(mock_timestamp); - std::string desc = manager.getStatusDescription(); - EXPECT_NE(desc.find("有1个活动预警"), std::string::npos); - EXPECT_NE(desc.find("传感器故障"), std::string::npos); -} - -TEST_F(AlertManagerTest, testClearHistory) { - mock_timestamp = 1640995200; - manager.triggerSensorFailure(mock_timestamp); - manager.clearHistory(); - EXPECT_TRUE(manager.getAlertHistory(10).empty()); -} - -TEST_F(AlertManagerTest, testAddToHistoryLimit) { - mock_timestamp = 1640995200; - for (int i = 0; i < 100; ++i) { - manager.triggerSensorFailure(mock_timestamp + i); - } - auto history = manager.getAlertHistory(10); - EXPECT_EQ(history.size(), 10); -} - -TEST_F(AlertManagerTest, testTriggerAlert) { - mock_timestamp = 1640995200; - manager.triggerAlert(AlertType::SENSOR_FAILURE, mock_timestamp, "Test alert"); - auto alerts = manager.getActiveAlerts(); - EXPECT_EQ(alerts.size(), 1); - EXPECT_EQ(alerts[0].description, "Test alert"); -} - -TEST_F(AlertManagerTest, testUpdateAlertStatusAcknowledged) { - mock_timestamp = 1640995200; - manager.triggerSensorFailure(mock_timestamp); - mock_timestamp = 1640995201; - manager.updateAlertStatus(AlertType::SENSOR_FAILURE, AlertStatus::ACKNOWLEDGED, mock_timestamp); - auto alerts = manager.getActiveAlerts(); - EXPECT_TRUE(alerts.empty()); -} - -TEST_F(AlertManagerTest, testUpdateAlertStatusResolved) { - mock_timestamp = 1640995200; - manager.triggerSensorFailure(mock_timestamp); - mock_timestamp = 1640995201; - manager.updateAlertStatus(AlertType::SENSOR_FAILURE, AlertStatus::RESOLVED, mock_timestamp); - auto alerts = manager.getActiveAlerts(); - EXPECT_TRUE(alerts.empty()); -} - -TEST_F(AlertManagerTest, testGetAlertTypeName) { - EXPECT_EQ(manager.getAlertTypeName(AlertType::ALTITUDE_UPPER), "高度上限预警"); - EXPECT_EQ(manager.getAlertTypeName(AlertType::NONE), "无"); - EXPECT_EQ(manager.getAlertTypeName(static_cast(99)), "未知预警"); -} - -TEST_F(AlertManagerTest, testGetAlertStatusName) { - EXPECT_EQ(manager.getAlertStatusName(AlertStatus::ACTIVE), "预警中"); - EXPECT_EQ(manager.getAlertStatusName(AlertStatus::INACTIVE), "未激活"); - EXPECT_EQ(manager.getAlertStatusName(static_cast(99)), "未知状态"); -} - -TEST_F(AlertManagerTest, testDestructorDoesNotCrash) { - // Verify destructor doesn't throw or crash - EXPECT_NO_FATAL_FAILURE({ - AlertManager* mgr = new AlertManager(); - delete mgr; - }); -} \ No newline at end of file