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

use indoc::indoc;
use pretty_assertions::assert_eq;

#[test]
fn test_yield() -> anyhow::Result<()> {
    let compiled = compile! {
        check
        "
        yield();
        "
    };

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

    assert_eq!(
        compiled.output,
        indoc! {
            "
            j main
            main:
            yield
            "
        }
    );

    Ok(())
}

#[test]
fn test_sleep() -> anyhow::Result<()> {
    let compiled = compile! {
        check
        "
        sleep(3);
        let sleepAmount = 15;
        sleep(sleepAmount);
        sleep(sleepAmount * 2);
        "
    };

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

    assert_eq!(
        compiled.output,
        indoc! {
            "
            j main
            main:
            sleep 3
            move r8 15
            sleep r8
            mul r1 r8 2
            sleep r1
            "
        }
    );

    Ok(())
}

#[test]
fn test_set_on_device() -> anyhow::Result<()> {
    let compiled = compile! {
        check
        r#"
            device airConditioner = "d0";
            let internalTemp = 20c;

            set(airConditioner, "On", internalTemp > 25c);
        "#
    };

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

    assert_eq!(
        compiled.output,
        indoc! {
            "
            j main
            main:
            move r8 293.15
            sgt r1 r8 298.15
            s d0 On r1
            "
        }
    );

    Ok(())
}

#[test]
fn test_set_on_device_batched() -> anyhow::Result<()> {
    let compiled = compile! {
        check
        r#"
        const doorHash = hash("Door");
        setBatched(doorHash, "Lock", true);
        "#
    };

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

    assert_eq!(
        compiled.output,
        indoc! {
            r#"
            j main
            main:
            sb 718797587 Lock 1
            "#
        }
    );
    Ok(())
}

#[test]
fn test_set_on_device_batched_named() -> anyhow::Result<()> {
    let compiled = compile! {
        check
        r#"
        device dev = "d0";
        const devName = hash("test");

        sbn(dev, devName, "On", 12);
        "#
    };

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

    assert_eq!(
        compiled.output,
        indoc! {
            "
            j main
            main:
            sbn d0 -662733300 On 12
            "
        }
    );

    Ok(())
}

#[test]
fn test_load_from_device() -> anyhow::Result<()> {
    let compiled = compile! {
        check
        r#"
        device airCon = "d0";

        let setting = load(airCon, "On");
        "#
    };

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

    assert_eq!(
        compiled.output,
        indoc! {
            "
            j main
            main:
            l r15 d0 On
            move r8 r15
            "
        }
    );

    Ok(())
}

#[test]
fn test_load_from_slot() -> anyhow::Result<()> {
    let compiled = compile! {
        check
        r#"
        device airCon = "d0";

        let setting = ls(airCon, 0, "Occupied");
        "#
    };

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

    assert_eq!(
        compiled.output,
        indoc! {
            "
            j main
            main:
            ls r15 d0 0 Occupied
            move r8 r15
            "
        }
    );

    Ok(())
}

#[test]
fn test_set_slot() -> anyhow::Result<()> {
    let compiled = compile! {
        check
        r#"
        device airCon = "d0";

        ss(airCon, 0, "Occupied", true);
        "#
    };

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

    assert_eq!(
        compiled.output,
        indoc! {
            "
            j main
            main:
            ss d0 0 Occupied 1
            "
        }
    );

    Ok(())
}

#[test]
fn test_load_reagent() -> anyhow::Result<()> {
    let compiled = compile! {
        check
        r#"
        device thingy = "d0";

        let something = lr(thingy, "Contents", hash("Iron"));
        "#
    };

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

    assert_eq!(
        compiled.output,
        indoc! {
            "
            j main
            main:
            lr r15 d0 Contents -666742878
            move r8 r15
            "
        }
    );

    Ok(())
}
#[test]
fn test_clr() -> anyhow::Result<()> {
    let compiled = compile! {
        check
        "
        device stackDevice = \"d0\";
        clr(stackDevice);
        let deviceRef = 5;
        clr(deviceRef);
        "
    };

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

    assert_eq!(
        compiled.output,
        indoc! {
            "
            j main
            main:
            clr d0
            move r8 5
            clr r8
            "
        }
    );

    Ok(())
}
#[test]
fn test_rmap() -> anyhow::Result<()> {
    let compiled = compile! {
        check
        "
        device printer = \"d0\";
        let reagentHash = 12345;
        let itemHash = rmap(printer, reagentHash);
        "
    };

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

    assert_eq!(
        compiled.output,
        indoc! {
            "
            j main
            main:
            move r8 12345
            rmap r15 d0 r8
            move r9 r15
            "
        }
    );

    Ok(())
}