AI 自动生成测试用例

This commit is contained in:
lids 2026-05-12 14:24:52 +08:00
parent 3b3119277e
commit dce30619ed
1 changed files with 240 additions and 215 deletions

View File

@ -1,267 +1,292 @@
#include "gtest/gtest.h"
#include <gtest/gtest.h>
#include <gmock/gmock.h>
#include "alert_manager.hpp"
#include <vector>
#include <string>
#include <sstream>
#include <iostream>
// Mock for time to ensure deterministic tests
static uint32_t mock_timestamp = 1640995200; // 2022-01-01 00:00:00 UTC
using namespace testing;
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;
AlertManager alertManager;
void SetUp() override {
mock_timestamp = 1640995200;
manager.initialize();
// 重置预警管理器状态
alertManager.resetAllAlerts();
}
void TearDown() override {
manager.clearHistory();
// 清理
}
};
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) {
alertManager.setAltitudeThresholds(100.0f, -50.0f);
alertManager.checkAltitudeAlert(50.0f, 1000);
EXPECT_FALSE(alertManager.hasActiveAlerts());
EXPECT_FALSE(alertManager.hasUnacknowledgedAlerts());
}
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) {
alertManager.setAltitudeThresholds(100.0f, -50.0f);
alertManager.checkAltitudeAlert(150.0f, 1001);
EXPECT_TRUE(alertManager.hasActiveAlerts());
EXPECT_TRUE(alertManager.hasUnacknowledgedAlerts());
auto activeAlerts = alertManager.getActiveAlerts();
ASSERT_EQ(activeAlerts.size(), 1);
EXPECT_EQ(activeAlerts[0].type, AlertType::ALTITUDE_UPPER);
EXPECT_EQ(activeAlerts[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_F(AlertManagerTest, testCheckAltitudeAlertLowerThresholdExceeded) {
alertManager.setAltitudeThresholds(100.0f, -50.0f);
alertManager.checkAltitudeAlert(-100.0f, 1002);
EXPECT_TRUE(alertManager.hasActiveAlerts());
EXPECT_TRUE(alertManager.hasUnacknowledgedAlerts());
auto activeAlerts = alertManager.getActiveAlerts();
ASSERT_EQ(activeAlerts.size(), 1);
EXPECT_EQ(activeAlerts[0].type, AlertType::ALTITUDE_LOWER);
EXPECT_EQ(activeAlerts[0].status, AlertStatus::ACTIVE);
}
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) {
alertManager.setAltitudeThresholds(100.0f, -50.0f);
alertManager.checkAltitudeAlert(100.0f, 1003);
EXPECT_FALSE(alertManager.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) {
alertManager.setAltitudeThresholds(100.0f, -50.0f);
alertManager.checkAltitudeAlert(-50.0f, 1004);
EXPECT_FALSE(alertManager.hasActiveAlerts());
}
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, testCheckAltitudeAlertJustAboveUpper) {
alertManager.setAltitudeThresholds(100.0f, -50.0f);
alertManager.checkAltitudeAlert(100.001f, 1005);
EXPECT_TRUE(alertManager.hasActiveAlerts());
auto activeAlerts = alertManager.getActiveAlerts();
EXPECT_EQ(activeAlerts[0].type, AlertType::ALTITUDE_UPPER);
}
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, testCheckAltitudeAlertJustBelowLower) {
alertManager.setAltitudeThresholds(100.0f, -50.0f);
alertManager.checkAltitudeAlert(-50.001f, 1006);
EXPECT_TRUE(alertManager.hasActiveAlerts());
auto activeAlerts = alertManager.getActiveAlerts();
EXPECT_EQ(activeAlerts[0].type, AlertType::ALTITUDE_LOWER);
}
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, testCheckAltitudeAlertExtremeHighValue) {
alertManager.setAltitudeThresholds(100.0f, -50.0f);
alertManager.checkAltitudeAlert(999999.0f, 1007);
EXPECT_TRUE(alertManager.hasActiveAlerts());
auto activeAlerts = alertManager.getActiveAlerts();
EXPECT_EQ(activeAlerts[0].type, AlertType::ALTITUDE_UPPER);
}
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, testCheckAltitudeAlertExtremeLowValue) {
alertManager.setAltitudeThresholds(100.0f, -50.0f);
alertManager.checkAltitudeAlert(-999999.0f, 1008);
EXPECT_TRUE(alertManager.hasActiveAlerts());
auto activeAlerts = alertManager.getActiveAlerts();
EXPECT_EQ(activeAlerts[0].type, AlertType::ALTITUDE_LOWER);
}
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, testCheckAltitudeAlertResolveAfterNormalization) {
alertManager.setAltitudeThresholds(100.0f, -50.0f);
// 先触发预警
alertManager.checkAltitudeAlert(150.0f, 1009);
EXPECT_TRUE(alertManager.hasActiveAlerts());
// 高度恢复正常
alertManager.checkAltitudeAlert(50.0f, 1010);
EXPECT_FALSE(alertManager.hasActiveAlerts());
// 检查历史记录中预警状态已更新
auto history = alertManager.getAlertHistory(10);
bool foundResolved = false;
for (const auto& alert : history) {
if (alert.type == AlertType::ALTITUDE_UPPER &&
alert.status == AlertStatus::RESOLVED) {
foundResolved = true;
break;
}
}
EXPECT_TRUE(foundResolved);
}
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, testCheckAltitudeAlertBothThresholdsExceeded) {
alertManager.setAltitudeThresholds(50.0f, -50.0f);
// 先触发上限预警
alertManager.checkAltitudeAlert(100.0f, 1011);
EXPECT_TRUE(alertManager.hasActiveAlerts());
// 再触发下限预警(高度从上限直接跳到下限)
alertManager.checkAltitudeAlert(-100.0f, 1012);
auto activeAlerts = alertManager.getActiveAlerts();
EXPECT_EQ(activeAlerts.size(), 1); // 只有下限预警活动
EXPECT_EQ(activeAlerts[0].type, AlertType::ALTITUDE_LOWER);
}
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, testCheckAltitudeAlertDuplicateTrigger) {
alertManager.setAltitudeThresholds(100.0f, -50.0f);
// 第一次触发
alertManager.checkAltitudeAlert(150.0f, 1013);
auto activeAlerts1 = alertManager.getActiveAlerts();
ASSERT_EQ(activeAlerts1.size(), 1);
// 第二次触发(高度仍然超出)
alertManager.checkAltitudeAlert(200.0f, 1014);
auto activeAlerts2 = alertManager.getActiveAlerts();
EXPECT_EQ(activeAlerts2.size(), 1); // 仍然只有一个预警
EXPECT_EQ(activeAlerts2[0].trigger_time, 1013); // 触发时间不应更新
}
TEST_F(AlertManagerTest, testResetAlert) {
mock_timestamp = 1640995200;
manager.triggerSensorFailure(mock_timestamp);
manager.resetAlert(AlertType::SENSOR_FAILURE);
auto alerts = manager.getActiveAlerts();
EXPECT_TRUE(alerts.empty());
// ==================== setAltitudeThresholds 测试 ====================
// 正常输入测试:设置有效的上下限阈值
TEST_F(AlertManagerTest, testSetAltitudeThresholdsValidValues) {
testing::internal::CaptureStdout();
alertManager.setAltitudeThresholds(200.0f, -100.0f);
std::string output = testing::internal::GetCapturedStdout();
EXPECT_THAT(output, HasSubstr("预警阈值已更新"));
EXPECT_THAT(output, HasSubstr("上限=200"));
EXPECT_THAT(output, HasSubstr("下限=-100"));
// 验证新阈值生效
alertManager.checkAltitudeAlert(150.0f, 2000);
EXPECT_FALSE(alertManager.hasActiveAlerts());
alertManager.checkAltitudeAlert(250.0f, 2001);
EXPECT_TRUE(alertManager.hasActiveAlerts());
}
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, testSetAltitudeThresholdsEqualValues) {
testing::internal::CaptureStderr();
alertManager.setAltitudeThresholds(100.0f, 100.0f);
std::string errorOutput = testing::internal::GetCapturedStderr();
EXPECT_THAT(errorOutput, HasSubstr("错误:上限阈值必须大于下限阈值"));
// 验证阈值未改变
alertManager.checkAltitudeAlert(150.0f, 2002);
EXPECT_TRUE(alertManager.hasActiveAlerts());
}
TEST_F(AlertManagerTest, testHasActiveAlerts) {
EXPECT_FALSE(manager.hasActiveAlerts());
mock_timestamp = 1640995200;
manager.triggerSensorFailure(mock_timestamp);
EXPECT_TRUE(manager.hasActiveAlerts());
// 异常输入测试:设置上限小于下限,应输出错误并拒绝
TEST_F(AlertManagerTest, testSetAltitudeThresholdsInvalidOrder) {
testing::internal::CaptureStderr();
alertManager.setAltitudeThresholds(-50.0f, 100.0f);
std::string errorOutput = testing::internal::GetCapturedStderr();
EXPECT_THAT(errorOutput, HasSubstr("错误:上限阈值必须大于下限阈值"));
// 验证阈值未改变
alertManager.checkAltitudeAlert(150.0f, 2003);
EXPECT_TRUE(alertManager.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, testSetAltitudeThresholdsExtremeValues) {
testing::internal::CaptureStdout();
alertManager.setAltitudeThresholds(1000000.0f, -1000000.0f);
std::string output = testing::internal::GetCapturedStdout();
EXPECT_THAT(output, HasSubstr("预警阈值已更新"));
// 验证新阈值生效
alertManager.checkAltitudeAlert(500000.0f, 2004);
EXPECT_FALSE(alertManager.hasActiveAlerts());
alertManager.checkAltitudeAlert(1500000.0f, 2005);
EXPECT_TRUE(alertManager.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, testSetAltitudeThresholdsResetsExistingAlerts) {
// 先触发预警
alertManager.checkAltitudeAlert(150.0f, 2006);
EXPECT_TRUE(alertManager.hasActiveAlerts());
// 设置新阈值
testing::internal::CaptureStdout();
alertManager.setAltitudeThresholds(200.0f, -100.0f);
testing::internal::GetCapturedStdout();
// 原有预警应被重置
EXPECT_FALSE(alertManager.hasActiveAlerts());
}
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, testSetAltitudeThresholdsMultipleTimes) {
testing::internal::CaptureStdout();
alertManager.setAltitudeThresholds(200.0f, -100.0f);
alertManager.setAltitudeThresholds(300.0f, -200.0f);
alertManager.setAltitudeThresholds(400.0f, -300.0f);
std::string output = testing::internal::GetCapturedStdout();
// 验证最后一次设置生效
EXPECT_THAT(output, HasSubstr("上限=400"));
EXPECT_THAT(output, HasSubstr("下限=-300"));
// 验证新阈值
alertManager.checkAltitudeAlert(350.0f, 2007);
EXPECT_FALSE(alertManager.hasActiveAlerts());
alertManager.checkAltitudeAlert(450.0f, 2008);
EXPECT_TRUE(alertManager.hasActiveAlerts());
}
TEST_F(AlertManagerTest, testGetStatusDescriptionNoAlerts) {
std::string desc = manager.getStatusDescription();
EXPECT_EQ(desc, "系统状态:正常");
// 特殊场景测试:设置阈值后立即检查高度
TEST_F(AlertManagerTest, testSetAltitudeThresholdsImmediateCheck) {
testing::internal::CaptureStdout();
alertManager.setAltitudeThresholds(50.0f, -50.0f);
testing::internal::GetCapturedStdout();
// 新阈值下,当前高度应触发预警
alertManager.checkAltitudeAlert(75.0f, 2009);
EXPECT_TRUE(alertManager.hasActiveAlerts());
auto activeAlerts = alertManager.getActiveAlerts();
EXPECT_EQ(activeAlerts[0].type, AlertType::ALTITUDE_UPPER);
}
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<AlertType>(99)), "未知预警");
}
TEST_F(AlertManagerTest, testGetAlertStatusName) {
EXPECT_EQ(manager.getAlertStatusName(AlertStatus::ACTIVE), "预警中");
EXPECT_EQ(manager.getAlertStatusName(AlertStatus::INACTIVE), "未激活");
EXPECT_EQ(manager.getAlertStatusName(static_cast<AlertStatus>(99)), "未知状态");
}
TEST_F(AlertManagerTest, testDestructorDoesNotCrash) {
// Verify destructor doesn't throw or crash
EXPECT_NO_FATAL_FAILURE({
AlertManager* mgr = new AlertManager();
delete mgr;
});
// 主函数
int main(int argc, char **argv) {
::testing::InitGoogleTest(&argc, argv);
return RUN_ALL_TESTS();
}