Rebrand to Galaxy, major improvements to Bedrock support, still needs some TLC though

crates/galaxy_cli/src/mcp_tests.rs

@@ -0,0 +1,75 @@
+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(_)));
+}