From e5c2de94f055d3aa80597ee6b4b0935fb4316de2 Mon Sep 17 00:00:00 2001 From: lids <1713278948@qq.com> Date: Mon, 11 May 2026 11:14:49 +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 | 492 ++++++++++++++++++----------------- 1 file changed, 257 insertions(+), 235 deletions(-) diff --git a/tests/test_alert_manager.cpp b/tests/test_alert_manager.cpp index 5bc247a..b91104f 100644 --- a/tests/test_alert_manager.cpp +++ b/tests/test_alert_manager.cpp @@ -1,267 +1,289 @@ -#include "gtest/gtest.h" +#include #include "alert_manager.hpp" -#include -#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 { -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; -} - -// Test fixture class AlertManagerTest : public ::testing::Test { protected: AlertManager manager; void SetUp() override { - mock_timestamp = 1640995200; - manager.initialize(); + // 重置阈值到默认值 + manager.setAltitudeThresholds(100.0f, -50.0f); + // 清空预警 + manager.resetAllAlerts(); } void TearDown() override { - manager.clearHistory(); + // 清理 + } + + // 辅助函数：捕获标准输出 + std::string captureStdout(std::function func) { + std::stringstream buffer; + std::streambuf* old = std::cout.rdbuf(buffer.rdbuf()); + func(); + std::cout.rdbuf(old); + return buffer.str(); + } + + // 辅助函数：捕获标准错误输出 + std::string captureStderr(std::function func) { + std::stringstream buffer; + std::streambuf* old = std::cerr.rdbuf(buffer.rdbuf()); + func(); + std::cerr.rdbuf(old); + return buffer.str(); } }; -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()); +// ==================== checkAltitudeAlert 测试 ==================== + +// 正常输入测试：高度在阈值范围内，不应触发预警 +TEST_F(AlertManagerTest, testCheckAltitudeAlertNormalRange) { + manager.checkAltitudeAlert(50.0f, 1000); + EXPECT_FALSE(manager.hasActiveAlerts()); } -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_F(AlertManagerTest, testCheckAltitudeAlertUpperThresholdExceeded) { + manager.checkAltitudeAlert(150.0f, 1000); + EXPECT_TRUE(manager.hasActiveAlerts()); + auto activeAlerts = manager.getActiveAlerts(); + ASSERT_EQ(1, activeAlerts.size()); + EXPECT_EQ(AlertType::ALTITUDE_UPPER, activeAlerts[0].type); + EXPECT_EQ(AlertStatus::ACTIVE, activeAlerts[0].status); } -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_F(AlertManagerTest, testCheckAltitudeAlertLowerThresholdExceeded) { + manager.checkAltitudeAlert(-100.0f, 1000); + EXPECT_TRUE(manager.hasActiveAlerts()); + auto activeAlerts = manager.getActiveAlerts(); + ASSERT_EQ(1, activeAlerts.size()); + EXPECT_EQ(AlertType::ALTITUDE_LOWER, activeAlerts[0].type); + EXPECT_EQ(AlertStatus::ACTIVE, activeAlerts[0].status); } -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_F(AlertManagerTest, testCheckAltitudeAlertAtUpperBoundary) { + manager.checkAltitudeAlert(100.0f, 1000); + EXPECT_FALSE(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_F(AlertManagerTest, testCheckAltitudeAlertAtLowerBoundary) { + manager.checkAltitudeAlert(-50.0f, 1000); + EXPECT_FALSE(manager.hasActiveAlerts()); } +// 边界值测试：高度略高于上限阈值，应触发预警 +TEST_F(AlertManagerTest, testCheckAltitudeAlertJustAboveUpper) { + manager.checkAltitudeAlert(100.001f, 1000); + EXPECT_TRUE(manager.hasActiveAlerts()); + auto activeAlerts = manager.getActiveAlerts(); + ASSERT_EQ(1, activeAlerts.size()); + EXPECT_EQ(AlertType::ALTITUDE_UPPER, activeAlerts[0].type); +} + +// 边界值测试：高度略低于下限阈值，应触发预警 +TEST_F(AlertManagerTest, testCheckAltitudeAlertJustBelowLower) { + manager.checkAltitudeAlert(-50.001f, 1000); + EXPECT_TRUE(manager.hasActiveAlerts()); + auto activeAlerts = manager.getActiveAlerts(); + ASSERT_EQ(1, activeAlerts.size()); + EXPECT_EQ(AlertType::ALTITUDE_LOWER, activeAlerts[0].type); +} + +// 特殊场景测试：高度从超出上限恢复到正常范围，应解除预警 +TEST_F(AlertManagerTest, testCheckAltitudeAlertRecoverFromUpper) { + manager.checkAltitudeAlert(150.0f, 1000); + EXPECT_TRUE(manager.hasActiveAlerts()); + + manager.checkAltitudeAlert(50.0f, 1001); + EXPECT_FALSE(manager.hasActiveAlerts()); +} + +// 特殊场景测试：高度从超出下限恢复到正常范围，应解除预警 +TEST_F(AlertManagerTest, testCheckAltitudeAlertRecoverFromLower) { + manager.checkAltitudeAlert(-100.0f, 1000); + EXPECT_TRUE(manager.hasActiveAlerts()); + + manager.checkAltitudeAlert(0.0f, 1001); + EXPECT_FALSE(manager.hasActiveAlerts()); +} + +// 特殊场景测试：同时超出上下限（不可能，但测试逻辑），应触发上限预警 +TEST_F(AlertManagerTest, testCheckAltitudeAlertBothThresholdsExceeded) { + // 设置阈值使上下限重叠，测试逻辑 + manager.setAltitudeThresholds(50.0f, -50.0f); + manager.checkAltitudeAlert(100.0f, 1000); + auto activeAlerts = manager.getActiveAlerts(); + ASSERT_EQ(1, activeAlerts.size()); + EXPECT_EQ(AlertType::ALTITUDE_UPPER, activeAlerts[0].type); +} + +// 特殊场景测试：连续触发相同预警，不应重复触发 +TEST_F(AlertManagerTest, testCheckAltitudeAlertDuplicateTrigger) { + manager.checkAltitudeAlert(150.0f, 1000); + EXPECT_TRUE(manager.hasActiveAlerts()); + + // 再次触发相同预警 + manager.checkAltitudeAlert(200.0f, 1001); + auto activeAlerts = manager.getActiveAlerts(); + ASSERT_EQ(1, activeAlerts.size()); + EXPECT_EQ(AlertType::ALTITUDE_UPPER, activeAlerts[0].type); +} + +// 异常输入测试：高度为极大值 +TEST_F(AlertManagerTest, testCheckAltitudeAlertExtremeHigh) { + manager.checkAltitudeAlert(1e10f, 1000); + EXPECT_TRUE(manager.hasActiveAlerts()); + auto activeAlerts = manager.getActiveAlerts(); + ASSERT_EQ(1, activeAlerts.size()); + EXPECT_EQ(AlertType::ALTITUDE_UPPER, activeAlerts[0].type); +} + +// 异常输入测试：高度为极小值 +TEST_F(AlertManagerTest, testCheckAltitudeAlertExtremeLow) { + manager.checkAltitudeAlert(-1e10f, 1000); + EXPECT_TRUE(manager.hasActiveAlerts()); + auto activeAlerts = manager.getActiveAlerts(); + ASSERT_EQ(1, activeAlerts.size()); + EXPECT_EQ(AlertType::ALTITUDE_LOWER, activeAlerts[0].type); +} + +// 异常输入测试：高度为0 +TEST_F(AlertManagerTest, testCheckAltitudeAlertZero) { + manager.checkAltitudeAlert(0.0f, 1000); + EXPECT_FALSE(manager.hasActiveAlerts()); +} + +// 异常输入测试：时间戳为0 +TEST_F(AlertManagerTest, testCheckAltitudeAlertTimestampZero) { + manager.checkAltitudeAlert(150.0f, 0); + EXPECT_TRUE(manager.hasActiveAlerts()); + auto activeAlerts = manager.getActiveAlerts(); + ASSERT_EQ(1, activeAlerts.size()); + EXPECT_EQ(0, activeAlerts[0].trigger_time); +} + +// ==================== setAltitudeThresholds 测试 ==================== + +// 正常输入测试：设置有效的上下限阈值 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_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_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_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_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_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_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()); -} - -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) { + + // 验证新阈值生效 + manager.checkAltitudeAlert(150.0f, 1000); EXPECT_FALSE(manager.hasActiveAlerts()); - mock_timestamp = 1640995200; - manager.triggerSensorFailure(mock_timestamp); + + manager.checkAltitudeAlert(250.0f, 1001); 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, testSetAltitudeThresholdsResetsOldAlerts) { + // 先触发预警 + manager.checkAltitudeAlert(150.0f, 1000); + EXPECT_TRUE(manager.hasActiveAlerts()); + + // 设置新阈值 + manager.setAltitudeThresholds(200.0f, -100.0f); + EXPECT_FALSE(manager.hasActiveAlerts()); } -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; +// 边界值测试：上下限相等，应输出错误并拒绝 +TEST_F(AlertManagerTest, testSetAltitudeThresholdsEqual) { + std::string errOutput = captureStderr([this]() { + manager.setAltitudeThresholds(100.0f, 100.0f); }); -} \ No newline at end of file + EXPECT_FALSE(errOutput.empty()); + EXPECT_NE(std::string::npos, errOutput.find("错误")); + + // 验证阈值未改变 + manager.checkAltitudeAlert(100.0f, 1000); + EXPECT_FALSE(manager.hasActiveAlerts()); + + manager.checkAltitudeAlert(150.0f, 1001); + EXPECT_TRUE(manager.hasActiveAlerts()); +} + +// 边界值测试：上限小于下限，应输出错误并拒绝 +TEST_F(AlertManagerTest, testSetAltitudeThresholdsUpperLessThanLower) { + std::string errOutput = captureStderr([this]() { + manager.setAltitudeThresholds(-50.0f, 100.0f); + }); + EXPECT_FALSE(errOutput.empty()); + EXPECT_NE(std::string::npos, errOutput.find("错误")); + + // 验证阈值未改变 + EXPECT_TRUE(manager.hasActiveAlerts() == false); +} + +// 边界值测试：设置极大值 +TEST_F(AlertManagerTest, testSetAltitudeThresholdsExtremeValues) { + manager.setAltitudeThresholds(1e10f, -1e10f); + + manager.checkAltitudeAlert(1e9f, 1000); + EXPECT_FALSE(manager.hasActiveAlerts()); + + manager.checkAltitudeAlert(1e11f, 1001); + EXPECT_TRUE(manager.hasActiveAlerts()); +} + +// 边界值测试：设置极小值 +TEST_F(AlertManagerTest, testSetAltitudeThresholdsTinyValues) { + manager.setAltitudeThresholds(0.001f, -0.001f); + + manager.checkAltitudeAlert(0.0f, 1000); + EXPECT_FALSE(manager.hasActiveAlerts()); + + manager.checkAltitudeAlert(0.002f, 1001); + EXPECT_TRUE(manager.hasActiveAlerts()); +} + +// 特殊场景测试：多次设置阈值，验证最终生效 +TEST_F(AlertManagerTest, testSetAltitudeThresholdsMultipleTimes) { + manager.setAltitudeThresholds(200.0f, -100.0f); + manager.setAltitudeThresholds(300.0f, -200.0f); + manager.setAltitudeThresholds(150.0f, -50.0f); + + // 验证最终阈值 + manager.checkAltitudeAlert(100.0f, 1000); + EXPECT_FALSE(manager.hasActiveAlerts()); + + manager.checkAltitudeAlert(200.0f, 1001); + EXPECT_TRUE(manager.hasActiveAlerts()); +} + +// 特殊场景测试：设置阈值后立即检查边界值 +TEST_F(AlertManagerTest, testSetAltitudeThresholdsCheckBoundary) { + manager.setAltitudeThresholds(100.5f, -50.5f); + + // 边界值不应触发预警 + manager.checkAltitudeAlert(100.5f, 1000); + EXPECT_FALSE(manager.hasActiveAlerts()); + + manager.checkAltitudeAlert(-50.5f, 1001); + EXPECT_FALSE(manager.hasActiveAlerts()); +} + +// 异常输入测试：设置NaN值（如果编译器支持） +TEST_F(AlertManagerTest, testSetAltitudeThresholdsNaN) { + float nan = std::numeric_limits::quiet_NaN(); + // NaN比较总是false，所以upper <= lower可能为false，导致设置成功 + manager.setAltitudeThresholds(nan, nan); + // 验证不会崩溃 + manager.checkAltitudeAlert(50.0f, 1000); +} + +// 异常输入测试：设置INF值 +TEST_F(AlertManagerTest, testSetAltitudeThresholdsInfinity) { + float inf = std::numeric_limits::infinity(); + manager.setAltitudeThresholds(inf, -inf); + + // 验证不会崩溃 + manager.checkAltitudeAlert(50.0f, 1000); + EXPECT_FALSE(manager.hasActiveAlerts()); +}