use super::*;

#[allow(dead_code)]
fn parse_test_input(raw: &str) -> Vec<i64> {
    raw.split(",").map(|x| x.parse().unwrap()).collect()
}

#[test]
fn test_find_max() {
    assert_eq!(
        find_max(
            0..5,
            &parse_test_input("3,15,3,16,1002,16,10,16,1,16,15,15,4,15,99,0,0")
        ),
        Some(43210)
    );
    assert_eq!(
        find_max(
            0..5,
            &parse_test_input(
                "3,23,3,24,1002,24,10,24,1002,23,-1,23,101,5,23,23,1,24,23,23,4,23,99,0,0"
            )
        ),
        Some(54321)
    );
    assert_eq!(
        find_max(
            0..5,
            &parse_test_input("3,31,3,32,1002,32,10,32,1001,31,-2,31,1007,31,0,33,1002,33,7,33,1,33,31,31,1,32,31,31,4,31,99,0,0,0")
        ),
        Some(65210)
    );
}

#[test]
fn test_find_max_with_loops() {
    assert_eq!(
        find_max(
            5..10,
            &parse_test_input("3,26,1001,26,-4,26,3,27,1002,27,2,27,1,27,26,27,4,27,1001,28,-1,28,1005,28,6,99,0,0,5")
        ),
        Some(139629729)
    );
    assert_eq!(
        find_max(
            5..10,
            &parse_test_input("3,52,1001,52,-5,52,3,53,1,52,56,54,1007,54,5,55,1005,55,26,1001,54,-5,54,1105,1,12,1,53,54,53,1008,54,0,55,1001,55,1,55,2,53,55,53,4,53,1001,56,-1,56,1005,56,6,99,0,0,0,0,10")
        ),
        Some(18216)
    );
}

#[test]
fn test_position_less_than() {
    assert_eq!(
        run_for_input(
            &parse_test_input("3,9,8,9,10,9,4,9,99,-1,8"),
            &mut 0,
            vec![8]
        ),
        1
    );
    assert_eq!(
        run_for_input(
            &parse_test_input("3,9,8,9,10,9,4,9,99,-1,8"),
            &mut 0,
            vec![7]
        ),
        0
    );
}

#[test]
fn test_position_equals() {
    assert_eq!(
        run_for_input(
            &parse_test_input("3,9,7,9,10,9,4,9,99,-1,8"),
            &mut 0,
            vec![8]
        ),
        0
    );
    assert_eq!(
        run_for_input(
            &parse_test_input("3,9,7,9,10,9,4,9,99,-1,8"),
            &mut 0,
            vec![7]
        ),
        1
    );
}

#[test]
fn test_immediate_less_than() {
    assert_eq!(
        run_for_input(
            &parse_test_input("3,3,1107,-1,8,3,4,3,99"),
            &mut 0,
            vec![8]
        ),
        0
    );
    assert_eq!(
        run_for_input(
            &parse_test_input("3,3,1107,-1,8,3,4,3,99"),
            &mut 0,
            vec![7]
        ),
        1
    );
}

#[test]
fn test_immediate_equals() {
    assert_eq!(
        run_for_input(
            &parse_test_input("3,3,1108,-1,8,3,4,3,99"),
            &mut 0,
            vec![8]
        ),
        1
    );
    assert_eq!(
        run_for_input(
            &parse_test_input("3,3,1108,-1,8,3,4,3,99"),
            &mut 0,
            vec![7]
        ),
        0
    );
}