236 lines
9.3 KiB
C++
236 lines
9.3 KiB
C++
#include "gtest/gtest.h"
|
||
#include "alert_manager.hpp"
|
||
#include <vector>
|
||
#include <string>
|
||
#include <ctime>
|
||
|
||
// Mock time for deterministic testing
|
||
static uint32_t mock_timestamp = 1640995200; // 2022-01-01 00:00:00 UTC
|
||
|
||
class MockTime {
|
||
public:
|
||
static void setTimestamp(uint32_t ts) { mock_timestamp = ts; }
|
||
static uint32_t getTimestamp() { return mock_timestamp; }
|
||
};
|
||
|
||
// Helper function to get current timestamp
|
||
uint32_t getCurrentTimestamp() {
|
||
return MockTime::getTimestamp();
|
||
}
|
||
|
||
// Test fixture for AlertManager
|
||
class AlertManagerTest : public ::testing::Test {
|
||
protected:
|
||
AlertManager alert_manager;
|
||
|
||
void SetUp() override {
|
||
MockTime::setTimestamp(1640995200);
|
||
alert_manager.initialize();
|
||
}
|
||
|
||
void TearDown() override {
|
||
alert_manager.resetAllAlerts();
|
||
}
|
||
};
|
||
|
||
// Test case: AlertManager 构造函数初始化默认阈值
|
||
TEST_F(AlertManagerTest, testAlertManagerDefaultThresholds) {
|
||
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 case: AlertManager 析构函数无副作用
|
||
TEST_F(AlertManagerTest, testAlertManagerDestructor) {
|
||
// No specific assertion needed as destructor has no logic
|
||
// Just ensure it doesn't crash
|
||
AlertManager* manager = new AlertManager();
|
||
delete manager;
|
||
EXPECT_TRUE(true); // Dummy assertion to pass test
|
||
}
|
||
|
||
// Test case: initialize 输出初始化信息
|
||
TEST_F(AlertManagerTest, testInitializeOutput) {
|
||
std::stringstream buffer;
|
||
std::streambuf* old_cout = std::cout.rdbuf(buffer.rdbuf());
|
||
alert_manager.initialize();
|
||
std::cout.rdbuf(old_cout);
|
||
EXPECT_NE(buffer.str().find("预警管理器初始化完成"), std::string::npos);
|
||
EXPECT_NE(buffer.str().find("默认预警阈值：上限=100米，下限=-50米"), std::string::npos);
|
||
}
|
||
|
||
// Test case: checkAltitudeAlert 正常高度不触发预警
|
||
TEST_F(AlertManagerTest, testCheckAltitudeAlertNormalHeight) {
|
||
alert_manager.checkAltitudeAlert(50.0f, getCurrentTimestamp());
|
||
EXPECT_FALSE(alert_manager.hasActiveAlerts());
|
||
}
|
||
|
||
// Test case: checkAltitudeAlert 高于上限触发预警
|
||
TEST_F(AlertManagerTest, testCheckAltitudeAlertUpperThresholdTrigger) {
|
||
alert_manager.checkAltitudeAlert(101.0f, getCurrentTimestamp());
|
||
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);
|
||
}
|
||
|
||
// Test case: checkAltitudeAlert 低于下限触发预警
|
||
TEST_F(AlertManagerTest, testCheckAltitudeAlertLowerThresholdTrigger) {
|
||
alert_manager.checkAltitudeAlert(-51.0f, getCurrentTimestamp());
|
||
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 case: checkAltitudeAlert 恢复正常后解除预警
|
||
TEST_F(AlertManagerTest, testCheckAltitudeAlertResolution) {
|
||
alert_manager.checkAltitudeAlert(101.0f, getCurrentTimestamp());
|
||
alert_manager.checkAltitudeAlert(99.0f, getCurrentTimestamp());
|
||
EXPECT_FALSE(alert_manager.hasActiveAlerts());
|
||
}
|
||
|
||
// 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::stringstream buffer;
|
||
std::streambuf* old_cerr = std::cerr.rdbuf(buffer.rdbuf());
|
||
alert_manager.setAltitudeThresholds(50.0f, 100.0f);
|
||
std::cerr.rdbuf(old_cerr);
|
||
EXPECT_NE(buffer.str().find("错误：上限阈值必须大于下限阈值"), std::string::npos);
|
||
}
|
||
|
||
// Test case: triggerSensorFailure 触发传感器故障预警
|
||
TEST_F(AlertManagerTest, testTriggerSensorFailure) {
|
||
alert_manager.triggerSensorFailure(getCurrentTimestamp());
|
||
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);
|
||
}
|
||
|
||
// Test case: triggerLowBattery 触发低电量预警
|
||
TEST_F(AlertManagerTest, testTriggerLowBattery) {
|
||
alert_manager.triggerLowBattery(getCurrentTimestamp(), 15.0f);
|
||
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_NE(active_alerts[0].description.find("当前电量15%"), std::string::npos);
|
||
}
|
||
|
||
// Test case: triggerCommunicationError 触发通信错误预警
|
||
TEST_F(AlertManagerTest, testTriggerCommunicationError) {
|
||
alert_manager.triggerCommunicationError(getCurrentTimestamp());
|
||
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);
|
||
}
|
||
|
||
// Test case: acknowledgeAllAlerts 确认所有活动预警
|
||
TEST_F(AlertManagerTest, testAcknowledgeAllAlerts) {
|
||
alert_manager.triggerSensorFailure(getCurrentTimestamp());
|
||
alert_manager.triggerLowBattery(getCurrentTimestamp(), 10.0f);
|
||
alert_manager.acknowledgeAllAlerts(getCurrentTimestamp());
|
||
auto active_alerts = alert_manager.getActiveAlerts();
|
||
EXPECT_TRUE(active_alerts.empty());
|
||
EXPECT_TRUE(alert_manager.hasUnacknowledgedAlerts());
|
||
}
|
||
|
||
// Test case: acknowledgeAlert 确认指定类型预警
|
||
TEST_F(AlertManagerTest, testAcknowledgeAlert) {
|
||
alert_manager.triggerSensorFailure(getCurrentTimestamp());
|
||
alert_manager.acknowledgeAlert(AlertType::SENSOR_FAILURE, getCurrentTimestamp());
|
||
auto active_alerts = alert_manager.getActiveAlerts();
|
||
EXPECT_TRUE(active_alerts.empty());
|
||
}
|
||
|
||
// Test case: resetAlert 重置指定类型预警
|
||
TEST_F(AlertManagerTest, testResetAlert) {
|
||
alert_manager.triggerSensorFailure(getCurrentTimestamp());
|
||
alert_manager.resetAlert(AlertType::SENSOR_FAILURE);
|
||
EXPECT_FALSE(alert_manager.hasActiveAlerts());
|
||
}
|
||
|
||
// Test case: resetAllAlerts 清除所有活动预警
|
||
TEST_F(AlertManagerTest, testResetAllAlerts) {
|
||
alert_manager.triggerSensorFailure(getCurrentTimestamp());
|
||
alert_manager.triggerLowBattery(getCurrentTimestamp(), 10.0f);
|
||
alert_manager.resetAllAlerts();
|
||
EXPECT_FALSE(alert_manager.hasActiveAlerts());
|
||
EXPECT_FALSE(alert_manager.hasUnacknowledgedAlerts());
|
||
}
|
||
|
||
// Test case: hasActiveAlerts 检查是否存在活动预警
|
||
TEST_F(AlertManagerTest, testHasActiveAlerts) {
|
||
EXPECT_FALSE(alert_manager.hasActiveAlerts());
|
||
alert_manager.triggerSensorFailure(getCurrentTimestamp());
|
||
EXPECT_TRUE(alert_manager.hasActiveAlerts());
|
||
}
|
||
|
||
// Test case: hasUnacknowledgedAlerts 检查是否存在未确认预警
|
||
TEST_F(AlertManagerTest, testHasUnacknowledgedAlerts) {
|
||
EXPECT_FALSE(alert_manager.hasUnacknowledgedAlerts());
|
||
alert_manager.triggerSensorFailure(getCurrentTimestamp());
|
||
EXPECT_TRUE(alert_manager.hasUnacknowledgedAlerts());
|
||
}
|
||
|
||
// Test case: getActiveAlerts 返回活动预警列表
|
||
TEST_F(AlertManagerTest, testGetActiveAlerts) {
|
||
alert_manager.triggerSensorFailure(getCurrentTimestamp());
|
||
auto alerts = alert_manager.getActiveAlerts();
|
||
EXPECT_EQ(alerts.size(), 1);
|
||
EXPECT_EQ(alerts[0].type, AlertType::SENSOR_FAILURE);
|
||
}
|
||
|
||
// Test case: getAlertHistory 获取最近历史记录
|
||
TEST_F(AlertManagerTest, testGetAlertHistory) {
|
||
for (int i = 0; i < 5; ++i) {
|
||
alert_manager.triggerSensorFailure(getCurrentTimestamp() + i);
|
||
}
|
||
auto history = alert_manager.getAlertHistory(3);
|
||
EXPECT_EQ(history.size(), 3);
|
||
EXPECT_EQ(history[0].type, AlertType::SENSOR_FAILURE);
|
||
}
|
||
|
||
// Test case: getStatusDescription 生成系统状态描述
|
||
TEST_F(AlertManagerTest, testGetStatusDescription) {
|
||
std::string status_desc = alert_manager.getStatusDescription();
|
||
EXPECT_NE(status_desc.find("系统状态：正常"), std::string::npos);
|
||
alert_manager.triggerSensorFailure(getCurrentTimestamp());
|
||
status_desc = alert_manager.getStatusDescription();
|
||
EXPECT_NE(status_desc.find("有1个活动预警"), std::string::npos);
|
||
}
|
||
|
||
// Test case: clearHistory 清空历史记录
|
||
TEST_F(AlertManagerTest, testClearHistory) {
|
||
alert_manager.triggerSensorFailure(getCurrentTimestamp());
|
||
alert_manager.clearHistory();
|
||
EXPECT_TRUE(alert_manager.getAlertHistory(10).empty());
|
||
}
|
||
|
||
// Test case: addToHistory 添加历史记录并限制数量
|
||
TEST_F(AlertManagerTest, testAddToHistoryLimit) {
|
||
const size_t MAX_COUNT = 5;
|
||
for (size_t i = 0; i < MAX_COUNT + 2; ++i) {
|
||
AlertInfo alert;
|
||
alert.type = AlertType::SENSOR_FAILURE;
|
||
alert.status = AlertStatus::ACTIVE;
|
||
alert.trigger_time = getCurrentTimestamp() + i;
|
||
alert.description = "Test alert";
|
||
alert_manager.addToHistory(alert);
|
||
}
|
||
auto history = alert_manager.getAlertHistory(MAX_COUNT);
|
||
EXPECT_EQ(history.size(), MAX_COUNT);
|
||
EXPECT_EQ(history[0].trigger_time, getCurrentTimestamp() + 2);
|
||
}
|