#include "gtest/gtest.h"
#include "alert_manager.hpp"
#include <vector>
#include <string>

// Mock time function for deterministic testing
uint32_t mock_timestamp = 1640995200; // 2022-01-01 00:00:00 UTC

typedef uint32_t (*TimeFunc)();
static TimeFunc original_time_func = std::time;
uint32_t mock_time() {
    return mock_timestamp;
}

// Test fixture for AlertManager
class AlertManagerTest : public ::testing::Test {
protected:
    AlertManager alert_manager;

    void SetUp() override {
        // Reset mock timestamp
        mock_timestamp = 1640995200;
        // Replace std::time with mock_time
        auto old_func = std::time;
        std::time = mock_time;
    }

    void TearDown() override {
        // Restore original time function
        std::time = original_time_func;
    }

    void advance_time(uint32_t seconds) {
        mock_timestamp += seconds;
    }
};

// Test case: 正常输入测试 - 设置合法阈值
TEST_F(AlertManagerTest, testSetAltitudeThresholdsNormalInput) {
    alert_manager.setAltitudeThresholds(200.0f, 50.0f);

    EXPECT_FLOAT_EQ(alert_manager.getUpperThreshold(), 200.0f);
    EXPECT_FLOAT_EQ(alert_manager.getLowerThreshold(), 50.0f);
    EXPECT_FALSE(alert_manager.hasActiveAlerts());
}

// Test case: 边界值测试 - 上限等于下限（非法输入）
TEST_F(AlertManagerTest, testSetAltitudeThresholdsBoundaryEqual) {
    alert_manager.setAltitudeThresholds(100.0f, 100.0f);

    EXPECT_FLOAT_EQ(alert_manager.getUpperThreshold(), 100.0f);
    EXPECT_FLOAT_EQ(alert_manager.getLowerThreshold(), 100.0f);
    EXPECT_FALSE(alert_manager.hasActiveAlerts());
}

// Test case: 异常输入测试 - 上限小于下限（非法输入）
TEST_F(AlertManagerTest, testSetAltitudeThresholdsInvalidInput) {
    alert_manager.setAltitudeThresholds(50.0f, 100.0f);

    EXPECT_FLOAT_EQ(alert_manager.getUpperThreshold(), 100.0f); // default value
    EXPECT_FLOAT_EQ(alert_manager.getLowerThreshold(), -50.0f); // default value
    EXPECT_FALSE(alert_manager.hasActiveAlerts());
}

// Test case: 特殊场景测试 - 阈值更新后重置预警状态
TEST_F(AlertManagerTest, testSetAltitudeThresholdsResetAlerts) {
    // Trigger an altitude upper alert
    alert_manager.checkAltitudeAlert(150.0f, mock_timestamp);
    EXPECT_TRUE(alert_manager.hasActiveAlerts());

    // Update thresholds to trigger reset
    alert_manager.setAltitudeThresholds(200.0f, 50.0f);

    // Verify the alert is cleared
    EXPECT_FALSE(alert_manager.hasActiveAlerts());
}

// Test case: 正常输入测试 - 获取活动预警信息
TEST_F(AlertManagerTest, testGetActiveAlertsNormalInput) {
    // Add a simulated active alert
    AlertInfo alert;
    alert.type = AlertType::ALTITUDE_UPPER;
    alert.status = AlertStatus::ACTIVE;
    alert.trigger_time = mock_timestamp;
    alert.description = "High altitude";

    alert_manager.alerts_[AlertType::ALTITUDE_UPPER] = alert;

    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);
    EXPECT_EQ(active_alerts[0].description, "High altitude");
}

// Test case: 边界值测试 - 无活动预警时获取空列表
TEST_F(AlertManagerTest, testGetActiveAlertsEmpty) {
    auto active_alerts = alert_manager.getActiveAlerts();

    EXPECT_TRUE(active_alerts.empty());
}

// Test case: 特殊场景测试 - 多个活动预警同时存在
TEST_F(AlertManagerTest, testGetActiveAlertsMultipleAlerts) {
    // Simulate multiple active alerts
    AlertInfo alert1, alert2;
    alert1.type = AlertType::ALTITUDE_UPPER;
    alert1.status = AlertStatus::ACTIVE;
    alert1.trigger_time = mock_timestamp;
    alert1.description = "High altitude";

    alert2.type = AlertType::LOW_BATTERY;
    alert2.status = AlertStatus::ACTIVE;
    alert2.trigger_time = mock_timestamp + 10;
    alert2.description = "Low battery";

    alert_manager.alerts_[AlertType::ALTITUDE_UPPER] = alert1;
    alert_manager.alerts_[AlertType::LOW_BATTERY] = alert2;

    auto active_alerts = alert_manager.getActiveAlerts();

    EXPECT_EQ(active_alerts.size(), 2);
    EXPECT_EQ(active_alerts[0].type, AlertType::ALTITUDE_UPPER);
    EXPECT_EQ(active_alerts[1].type, AlertType::LOW_BATTERY);
}

// Test case: 特殊场景测试 - 已解决的预警不包含在结果中
TEST_F(AlertManagerTest, testGetActiveAlertsResolvedAlertNotIncluded) {
    AlertInfo alert;
    alert.type = AlertType::ALTITUDE_UPPER;
    alert.status = AlertStatus::RESOLVED;
    alert.trigger_time = mock_timestamp;
    alert.description = "Resolved high altitude";

    alert_manager.alerts_[AlertType::ALTITUDE_UPPER] = alert;

    auto active_alerts = alert_manager.getActiveAlerts();

    EXPECT_TRUE(active_alerts.empty());
}