use bladeink::{story::Story, story_error::StoryError};

mod common;

#[test]
fn simple_divert_test() -> Result<(), StoryError> {
    let json_string = common::get_json_string("inkfiles/divert/simple-divert.ink.json").unwrap();
    let mut story = Story::new(&json_string)?;

    let mut text: Vec<String> = Vec::new();
    common::next_all(&mut story, &mut text)?;
    assert_eq!(2, text.len());
    assert_eq!("We arrived into London at 9.45pm exactly.", text[0]);
    assert_eq!(
        "We hurried home to Savile Row as fast as we could.",
        text[1]
    );

    Ok(())
}

#[test]
fn invisible_divert_test() -> Result<(), StoryError> {
    let json_string = common::get_json_string("inkfiles/divert/invisible-divert.ink.json").unwrap();
    let mut story = Story::new(&json_string)?;

    let mut text: Vec<String> = Vec::new();
    common::next_all(&mut story, &mut text)?;
    assert_eq!(1, text.len());
    assert_eq!(
        "We hurried home to Savile Row as fast as we could.",
        text[0]
    );

    Ok(())
}

#[test]
fn divert_on_choice_test() -> Result<(), StoryError> {
    let json_string = common::get_json_string("inkfiles/divert/divert-on-choice.ink.json").unwrap();
    let mut story = Story::new(&json_string)?;

    let mut text: Vec<String> = Vec::new();
    common::next_all(&mut story, &mut text)?;
    story.choose_choice_index(0)?;

    text.clear();
    common::next_all(&mut story, &mut text)?;

    assert_eq!(1, text.len());
    assert_eq!("You open the gate, and step out onto the path.", text[0]);

    Ok(())
}

#[test]
fn complex_branching1_test() -> Result<(), StoryError> {
    let json_string =
        common::get_json_string("inkfiles/divert/complex-branching.ink.json").unwrap();
    let mut story = Story::new(&json_string)?;
    let mut text: Vec<String> = Vec::new();

    common::next_all(&mut story, &mut text)?;
    story.choose_choice_index(0)?;

    text.clear();
    common::next_all(&mut story, &mut text)?;

    assert_eq!(2, text.len());
    assert_eq!("\"There is not a moment to lose!\" I declared.", text[0]);
    assert_eq!(
        "We hurried home to Savile Row as fast as we could.",
        text[1]
    );

    Ok(())
}

#[test]
fn complex_branching2_test() -> Result<(), StoryError> {
    let json_string =
        common::get_json_string("inkfiles/divert/complex-branching.ink.json").unwrap();
    let mut story = Story::new(&json_string)?;
    let mut text: Vec<String> = Vec::new();

    common::next_all(&mut story, &mut text)?;
    story.choose_choice_index(1)?;

    text.clear();
    common::next_all(&mut story, &mut text)?;

    assert_eq!(3, text.len());
    assert_eq!(
        "\"Monsieur, let us savour this moment!\" I declared.",
        text[0]
    );
    assert_eq!(
        "My master clouted me firmly around the head and dragged me out of the door.",
        text[1]
    );
    assert_eq!(
        "He insisted that we hurried home to Savile Row as fast as we could.",
        text[2]
    );

    Ok(())
}