use dust_lang::*;

#[test]
fn true_and_true() {
    let source = "let a = true; let b = true; a && b";

    assert_eq!(
        compile(source),
        Ok(Chunk::with_data(
            None,
            FunctionType {
                type_parameters: None,
                value_parameters: None,
                return_type: Box::new(Type::Boolean),
            },
            vec![
                (
                    Instruction::load_boolean(Destination::Register(0), true, false),
                    Span(8, 12)
                ),
                (Instruction::define_local(0, 0, false), Span(4, 5)),
                (
                    Instruction::load_boolean(Destination::Register(1), true, false),
                    Span(22, 26)
                ),
                (Instruction::define_local(1, 1, false), Span(18, 19)),
                (
                    Instruction::get_local(Destination::Register(2), 0),
                    Span(28, 29)
                ),
                (Instruction::test(Argument::Local(0), true), Span(30, 32)),
                (Instruction::jump(1, true), Span(30, 32)),
                (
                    Instruction::get_local(Destination::Register(3), 1),
                    Span(33, 34)
                ),
                (Instruction::r#return(true), Span(34, 34)),
            ],
            vec![ConcreteValue::string("a"), ConcreteValue::string("b")],
            vec![
                Local::new(0, Type::Boolean, false, Scope::default()),
                Local::new(1, Type::Boolean, false, Scope::default())
            ]
        ))
    );

    assert_eq!(run(source), Ok(Some(ConcreteValue::Boolean(true))));
}

#[test]
fn false_and_false() {
    let source = "let a = false; let b = false; a && b";

    assert_eq!(
        compile(source),
        Ok(Chunk::with_data(
            None,
            FunctionType {
                type_parameters: None,
                value_parameters: None,
                return_type: Box::new(Type::Boolean),
            },
            vec![
                (
                    Instruction::load_boolean(Destination::Register(0), false, false),
                    Span(8, 13)
                ),
                (Instruction::define_local(0, 0, false), Span(4, 5)),
                (
                    Instruction::load_boolean(Destination::Register(1), false, false),
                    Span(23, 28)
                ),
                (Instruction::define_local(1, 1, false), Span(19, 20)),
                (
                    Instruction::get_local(Destination::Register(2), 0),
                    Span(30, 31)
                ),
                (Instruction::test(Argument::Local(0), true), Span(32, 34)),
                (Instruction::jump(1, true), Span(32, 34)),
                (
                    Instruction::get_local(Destination::Register(3), 1),
                    Span(35, 36)
                ),
                (Instruction::r#return(true), Span(36, 36)),
            ],
            vec![ConcreteValue::string("a"), ConcreteValue::string("b")],
            vec![
                Local::new(0, Type::Boolean, false, Scope::default()),
                Local::new(1, Type::Boolean, false, Scope::default())
            ]
        ))
    );

    assert_eq!(run(source), Ok(Some(ConcreteValue::Boolean(false))));
}

#[test]
fn true_and_false() {
    let source = "let a = true; let b = false; a && b";

    assert_eq!(
        compile(source),
        Ok(Chunk::with_data(
            None,
            FunctionType {
                type_parameters: None,
                value_parameters: None,
                return_type: Box::new(Type::Boolean),
            },
            vec![
                (
                    Instruction::load_boolean(Destination::Register(0), true, false),
                    Span(8, 12)
                ),
                (Instruction::define_local(0, 0, false), Span(4, 5)),
                (
                    Instruction::load_boolean(Destination::Register(1), false, false),
                    Span(22, 27)
                ),
                (Instruction::define_local(1, 1, false), Span(18, 19)),
                (
                    Instruction::get_local(Destination::Register(2), 0),
                    Span(29, 30)
                ),
                (Instruction::test(Argument::Local(0), true), Span(31, 33)),
                (Instruction::jump(1, true), Span(31, 33)),
                (
                    Instruction::get_local(Destination::Register(3), 1),
                    Span(34, 35)
                ),
                (Instruction::r#return(true), Span(35, 35)),
            ],
            vec![ConcreteValue::string("a"), ConcreteValue::string("b")],
            vec![
                Local::new(0, Type::Boolean, false, Scope::default()),
                Local::new(1, Type::Boolean, false, Scope::default())
            ]
        ))
    );

    assert_eq!(run(source), Ok(Some(ConcreteValue::Boolean(false))));
}

#[test]
fn false_and_true() {
    let source = "let a = false; let b = true; a && b";

    assert_eq!(
        compile(source),
        Ok(Chunk::with_data(
            None,
            FunctionType {
                type_parameters: None,
                value_parameters: None,
                return_type: Box::new(Type::Boolean),
            },
            vec![
                (
                    Instruction::load_boolean(Destination::Register(0), false, false),
                    Span(8, 13)
                ),
                (Instruction::define_local(0, 0, false), Span(4, 5)),
                (
                    Instruction::load_boolean(Destination::Register(1), true, false),
                    Span(23, 27)
                ),
                (Instruction::define_local(1, 1, false), Span(19, 20)),
                (
                    Instruction::get_local(Destination::Register(2), 0),
                    Span(29, 30)
                ),
                (Instruction::test(Argument::Local(0), true), Span(31, 33)),
                (Instruction::jump(1, true), Span(31, 33)),
                (
                    Instruction::get_local(Destination::Register(3), 1),
                    Span(34, 35)
                ),
                (Instruction::r#return(true), Span(35, 35)),
            ],
            vec![ConcreteValue::string("a"), ConcreteValue::string("b")],
            vec![
                Local::new(0, Type::Boolean, false, Scope::default()),
                Local::new(1, Type::Boolean, false, Scope::default())
            ]
        ))
    );

    assert_eq!(run(source), Ok(Some(ConcreteValue::Boolean(false))));
}

#[test]
fn true_and_true_and_true() {
    let source = "let a = true; let b = true; let c = true; a && b && c";

    assert_eq!(
        compile(source),
        Ok(Chunk::with_data(
            None,
            FunctionType {
                type_parameters: None,
                value_parameters: None,
                return_type: Box::new(Type::Boolean),
            },
            vec![
                (
                    Instruction::load_boolean(Destination::Register(0), true, false),
                    Span(8, 12)
                ),
                (Instruction::define_local(0, 0, false), Span(4, 5)),
                (
                    Instruction::load_boolean(Destination::Register(1), true, false),
                    Span(22, 26)
                ),
                (Instruction::define_local(1, 1, false), Span(18, 19)),
                (
                    Instruction::load_boolean(Destination::Register(2), true, false),
                    Span(36, 40)
                ),
                (Instruction::define_local(2, 2, false), Span(32, 33)),
                (
                    Instruction::get_local(Destination::Register(3), 0),
                    Span(42, 43)
                ),
                (Instruction::test(Argument::Local(0), true), Span(44, 46)),
                (Instruction::jump(1, true), Span(44, 46)),
                (
                    Instruction::get_local(Destination::Register(4), 1),
                    Span(47, 48)
                ),
                (Instruction::test(Argument::Local(1), true), Span(49, 51)),
                (Instruction::jump(1, true), Span(49, 51)),
                (
                    Instruction::get_local(Destination::Register(5), 2),
                    Span(52, 53)
                ),
                (Instruction::r#return(true), Span(53, 53)),
            ],
            vec![
                ConcreteValue::string("a"),
                ConcreteValue::string("b"),
                ConcreteValue::string("c")
            ],
            vec![
                Local::new(0, Type::Boolean, false, Scope::default()),
                Local::new(1, Type::Boolean, false, Scope::default()),
                Local::new(2, Type::Boolean, false, Scope::default())
            ]
        ))
    );

    assert_eq!(run(source), Ok(Some(ConcreteValue::Boolean(true))));
}

#[test]
fn false_and_false_and_false() {
    let source = "let a = false; let b = false; let c = false; a && b && c";

    assert_eq!(
        compile(source),
        Ok(Chunk::with_data(
            None,
            FunctionType {
                type_parameters: None,
                value_parameters: None,
                return_type: Box::new(Type::Boolean),
            },
            vec![
                (
                    Instruction::load_boolean(Destination::Register(0), false, false),
                    Span(8, 13)
                ),
                (Instruction::define_local(0, 0, false), Span(4, 5)),
                (
                    Instruction::load_boolean(Destination::Register(1), false, false),
                    Span(23, 28)
                ),
                (Instruction::define_local(1, 1, false), Span(19, 20)),
                (
                    Instruction::load_boolean(Destination::Register(2), false, false),
                    Span(38, 43)
                ),
                (Instruction::define_local(2, 2, false), Span(34, 35)),
                (
                    Instruction::get_local(Destination::Register(3), 0),
                    Span(45, 46)
                ),
                (Instruction::test(Argument::Local(0), true), Span(47, 49)),
                (Instruction::jump(1, true), Span(47, 49)),
                (
                    Instruction::get_local(Destination::Register(4), 1),
                    Span(50, 51)
                ),
                (Instruction::test(Argument::Local(1), true), Span(52, 54)),
                (Instruction::jump(1, true), Span(52, 54)),
                (
                    Instruction::get_local(Destination::Register(5), 2),
                    Span(55, 56)
                ),
                (Instruction::r#return(true), Span(56, 56)),
            ],
            vec![
                ConcreteValue::string("a"),
                ConcreteValue::string("b"),
                ConcreteValue::string("c")
            ],
            vec![
                Local::new(0, Type::Boolean, false, Scope::default()),
                Local::new(1, Type::Boolean, false, Scope::default()),
                Local::new(2, Type::Boolean, false, Scope::default())
            ]
        ))
    );

    assert_eq!(run(source), Ok(Some(ConcreteValue::Boolean(false))));
}

#[test]
fn true_and_true_or_false() {
    let source = "let a = true; let b = true; let c = false; a && b || c";

    assert_eq!(
        compile(source),
        Ok(Chunk::with_data(
            None,
            FunctionType {
                type_parameters: None,
                value_parameters: None,
                return_type: Box::new(Type::Boolean),
            },
            vec![
                (
                    Instruction::load_boolean(Destination::Register(0), true, false),
                    Span(8, 12)
                ),
                (Instruction::define_local(0, 0, false), Span(4, 5)),
                (
                    Instruction::load_boolean(Destination::Register(1), true, false),
                    Span(22, 26)
                ),
                (Instruction::define_local(1, 1, false), Span(18, 19)),
                (
                    Instruction::load_boolean(Destination::Register(2), false, false),
                    Span(36, 41)
                ),
                (Instruction::define_local(2, 2, false), Span(32, 33)),
                (
                    Instruction::get_local(Destination::Register(3), 0),
                    Span(43, 44)
                ),
                (Instruction::test(Argument::Local(0), true), Span(45, 47)),
                (Instruction::jump(1, true), Span(45, 47)),
                (
                    Instruction::get_local(Destination::Register(4), 1),
                    Span(48, 49)
                ),
                (Instruction::test(Argument::Local(1), false), Span(50, 52)),
                (Instruction::jump(1, true), Span(50, 52)),
                (
                    Instruction::get_local(Destination::Register(5), 2),
                    Span(53, 54)
                ),
                (Instruction::r#return(true), Span(54, 54)),
            ],
            vec![
                ConcreteValue::string("a"),
                ConcreteValue::string("b"),
                ConcreteValue::string("c")
            ],
            vec![
                Local::new(0, Type::Boolean, false, Scope::default()),
                Local::new(1, Type::Boolean, false, Scope::default()),
                Local::new(2, Type::Boolean, false, Scope::default())
            ]
        ))
    );

    assert_eq!(run(source), Ok(Some(ConcreteValue::Boolean(true))));
}