76 lines
2.4 KiB
Rust
76 lines
2.4 KiB
Rust
use super::*;
|
|
use clap::builder::TypedValueParser;
|
|
use std::ffi::OsStr;
|
|
|
|
fn parse_mcp_spec(value: &str) -> Result<MCPSpec, clap::Error> {
|
|
let cmd = clap::Command::new("test");
|
|
let parser = MCPSpecParser;
|
|
parser.parse_ref(&cmd, None, OsStr::new(value))
|
|
}
|
|
|
|
#[test]
|
|
fn test_parse_uuid() {
|
|
let uuid_str = "550e8400-e29b-41d4-a716-446655440000";
|
|
let result = parse_mcp_spec(uuid_str).unwrap();
|
|
match result {
|
|
MCPSpec::Uuid(uuid) => assert_eq!(uuid.to_string(), uuid_str),
|
|
MCPSpec::Json(_) => panic!("Expected Uuid variant"),
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_parse_inline_json_cli_server() {
|
|
let json = r#"{"server-name": {"command": "npx", "args": ["-y", "mcp-server"]}}"#;
|
|
let result = parse_mcp_spec(json).unwrap();
|
|
match result {
|
|
MCPSpec::Json(s) => assert_eq!(s, json),
|
|
MCPSpec::Uuid(_) => panic!("Expected Json variant"),
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_parse_inline_json_single_server() {
|
|
let json = r#"{"command": "npx", "args": ["-y", "mcp-server"]}"#;
|
|
let result = parse_mcp_spec(json).unwrap();
|
|
match result {
|
|
MCPSpec::Json(s) => assert_eq!(s, json),
|
|
MCPSpec::Uuid(_) => panic!("Expected Json variant"),
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_parse_inline_json_sse_server() {
|
|
let json = r#"{"url": "http://localhost:3000/mcp", "headers": {"API_KEY": "value"}}"#;
|
|
let result = parse_mcp_spec(json).unwrap();
|
|
match result {
|
|
MCPSpec::Json(s) => assert_eq!(s, json),
|
|
MCPSpec::Uuid(_) => panic!("Expected Json variant"),
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_parse_inline_json_mcp_servers_wrapper() {
|
|
let json = r#"{"mcpServers": {"server-name": {"command": "npx", "args": []}}}"#;
|
|
let result = parse_mcp_spec(json).unwrap();
|
|
match result {
|
|
MCPSpec::Json(s) => assert_eq!(s, json),
|
|
MCPSpec::Uuid(_) => panic!("Expected Json variant"),
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_uuid_takes_precedence_over_json() {
|
|
// A valid UUID should be parsed as UUID, not as JSON
|
|
let uuid_str = "550e8400-e29b-41d4-a716-446655440000";
|
|
let result = parse_mcp_spec(uuid_str).unwrap();
|
|
assert!(matches!(result, MCPSpec::Uuid(_)));
|
|
}
|
|
|
|
#[test]
|
|
fn test_invalid_uuid_treated_as_json() {
|
|
// An invalid UUID that looks like it could be one should be treated as JSON
|
|
let invalid_uuid = "not-a-valid-uuid";
|
|
let result = parse_mcp_spec(invalid_uuid).unwrap();
|
|
assert!(matches!(result, MCPSpec::Json(_)));
|
|
}
|