openpencil/crates/op-cli/src/cli_design_tests.rs

use super::*;

#[test]
fn parse_args_maps_ts_design_dsl_flags_to_operations_arg() {
    let dsl = r#"root=I(null, {"type":"frame","name":"Page","width":320,"height":240})"#;
    let p = parse_args(&[
        "design".to_string(),
        dsl.to_string(),
        "--canvas-width".to_string(),
        "375".to_string(),
        "--page".to_string(),
        "page-2".to_string(),
    ])
    .expect("parse design dsl");

    assert_eq!(
        p.command,
        Command::ToolCall {
            tool: "batch_design".to_string(),
            args: vec![
                ("operations".to_string(), dsl.to_string()),
                ("postProcess".to_string(), "true".to_string()),
                ("canvasWidth".to_string(), "375".to_string()),
                ("pageId".to_string(), "page-2".to_string()),
            ],
        }
    );
}

#[test]
fn parse_args_keeps_json_array_design_as_nodes_json() {
    let nodes = r#"[{"kind":"rect","name":"A","x":0,"y":0,"width":10,"height":10}]"#;
    let p = parse_args(&["design".to_string(), nodes.to_string()]).expect("parse design array");

    assert_eq!(
        p.command,
        Command::ToolCall {
            tool: "batch_design".to_string(),
            args: vec![
                ("nodes_json".to_string(), nodes.to_string()),
                ("postProcess".to_string(), "true".to_string()),
            ],
        }
    );
}

#[test]
fn parse_args_maps_design_content_to_section_children_args() {
    let payload = r#"{"children":[{"type":"text","name":"Title","content":"Hello","width":120,"height":24}]}"#;
    let p = parse_args(&[
        "design:content".to_string(),
        "section-1".to_string(),
        payload.to_string(),
        "--canvas-width".to_string(),
        "375".to_string(),
        "--page".to_string(),
        "page-2".to_string(),
    ])
    .expect("parse design content");

    match p.command {
        Command::ToolCallJson { tool, args_json } => {
            assert_eq!(tool, "design_content");
            assert_eq!(
                serde_json::from_str::<serde_json::Value>(&args_json).expect("args json"),
                serde_json::json!({
                    "sectionId": "section-1",
                    "children": [
                        {
                            "type": "text",
                            "name": "Title",
                            "content": "Hello",
                            "width": 120,
                            "height": 24,
                        },
                    ],
                    "canvasWidth": 375,
                    "pageId": "page-2",
                })
            );
        }
        other => panic!("expected structured design_content call, got {other:?}"),
    }
}

#[test]
fn parse_args_maps_design_skeleton_to_root_frame_sections_args() {
    let payload = r##"{"rootFrame":{"name":"Login","width":375,"height":812,"fill":[{"type":"solid","color":"#FFFFFF"}]},"sections":[{"name":"Hero","height":240,"padding":[24,24],"role":"hero"}]}"##;
    let p = parse_args(&[
        "design:skeleton".to_string(),
        payload.to_string(),
        "--canvas-width".to_string(),
        "375".to_string(),
        "--page".to_string(),
        "page-2".to_string(),
    ])
    .expect("parse design skeleton");

    match p.command {
        Command::ToolCallJson { tool, args_json } => {
            assert_eq!(tool, "design_skeleton");
            assert_eq!(
                serde_json::from_str::<serde_json::Value>(&args_json).expect("args json"),
                serde_json::json!({
                    "rootFrame": {
                        "name": "Login",
                        "width": 375,
                        "height": 812,
                        "fill": [{"type": "solid", "color": "#FFFFFF"}],
                    },
                    "sections": [
                        {
                            "name": "Hero",
                            "height": 240,
                            "padding": [24, 24],
                            "role": "hero",
                        },
                    ],
                    "canvasWidth": 375,
                    "pageId": "page-2",
                })
            );
        }
        other => panic!("expected structured design_skeleton call, got {other:?}"),
    }
}

#[test]
fn parse_args_maps_design_refine_root_id_args() {
    let p = parse_args(&[
        "design:refine".to_string(),
        "--root-id".to_string(),
        "root-1".to_string(),
        "--canvas-width".to_string(),
        "375".to_string(),
        "--page".to_string(),
        "page-2".to_string(),
    ])
    .expect("parse design refine");

    assert_eq!(
        p.command,
        Command::ToolCall {
            tool: "design_refine".to_string(),
            args: vec![
                ("rootId".to_string(), "root-1".to_string()),
                ("canvasWidth".to_string(), "375".to_string()),
                ("pageId".to_string(), "page-2".to_string()),
            ],
        }
    );
}