From 1d9a607a7681b1db1000aab4276a768e679c5ab9 Mon Sep 17 00:00:00 2001 From: lids <1713278948@qq.com> Date: Wed, 22 Apr 2026 17:32:55 +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/test_alert_manager.cpp | 360 ++++++++++++++++------------------- 1 file changed, 167 insertions(+), 193 deletions(-) diff --git a/tests/test_alert_manager.cpp b/tests/test_alert_manager.cpp index 5bc247a..d4ed53d 100644 --- a/tests/test_alert_manager.cpp +++ b/tests/test_alert_manager.cpp @@ -3,265 +3,239 @@ #include #include -// Mock for time to ensure deterministic tests -static uint32_t mock_timestamp = 1640995200; // 2022-01-01 00:00:00 UTC +// Mock time function for deterministic testing +uint32_t mock_timestamp = 1640995200; // 2022-01-01 00:00:00 UTC -class MockTime { -public: - static uint32_t getTimestamp() { return mock_timestamp; } - static void setTimestamp(uint32_t t) { mock_timestamp = t; } -}; - -// 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; +// Mock function to replace std::time +uint32_t get_mock_timestamp() { + return mock_timestamp; } -// Test fixture +// Override the global function in alert_manager.cpp +extern "C" uint32_t std_time() { + return get_mock_timestamp(); +} + +// Test fixture for AlertManager class AlertManagerTest : public ::testing::Test { protected: - AlertManager manager; + AlertManager alert_manager; void SetUp() override { mock_timestamp = 1640995200; - manager.initialize(); + alert_manager.initialize(); } - void TearDown() override { - manager.clearHistory(); + void advance_time(uint32_t seconds) { + mock_timestamp += seconds; } }; -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: AlertManager 构造函数初始化默认阈值 +TEST_F(AlertManagerTest, testAlertManagerInitialization) { + EXPECT_FLOAT_EQ(alert_manager.getAltitudeThresholds().first, 100.0f); + EXPECT_FLOAT_EQ(alert_manager.getAltitudeThresholds().second, -50.0f); + EXPECT_TRUE(alert_manager.getActiveAlerts().empty()); + EXPECT_TRUE(alert_manager.getAlertHistory(10).empty()); } -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: AlertManager 析构函数无副作用 +TEST_F(AlertManagerTest, testAlertManagerDestructor) { + // No specific assertion needed as destructor has no logic + // Just ensure it compiles and runs without crash + EXPECT_TRUE(true); } -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: initialize 输出初始化信息 +TEST_F(AlertManagerTest, testInitializeOutput) { + // Capture output using a mock stream + std::ostringstream buffer; + std::streambuf* orig_cout = std::cout.rdbuf(); + std::cout.rdbuf(buffer.rdbuf()); + + alert_manager.initialize(); + + std::cout.rdbuf(orig_cout); + + EXPECT_NE(buffer.str().find("预警管理器初始化完成。"), std::string::npos); + EXPECT_NE(buffer.str().find("默认预警阈值:上限=100米,下限=-50米"), std::string::npos); } -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: checkAltitudeAlert 正常高度不触发预警 +TEST_F(AlertManagerTest, testCheckAltitudeAlertNormalHeight) { + alert_manager.checkAltitudeAlert(50.0f, mock_timestamp); + EXPECT_TRUE(alert_manager.getActiveAlerts().empty()); } -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: checkAltitudeAlert 触发上限预警 +TEST_F(AlertManagerTest, testCheckAltitudeAlertUpperThreshold) { + alert_manager.checkAltitudeAlert(101.0f, mock_timestamp); + auto active_alerts = alert_manager.getActiveAlerts(); + EXPECT_EQ(active_alerts.size(), 1); + EXPECT_EQ(active_alerts[0].type, AlertType::ALTITUDE_UPPER); + EXPECT_EQ(active_alerts[0].status, AlertStatus::ACTIVE); + EXPECT_EQ(active_alerts[0].trigger_time, mock_timestamp); } -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: checkAltitudeAlert 触发下限预警 +TEST_F(AlertManagerTest, testCheckAltitudeAlertLowerThreshold) { + alert_manager.checkAltitudeAlert(-51.0f, mock_timestamp); + auto active_alerts = alert_manager.getActiveAlerts(); + EXPECT_EQ(active_alerts.size(), 1); + EXPECT_EQ(active_alerts[0].type, AlertType::ALTITUDE_LOWER); + EXPECT_EQ(active_alerts[0].status, AlertStatus::ACTIVE); } -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: checkAltitudeAlert 高度恢复后解除预警 +TEST_F(AlertManagerTest, testCheckAltitudeAlertResolution) { + alert_manager.checkAltitudeAlert(101.0f, mock_timestamp); // Trigger upper + advance_time(10); + alert_manager.checkAltitudeAlert(99.0f, mock_timestamp); // Resolve + auto active_alerts = alert_manager.getActiveAlerts(); + EXPECT_TRUE(active_alerts.empty()); } +// Test case: setAltitudeThresholds 合法输入更新阈值 +TEST_F(AlertManagerTest, testSetAltitudeThresholdsValidInput) { + alert_manager.setAltitudeThresholds(150.0f, -80.0f); + auto thresholds = alert_manager.getAltitudeThresholds(); + EXPECT_FLOAT_EQ(thresholds.first, 150.0f); + EXPECT_FLOAT_EQ(thresholds.second, -80.0f); +} + +// Test case: setAltitudeThresholds 非法输入(上界 <= 下界) +TEST_F(AlertManagerTest, testSetAltitudeThresholdsInvalidInput) { + std::ostringstream buffer; + std::streambuf* orig_cerr = std::cerr.rdbuf(); + std::cerr.rdbuf(buffer.rdbuf()); + + alert_manager.setAltitudeThresholds(50.0f, 100.0f); + + std::cerr.rdbuf(orig_cerr); + EXPECT_NE(buffer.str().find("错误:上限阈值必须大于下限阈值。"), std::string::npos); +} + +// Test case: triggerSensorFailure 触发传感器故障预警 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); + alert_manager.triggerSensorFailure(mock_timestamp); + auto active_alerts = alert_manager.getActiveAlerts(); + EXPECT_EQ(active_alerts.size(), 1); + EXPECT_EQ(active_alerts[0].type, AlertType::SENSOR_FAILURE); + EXPECT_EQ(active_alerts[0].status, AlertStatus::ACTIVE); + EXPECT_EQ(active_alerts[0].trigger_time, mock_timestamp); } +// Test case: triggerLowBattery 触发低电量预警 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); + alert_manager.triggerLowBattery(mock_timestamp, 15.5f); + auto active_alerts = alert_manager.getActiveAlerts(); + EXPECT_EQ(active_alerts.size(), 1); + EXPECT_EQ(active_alerts[0].type, AlertType::LOW_BATTERY); + EXPECT_EQ(active_alerts[0].status, AlertStatus::ACTIVE); + EXPECT_EQ(active_alerts[0].trigger_time, mock_timestamp); + EXPECT_NE(active_alerts[0].description.find("低电量警告:当前电量15.5%"), std::string::npos); } +// Test case: triggerCommunicationError 触发通信错误预警 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); + alert_manager.triggerCommunicationError(mock_timestamp); + auto active_alerts = alert_manager.getActiveAlerts(); + EXPECT_EQ(active_alerts.size(), 1); + EXPECT_EQ(active_alerts[0].type, AlertType::COMMUNICATION_ERROR); + EXPECT_EQ(active_alerts[0].status, AlertStatus::ACTIVE); + EXPECT_EQ(active_alerts[0].trigger_time, mock_timestamp); } +// Test case: acknowledgeAllAlerts 确认所有活动预警 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); + alert_manager.triggerSensorFailure(mock_timestamp); + alert_manager.triggerLowBattery(mock_timestamp + 1, 10.0f); + alert_manager.acknowledgeAllAlerts(mock_timestamp + 2); + auto active_alerts = alert_manager.getActiveAlerts(); + EXPECT_TRUE(active_alerts.empty()); + EXPECT_EQ(alert_manager.getActiveAlerts().size(), 0); } +// Test case: acknowledgeAlert 确认指定类型预警 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()); + alert_manager.triggerSensorFailure(mock_timestamp); + alert_manager.acknowledgeAlert(AlertType::SENSOR_FAILURE, mock_timestamp + 1); + auto active_alerts = alert_manager.getActiveAlerts(); + EXPECT_TRUE(active_alerts.empty()); } +// Test case: resetAlert 重置指定类型预警 TEST_F(AlertManagerTest, testResetAlert) { - mock_timestamp = 1640995200; - manager.triggerSensorFailure(mock_timestamp); - manager.resetAlert(AlertType::SENSOR_FAILURE); - auto alerts = manager.getActiveAlerts(); - EXPECT_TRUE(alerts.empty()); + alert_manager.triggerSensorFailure(mock_timestamp); + alert_manager.resetAlert(AlertType::SENSOR_FAILURE); + auto active_alerts = alert_manager.getActiveAlerts(); + EXPECT_TRUE(active_alerts.empty()); } +// Test case: resetAllAlerts 重置所有预警 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()); + alert_manager.triggerSensorFailure(mock_timestamp); + alert_manager.triggerLowBattery(mock_timestamp + 1, 10.0f); + alert_manager.resetAllAlerts(); + EXPECT_TRUE(alert_manager.getActiveAlerts().empty()); } +// Test case: hasActiveAlerts 检查是否存在活动预警 TEST_F(AlertManagerTest, testHasActiveAlerts) { - EXPECT_FALSE(manager.hasActiveAlerts()); - mock_timestamp = 1640995200; - manager.triggerSensorFailure(mock_timestamp); - EXPECT_TRUE(manager.hasActiveAlerts()); + EXPECT_FALSE(alert_manager.hasActiveAlerts()); + alert_manager.triggerSensorFailure(mock_timestamp); + EXPECT_TRUE(alert_manager.hasActiveAlerts()); } +// Test case: hasUnacknowledgedAlerts 检查是否存在未确认预警 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()); + EXPECT_FALSE(alert_manager.hasUnacknowledgedAlerts()); + alert_manager.triggerSensorFailure(mock_timestamp); + EXPECT_TRUE(alert_manager.hasUnacknowledgedAlerts()); + alert_manager.acknowledgeAlert(AlertType::SENSOR_FAILURE, mock_timestamp + 1); + EXPECT_FALSE(alert_manager.hasUnacknowledgedAlerts()); } +// Test case: getActiveAlerts 获取活动预警列表 TEST_F(AlertManagerTest, testGetActiveAlerts) { - mock_timestamp = 1640995200; - manager.triggerSensorFailure(mock_timestamp); - auto alerts = manager.getActiveAlerts(); + alert_manager.triggerSensorFailure(mock_timestamp); + auto alerts = alert_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 case: getAlertHistory 获取最近历史记录 +TEST_F(AlertManagerTest, testGetAlertHistory) { + alert_manager.triggerSensorFailure(mock_timestamp); + alert_manager.triggerLowBattery(mock_timestamp + 1, 10.0f); + auto history = alert_manager.getAlertHistory(1); + EXPECT_EQ(history.size(), 1); + EXPECT_EQ(history[0].type, AlertType::LOW_BATTERY); } -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 case: getStatusDescription 生成系统状态描述 +TEST_F(AlertManagerTest, testGetStatusDescription) { + alert_manager.triggerSensorFailure(mock_timestamp); + std::string status_desc = alert_manager.getStatusDescription(); + EXPECT_NE(status_desc.find("系统状态:有1个活动预警"), std::string::npos); + EXPECT_NE(status_desc.find("传感器故障"), std::string::npos); } +// Test case: clearHistory 清空历史记录 TEST_F(AlertManagerTest, testClearHistory) { - mock_timestamp = 1640995200; - manager.triggerSensorFailure(mock_timestamp); - manager.clearHistory(); - EXPECT_TRUE(manager.getAlertHistory(10).empty()); + alert_manager.triggerSensorFailure(mock_timestamp); + alert_manager.clearHistory(); + EXPECT_TRUE(alert_manager.getAlertHistory(10).empty()); } +// Test case: addToHistory 添加历史记录并限制数量 TEST_F(AlertManagerTest, testAddToHistoryLimit) { - mock_timestamp = 1640995200; - for (int i = 0; i < 100; ++i) { - manager.triggerSensorFailure(mock_timestamp + i); + const size_t MAX_HISTORY = 5; + for (int i = 0; i < MAX_HISTORY + 1; ++i) { + alert_manager.triggerSensorFailure(mock_timestamp + i); } - auto history = manager.getAlertHistory(10); - EXPECT_EQ(history.size(), 10); + auto history = alert_manager.getAlertHistory(MAX_HISTORY); + EXPECT_EQ(history.size(), MAX_HISTORY); + EXPECT_EQ(history[0].trigger_time, mock_timestamp + 1); + EXPECT_EQ(history[MAX_HISTORY - 1].trigger_time, mock_timestamp + MAX_HISTORY); } - -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