rho-coding-agent 1.8.2

use crate::tui::{pending_input::AcceptedSteering, tests::test_app, QueuedPrompt};

fn prompt(model: &str, display: &str) -> QueuedPrompt {
    QueuedPrompt {
        prompt: model.into(),
        display_prompt: display.into(),
        paste_segments: Vec::new(),
    }
}

#[test]
fn interrupt_restores_accepted_local_and_follow_up_input() {
    let mut app = test_app();
    app.input = "draft".into();
    app.input_cursor = app.input_char_len();
    app.accepted_steering.push_back(AcceptedSteering {
        id: rho_sdk::SteeringId::new(),
        prompt: prompt("accepted steer", "accepted steer"),
    });
    app.steering_prompts
        .push_back(prompt("local steer", "local steer"));
    app.queued_prompts
        .push_back(prompt("expanded next turn", "next turn"));

    app.restore_pending_work_to_input();

    assert_eq!(
        app.input,
        "accepted steer\n\nlocal steer\n\nexpanded next turn\n\ndraft"
    );
    assert!(app.accepted_steering.is_empty());
    assert!(app.steering_prompts.is_empty());
    assert!(app.queued_prompts.is_empty());
    assert_eq!(app.input_cursor, app.input_char_len());
}

#[test]
fn failed_run_preserves_unapplied_steering_as_follow_ups() {
    let mut app = test_app();
    app.accepted_steering.push_back(AcceptedSteering {
        id: rho_sdk::SteeringId::new(),
        prompt: prompt("accepted model", "accepted display"),
    });
    app.steering_prompts
        .push_back(prompt("local model", "local display"));
    app.queued_prompts
        .push_back(prompt("existing model", "existing display"));

    app.preserve_unapplied_steering_as_follow_ups();

    assert!(app.accepted_steering.is_empty());
    assert!(app.steering_prompts.is_empty());
    assert_eq!(
        app.queued_prompts
            .iter()
            .map(|prompt| prompt.display_prompt.as_str())
            .collect::<Vec<_>>(),
        ["accepted display", "local display", "existing display"]
    );
}

#[test]
fn interrupt_expands_pasted_draft_before_restoring_it() {
    let mut app = test_app();
    app.insert_pasted_input_text("alpha\nbeta");
    app.steering_prompts.push_back(prompt("steer", "steer"));

    app.restore_pending_work_to_input();

    assert_eq!(app.input, "steer\n\nalpha\nbeta");
    assert!(app.paste_segments.is_empty());
}