stationeers_lang/rust_compiler/libs/compiler/src/test/binary_expression.rs

use anyhow::Result;
use indoc::indoc;
use pretty_assertions::assert_eq;

#[test]
fn simple_binary_expression() -> Result<()> {
    let result = compile! {
        check
        "
        let i = 1 + 2;
        "
    };

    assert!(
        result.errors.is_empty(),
        "Expected no errors, got: {:?}",
        result.errors
    );

    assert_eq!(
        result.output,
        indoc! {
            "
            j main
            main:
            move r8 3
            "
        }
    );

    Ok(())
}

#[test]
fn nested_binary_expressions() -> Result<()> {
    let result = compile! {
        check
        "
        fn calculateArgs(arg1, arg2, arg3) {
            return (arg1 + arg2) * arg3;
        };

        let returned = calculateArgs(10, 20, 30) + 100;
        "
    };

    assert!(
        result.errors.is_empty(),
        "Expected no errors, got: {:?}",
        result.errors
    );

    assert_eq!(
        result.output,
        indoc! {
            "
            j main
            calculateArgs:
            pop r8
            pop r9
            pop r10
            push sp
            push ra
            add r1 r10 r9
            mul r2 r1 r8
            move r15 r2
            j __internal_L1
            __internal_L1:
            pop ra
            pop sp
            j ra
            main:
            push 10
            push 20
            push 30
            jal calculateArgs
            move r1 r15
            add r2 r1 100
            move r8 r2
            "
        }
    );

    Ok(())
}

#[test]
fn stress_test_constant_folding() -> Result<()> {
    let result = compile! {
        check
        "
        let negationHell = (-1 + -2) * (-3 + (-4 * (-5 + -6)));
        "
    };

    assert!(
        result.errors.is_empty(),
        "Expected no errors, got: {:?}",
        result.errors
    );

    assert_eq!(
        result.output,
        indoc! {
            "
            j main
            main:
            move r8 -123
            "
        }
    );

    Ok(())
}

#[test]
fn test_constant_folding_with_variables_mixed_in() -> Result<()> {
    let result = compile! {
        check
        r#"
        device self = "db";
        let i = 1 - 3 * (1 + 123.4) * self.Setting + 245c;
        "#
    };

    assert!(
        result.errors.is_empty(),
        "Expected no errors, got: {:?}",
        result.errors
    );

    assert_eq!(
        result.output,
        indoc! {
            "
            j main
            main:
            l r1 db Setting
            mul r2 373.2 r1
            sub r3 1 r2
            add r4 r3 518.15
            move r8 r4
            "
        }
    );

    Ok(())
}

#[test]
fn test_ternary_expression() -> Result<()> {
    let result = compile! {
        check
        r#"
        let i = 1 > 2 ? 15 : 20;
        "#
    };

    assert!(
        result.errors.is_empty(),
        "Expected no errors, got: {:?}",
        result.errors
    );

    assert_eq!(
        result.output,
        indoc! {
            "
            j main
            main:
            sgt r1 1 2
            select r2 r1 15 20
            move r8 r2
            "
        }
    );

    Ok(())
}

#[test]
fn test_ternary_expression_assignment() -> Result<()> {
    let result = compile! {
        check
        r#"
        let i = 0;
        i = 1 > 2 ? 15 : 20;
        "#
    };

    assert!(
        result.errors.is_empty(),
        "Expected no errors, got: {:?}",
        result.errors
    );

    assert_eq!(
        result.output,
        indoc! {
            "
            j main
            main:
            move r8 0
            sgt r1 1 2
            select r2 r1 15 20
            move r8 r2
            "
        }
    );

    Ok(())
}

#[test]
fn test_negative_literals() -> Result<()> {
    let result = compile!(
    check
    r#"
        let item = -10c - 20c;
    "#
    );

    assert!(
        result.errors.is_empty(),
        "Expected no errors, got: {:?}",
        result.errors
    );

    assert_eq!(
        result.output,
        indoc! {
            "
            j main
            main:
            move r8 243.15
            "
        }
    );

    Ok(())
}

#[test]
fn test_mismatched_temperature_literals() -> Result<()> {
    let result = compile!(
    check
    r#"
        let item = -10c - 100k;
        let item2 = item + 500c;
    "#
    );

    assert!(
        result.errors.is_empty(),
        "Expected no errors, got: {:?}",
        result.errors
    );

    assert_eq!(
        result.output,
        indoc! {
            "
            j main
            main:
            move r8 163.15
            add r1 r8 773.15
            move r9 r1
            "
        }
    );

    Ok(())
}