320 lines
9.0 KiB
C++
320 lines
9.0 KiB
C++
|
/**
|
|||
|
|
* @file basic_test.cpp
|
|||
|
|
* @brief CMS 核心引擎基本单元测试
|
|||
|
|
*
|
|||
|
|
* 使用标准库 assert 进行验证，不依赖外部测试框架。
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
#include "app.hpp"
|
|||
|
|
#include <cassert>
|
|||
|
|
#include <iostream>
|
|||
|
|
#include <chrono>
|
|||
|
|
#include <string>
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 测试事件接入与处理
|
|||
|
|
*/
|
|||
|
|
static void testEventIngestion()
|
|||
|
|
{
|
|||
|
|
std::cout << "[TEST] testEventIngestion ... ";
|
|||
|
|
|
|||
|
|
CmsEngine engine;
|
|||
|
|
|
|||
|
|
// 空 ID 应被拒绝
|
|||
|
|
EventRecord bad;
|
|||
|
|
bad.id = "";
|
|||
|
|
assert(!engine.ingestEvent(bad));
|
|||
|
|
|
|||
|
|
// 有效事件应被接受
|
|||
|
|
EventRecord evt;
|
|||
|
|
evt.id = "EVT-TEST-001";
|
|||
|
|
evt.typeTag = "test_type";
|
|||
|
|
evt.priority = 100;
|
|||
|
|
evt.threatLevel = 3;
|
|||
|
|
evt.status = EventStatus::Pending;
|
|||
|
|
evt.receiveTime = std::chrono::system_clock::now();
|
|||
|
|
assert(engine.ingestEvent(evt));
|
|||
|
|
|
|||
|
|
// 查找事件
|
|||
|
|
const auto* found = engine.findEventById("EVT-TEST-001");
|
|||
|
|
assert(found != nullptr);
|
|||
|
|
assert(found->typeTag == "test_type");
|
|||
|
|
|
|||
|
|
// 处理待处理事件 → 应生成一个任务方案
|
|||
|
|
size_t count = engine.processPendingEvents();
|
|||
|
|
assert(count == 1);
|
|||
|
|
|
|||
|
|
// 事件状态已变为 Generated
|
|||
|
|
found = engine.findEventById("EVT-TEST-001");
|
|||
|
|
assert(found != nullptr);
|
|||
|
|
assert(found->status == EventStatus::Generated);
|
|||
|
|
|
|||
|
|
// 方案列表应有 1 个方案
|
|||
|
|
assert(engine.getAllPlans().size() == 1);
|
|||
|
|
|
|||
|
|
std::cout << "PASSED\n";
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 测试任务方案管理
|
|||
|
|
*/
|
|||
|
|
static void testPlanManagement()
|
|||
|
|
{
|
|||
|
|
std::cout << "[TEST] testPlanManagement ... ";
|
|||
|
|
|
|||
|
|
CmsEngine engine;
|
|||
|
|
|
|||
|
|
// 空名称应被拒绝
|
|||
|
|
TaskPlan badPlan;
|
|||
|
|
badPlan.name = "";
|
|||
|
|
assert(!engine.createTaskPlan(badPlan));
|
|||
|
|
|
|||
|
|
// 有效方案
|
|||
|
|
TaskPlan plan;
|
|||
|
|
plan.name = "Operation-Storm";
|
|||
|
|
plan.type = PlanType::Distributed;
|
|||
|
|
plan.status = PlanStatus::Drafting;
|
|||
|
|
plan.topologyNodes = {"Node-1", "Node-2", "Node-3"};
|
|||
|
|
plan.resourceQuota = 0.8;
|
|||
|
|
plan.constraints = "max_delay=5s";
|
|||
|
|
assert(engine.createTaskPlan(plan));
|
|||
|
|
|
|||
|
|
// 创建后应有 ID
|
|||
|
|
assert(!plan.id.empty());
|
|||
|
|
|
|||
|
|
// 按 ID 查找
|
|||
|
|
const auto* found = engine.findPlanById(plan.id);
|
|||
|
|
assert(found != nullptr);
|
|||
|
|
assert(found->name == "Operation-Storm");
|
|||
|
|
assert(found->type == PlanType::Distributed);
|
|||
|
|
|
|||
|
|
// 更新状态
|
|||
|
|
assert(engine.updatePlanStatus(plan.id, PlanStatus::Confirmed));
|
|||
|
|
found = engine.findPlanById(plan.id);
|
|||
|
|
assert(found->status == PlanStatus::Confirmed);
|
|||
|
|
|
|||
|
|
// 不存在的 ID 更新应失败
|
|||
|
|
assert(!engine.updatePlanStatus("NONEXIST", PlanStatus::Completed));
|
|||
|
|
|
|||
|
|
std::cout << "PASSED\n";
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 测试模板注册与匹配
|
|||
|
|
*/
|
|||
|
|
static void testTemplateMatching()
|
|||
|
|
{
|
|||
|
|
std::cout << "[TEST] testTemplateMatching ... ";
|
|||
|
|
|
|||
|
|
CmsEngine engine;
|
|||
|
|
|
|||
|
|
// 空库匹配应返回 nullptr
|
|||
|
|
assert(engine.matchTemplate("any") == nullptr);
|
|||
|
|
|
|||
|
|
TemplateInstance tmpl;
|
|||
|
|
tmpl.id = "TMPL-TEST";
|
|||
|
|
tmpl.scenario = "land_assault";
|
|||
|
|
tmpl.complexity = 0.8;
|
|||
|
|
tmpl.version = "1.0.0";
|
|||
|
|
tmpl.confidence = 0.95;
|
|||
|
|
engine.registerTemplate(tmpl);
|
|||
|
|
|
|||
|
|
const auto* matched = engine.matchTemplate("land_assault");
|
|||
|
|
assert(matched != nullptr);
|
|||
|
|
assert(matched->id == "TMPL-TEST");
|
|||
|
|
assert(matched->confidence == 0.95);
|
|||
|
|
|
|||
|
|
std::cout << "PASSED\n";
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 测试执行监控与健康检查
|
|||
|
|
*/
|
|||
|
|
static void testExecutionMonitoring()
|
|||
|
|
{
|
|||
|
|
std::cout << "[TEST] testExecutionMonitoring ... ";
|
|||
|
|
|
|||
|
|
CmsEngine engine;
|
|||
|
|
|
|||
|
|
// 健康节点
|
|||
|
|
ExecutionStatus healthy;
|
|||
|
|
healthy.nodeId = "NODE-H";
|
|||
|
|
healthy.healthIndex = 0.95;
|
|||
|
|
healthy.errorCode = 0;
|
|||
|
|
healthy.lastReport = std::chrono::system_clock::now();
|
|||
|
|
engine.reportExecutionStatus(healthy);
|
|||
|
|
|
|||
|
|
// 异常节点（健康指数低）
|
|||
|
|
ExecutionStatus sick;
|
|||
|
|
sick.nodeId = "NODE-S";
|
|||
|
|
sick.healthIndex = 0.3;
|
|||
|
|
sick.errorCode = 0;
|
|||
|
|
sick.lastReport = std::chrono::system_clock::now();
|
|||
|
|
engine.reportExecutionStatus(sick);
|
|||
|
|
|
|||
|
|
// 异常节点（异常码非零）
|
|||
|
|
ExecutionStatus error;
|
|||
|
|
error.nodeId = "NODE-E";
|
|||
|
|
error.healthIndex = 0.9;
|
|||
|
|
error.errorCode = 0xA001;
|
|||
|
|
error.lastReport = std::chrono::system_clock::now();
|
|||
|
|
engine.reportExecutionStatus(error);
|
|||
|
|
|
|||
|
|
// 异常节点（心跳超时 - 45秒前）
|
|||
|
|
ExecutionStatus timeout;
|
|||
|
|
timeout.nodeId = "NODE-T";
|
|||
|
|
timeout.healthIndex = 0.9;
|
|||
|
|
timeout.errorCode = 0;
|
|||
|
|
timeout.lastReport = std::chrono::system_clock::now()
|
|||
|
|
- std::chrono::seconds(45);
|
|||
|
|
engine.reportExecutionStatus(timeout);
|
|||
|
|
|
|||
|
|
auto unhealthy = engine.checkHealth();
|
|||
|
|
// 应有 3 个异常节点（NODE-S, NODE-E, NODE-T）
|
|||
|
|
assert(unhealthy.size() == 3);
|
|||
|
|
|
|||
|
|
// 验证具体节点
|
|||
|
|
bool foundS = false, foundE = false, foundT = false;
|
|||
|
|
for (const auto& u : unhealthy) {
|
|||
|
|
if (u.nodeId == "NODE-S") foundS = true;
|
|||
|
|
if (u.nodeId == "NODE-E") foundE = true;
|
|||
|
|
if (u.nodeId == "NODE-T") foundT = true;
|
|||
|
|
}
|
|||
|
|
assert(foundS && foundE && foundT);
|
|||
|
|
|
|||
|
|
std::cout << "PASSED\n";
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 测试用户会话与权限
|
|||
|
|
*/
|
|||
|
|
static void testUserSession()
|
|||
|
|
{
|
|||
|
|
std::cout << "[TEST] testUserSession ... ";
|
|||
|
|
|
|||
|
|
CmsEngine engine;
|
|||
|
|
|
|||
|
|
// 空用户 ID 应被拒绝
|
|||
|
|
UserSession bad;
|
|||
|
|
bad.userId = "";
|
|||
|
|
assert(!engine.createSession(bad));
|
|||
|
|
|
|||
|
|
// 创建有效用户
|
|||
|
|
UserSession user;
|
|||
|
|
user.userId = "USER-001";
|
|||
|
|
user.roleCode = "OPERATOR";
|
|||
|
|
user.permissionBitmap = 0x0000000000000003ULL; // 位0+位1
|
|||
|
|
assert(engine.createSession(user));
|
|||
|
|
|
|||
|
|
// 重复创建应失败
|
|||
|
|
assert(!engine.createSession(user));
|
|||
|
|
|
|||
|
|
// 权限校验
|
|||
|
|
assert(engine.checkPermission("USER-001", 0x0000000000000001ULL));
|
|||
|
|
assert(engine.checkPermission("USER-001", 0x0000000000000002ULL));
|
|||
|
|
assert(engine.checkPermission("USER-001", 0x0000000000000003ULL));
|
|||
|
|
assert(!engine.checkPermission("USER-001", 0x0000000000000004ULL));
|
|||
|
|
|
|||
|
|
// 不存在的用户
|
|||
|
|
assert(!engine.checkPermission("GHOST", 0x0000000000000001ULL));
|
|||
|
|
|
|||
|
|
std::cout << "PASSED\n";
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 测试通知管理
|
|||
|
|
*/
|
|||
|
|
static void testNotification()
|
|||
|
|
{
|
|||
|
|
std::cout << "[TEST] testNotification ... ";
|
|||
|
|
|
|||
|
|
CmsEngine engine;
|
|||
|
|
|
|||
|
|
// 初始无未读通知
|
|||
|
|
assert(engine.getUnreadNotifications().empty());
|
|||
|
|
|
|||
|
|
NotificationMessage msg;
|
|||
|
|
msg.id = "NTF-001";
|
|||
|
|
msg.priority = 200;
|
|||
|
|
msg.type = NotificationType::Alert;
|
|||
|
|
msg.summary = "Test alert";
|
|||
|
|
msg.triggerTime = std::chrono::system_clock::now();
|
|||
|
|
msg.isRead = false;
|
|||
|
|
engine.pushNotification(msg);
|
|||
|
|
|
|||
|
|
assert(engine.getUnreadNotifications().size() == 1);
|
|||
|
|
|
|||
|
|
// 推送已读消息
|
|||
|
|
NotificationMessage readMsg;
|
|||
|
|
readMsg.id = "NTF-002";
|
|||
|
|
readMsg.priority = 50;
|
|||
|
|
readMsg.type = NotificationType::Hint;
|
|||
|
|
readMsg.summary = "Read message";
|
|||
|
|
readMsg.triggerTime = std::chrono::system_clock::now();
|
|||
|
|
readMsg.isRead = true;
|
|||
|
|
engine.pushNotification(readMsg);
|
|||
|
|
|
|||
|
|
// 未读数量仍为 1
|
|||
|
|
assert(engine.getUnreadNotifications().size() == 1);
|
|||
|
|
|
|||
|
|
std::cout << "PASSED\n";
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 测试模式切换
|
|||
|
|
*/
|
|||
|
|
static void testModeSwitch()
|
|||
|
|
{
|
|||
|
|
std::cout << "[TEST] testModeSwitch ... ";
|
|||
|
|
|
|||
|
|
CmsEngine engine;
|
|||
|
|
|
|||
|
|
// 初始模式为 Idle
|
|||
|
|
assert(engine.getSystemContext().currentMode == RunMode::Idle);
|
|||
|
|
|
|||
|
|
// 正常切换
|
|||
|
|
assert(engine.switchMode(RunMode::HumanLoop));
|
|||
|
|
assert(engine.getSystemContext().currentMode == RunMode::HumanLoop);
|
|||
|
|
|
|||
|
|
assert(engine.switchMode(RunMode::AutoExec));
|
|||
|
|
assert(engine.getSystemContext().currentMode == RunMode::AutoExec);
|
|||
|
|
|
|||
|
|
// 进入降级模式
|
|||
|
|
assert(engine.switchMode(RunMode::Degraded));
|
|||
|
|
assert(engine.getSystemContext().currentMode == RunMode::Degraded);
|
|||
|
|
|
|||
|
|
// 降级模式下不允许切换回其他模式（安全策略）
|
|||
|
|
assert(!engine.switchMode(RunMode::Idle));
|
|||
|
|
assert(!engine.switchMode(RunMode::HumanLoop));
|
|||
|
|
|
|||
|
|
std::cout << "PASSED\n";
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 主测试入口
|
|||
|
|
*
|
|||
|
|
* 依次运行所有测试用例，汇总测试结果。
|
|||
|
|
*
|
|||
|
|
* @return int 退出码（0 全部通过，1 有失败）
|
|||
|
|
*/
|
|||
|
|
int main()
|
|||
|
|
{
|
|||
|
|
std::cout << "╔══════════════════════════════════════════════╗\n";
|
|||
|
|
std::cout << "║ CMS 单元测试 ║\n";
|
|||
|
|
std::cout << "╚══════════════════════════════════════════════╝\n\n";
|
|||
|
|
|
|||
|
|
testEventIngestion();
|
|||
|
|
testPlanManagement();
|
|||
|
|
testTemplateMatching();
|
|||
|
|
testExecutionMonitoring();
|
|||
|
|
testUserSession();
|
|||
|
|
testNotification();
|
|||
|
|
testModeSwitch();
|
|||
|
|
|
|||
|
|
std::cout << "\n╔══════════════════════════════════════════════╗\n";
|
|||
|
|
std::cout << "║ 全部测试通过 (ALL TESTS PASSED) ║\n";
|
|||
|
|
std::cout << "╚══════════════════════════════════════════════╝\n";
|
|||
|
|
|
|||
|
|
return 0;
|
|||
|
|
}
|