freedak
f7a720204a
- 移除 GovAI, nomifun-tauri, 算力盒子 的 submodule 引用 - 添加所有子项目的完整源代码 - 保留原始 .git 为 .git.bak 备份
85 lines
2.8 KiB
Rust
85 lines
2.8 KiB
Rust
use nomi_protocol::events::{Capabilities, ProtocolEvent};
|
|
|
|
#[test]
|
|
fn capabilities_serialize_with_all_fields() {
|
|
let caps = Capabilities {
|
|
tool_approval: true,
|
|
thinking: true,
|
|
effort: false,
|
|
effort_levels: vec![],
|
|
modes: vec!["default".into(), "auto_edit".into(), "yolo".into()],
|
|
current_mode: "default".into(),
|
|
mcp: true,
|
|
};
|
|
let event = ProtocolEvent::Ready {
|
|
version: "0.2.0".into(),
|
|
session_id: None,
|
|
capabilities: caps,
|
|
};
|
|
let json = serde_json::to_string(&event).unwrap();
|
|
let parsed: serde_json::Value = serde_json::from_str(&json).unwrap();
|
|
|
|
assert_eq!(parsed["type"], "ready");
|
|
assert_eq!(parsed["capabilities"]["thinking"], true);
|
|
assert_eq!(parsed["capabilities"]["effort"], false);
|
|
assert!(
|
|
parsed["capabilities"]["effort_levels"]
|
|
.as_array()
|
|
.unwrap()
|
|
.is_empty()
|
|
);
|
|
assert_eq!(parsed["capabilities"]["modes"].as_array().unwrap().len(), 3);
|
|
assert_eq!(parsed["capabilities"]["current_mode"], "default");
|
|
}
|
|
|
|
#[test]
|
|
fn config_changed_event_serializes_correctly() {
|
|
let caps = Capabilities {
|
|
tool_approval: true,
|
|
thinking: false,
|
|
effort: true,
|
|
effort_levels: vec!["low".into(), "medium".into(), "high".into()],
|
|
modes: vec!["default".into(), "auto_edit".into(), "yolo".into()],
|
|
current_mode: "default".into(),
|
|
mcp: false,
|
|
};
|
|
let event = ProtocolEvent::ConfigChanged { capabilities: caps };
|
|
let json = serde_json::to_string(&event).unwrap();
|
|
let parsed: serde_json::Value = serde_json::from_str(&json).unwrap();
|
|
|
|
assert_eq!(parsed["type"], "config_changed");
|
|
assert_eq!(parsed["capabilities"]["effort_levels"][1], "medium");
|
|
}
|
|
|
|
#[test]
|
|
fn capabilities_with_effort_levels_roundtrip() {
|
|
let caps = Capabilities {
|
|
tool_approval: true,
|
|
thinking: false,
|
|
effort: true,
|
|
effort_levels: vec!["low".into(), "medium".into(), "high".into()],
|
|
modes: vec!["default".into(), "auto_edit".into(), "yolo".into()],
|
|
current_mode: "default".into(),
|
|
mcp: true,
|
|
};
|
|
let event = ProtocolEvent::Ready {
|
|
version: "0.2.0".into(),
|
|
session_id: Some("test-session".into()),
|
|
capabilities: caps,
|
|
};
|
|
let json = serde_json::to_string(&event).unwrap();
|
|
let parsed: serde_json::Value = serde_json::from_str(&json).unwrap();
|
|
|
|
assert_eq!(parsed["capabilities"]["effort"], true);
|
|
assert_eq!(
|
|
parsed["capabilities"]["effort_levels"]
|
|
.as_array()
|
|
.unwrap()
|
|
.len(),
|
|
3
|
|
);
|
|
assert_eq!(parsed["capabilities"]["effort_levels"][0], "low");
|
|
assert_eq!(parsed["capabilities"]["effort_levels"][2], "high");
|
|
assert_eq!(parsed["session_id"], "test-session");
|
|
}
|