src/parser/tests/container_parser_tests.rs - Peter - George_Smith/RuSTy

src/parser/tests/container_parser_tests.rs
160 строк5 KB
Volkan
refactor: Rename `units` field to `pous` (#1438)
25 мар 2025, 17:53
25 мар 2025, 17:53f7ddafb
use crate::test_utils::tests::parse;
use insta::assert_debug_snapshot;
use pretty_assertions::*;

#[test]
fn action_container_parsed() {
    let src = "ACTIONS foo ACTION bar END_ACTION END_ACTIONS";
    let result = parse(src).0;

    let prg = &result.implementations[0];
    assert_eq!(prg.name, "foo.bar");
    assert_eq!(prg.type_name, "foo");
}

#[test]
fn two_action_containers_parsed() {
    let src = "ACTIONS foo ACTION bar END_ACTION ACTION buz END_ACTION END_ACTIONS";
    let result = parse(src).0;

    let prg = &result.implementations[0];
    assert_eq!(prg.name, "foo.bar");
    assert_eq!(prg.type_name, "foo");

    let prg2 = &result.implementations[1];
    assert_eq!(prg2.name, "foo.buz");
    assert_eq!(prg2.type_name, "foo");
}

#[test]
fn mixed_action_types_parsed() {
    let src =
        "PROGRAM foo END_PROGRAM ACTIONS foo ACTION bar END_ACTION END_ACTIONS ACTION foo.buz END_ACTION";
    let result = parse(src).0;

    let prg = &result.implementations[1];
    assert_eq!(prg.name, "foo.bar");
    assert_eq!(prg.type_name, "foo");

    let prg2 = &result.implementations[2];
    assert_eq!(prg2.name, "foo.buz");
    assert_eq!(prg2.type_name, "foo");
}

#[test]
fn actions_with_no_container_have_unknown_container() {
    let src = "ACTIONS ACTION bar END_ACTION END_ACTIONS";
    let (result, diagnostic) = parse(src);
    let prg = &result.implementations[0];
    assert_eq!(prg.name, "__unknown__.bar");
    assert_eq!(prg.type_name, "__unknown__");

    //Expect no diagnostic, actions container name is optional
    assert_eq!(diagnostic, []);
}

#[test]
fn actions_with_no_container_inherits_previous_pou() {
    let src = "PROGRAM buz END_PROGRAM PROGRAM foo END_PROGRAM ACTIONS ACTION bar END_ACTION END_ACTIONS";
    let (result, diagnostic) = parse(src);
    let prg = &result.implementations[0];
    assert_eq!(prg.name, "buz");
    assert_eq!(prg.type_name, "buz");

    let prg = &result.implementations[1];
    assert_eq!(prg.name, "foo");
    assert_eq!(prg.type_name, "foo");

    let prg = &result.implementations[2];
    assert_eq!(prg.name, "foo.bar");
    assert_eq!(prg.type_name, "foo");

    //Expect no diagnostic, actions container name is optional
    assert_eq!(diagnostic, []);
}

#[test]
fn actions_with_no_container_inherits_previous_pou_excluding_methods_and_actions() {
    let src = "PROGRAM buz END_PROGRAM PROGRAM foo METHOD x END_METHOD END_PROGRAM ACTIONS ACTION bar END_ACTION END_ACTIONS ACTIONS ACTION baz END_ACTION END_ACTIONS";
    let (result, diagnostic) = parse(src);
    let prg = &result.implementations[0];
    assert_eq!(prg.name, "buz");
    assert_eq!(prg.type_name, "buz");

    let prg = &result.implementations[2];
    assert_eq!(prg.name, "foo");
    assert_eq!(prg.type_name, "foo");

    let prg = &result.implementations[3];
    assert_eq!(prg.name, "foo.bar");
    assert_eq!(prg.type_name, "foo");

    let prg = &result.implementations[4];
    assert_eq!(prg.name, "foo.baz");
    assert_eq!(prg.type_name, "foo");

    //Expect no diagnostic, actions container name is optional
    assert_eq!(diagnostic, []);
}

#[test]
fn actions_with_invalid_token() {
    let src = "ACTIONS LIMA BRAVO END_ACTIONS";
    let errors = parse(src).1;
    assert_debug_snapshot!(errors.first().unwrap());
}

#[test]
fn two_programs_can_be_parsed() {
    let src = "PROGRAM foo END_PROGRAM  PROGRAM bar END_PROGRAM";
    let result = parse(src).0;

    let prg = &result.pous[0];
    assert_eq!(prg.name, "foo");
    let prg2 = &result.pous[1];
    assert_eq!(prg2.name, "bar");
}

#[test]
fn simple_program_with_varblock_can_be_parsed() {
    let src = "PROGRAM buz VAR END_VAR END_PROGRAM";
    let result = parse(src).0;

    let prg = &result.pous[0];

    assert_eq!(prg.variable_blocks.len(), 1);
}

#[test]
fn simple_program_with_two_varblocks_can_be_parsed() {
    let src = "PROGRAM buz VAR END_VAR VAR END_VAR END_PROGRAM";
    let result = parse(src).0;

    let prg = &result.pous[0];

    assert_eq!(prg.variable_blocks.len(), 2);
}

#[test]
fn single_action_parsed() {
    let src = "ACTION foo.bar END_ACTION";
    let result = parse(src).0;

    let prg = &result.implementations[0];
    assert_eq!(prg.name, "foo.bar");
    assert_eq!(prg.type_name, "foo");
}

#[test]
fn two_actions_parsed() {
    let src = "ACTION foo.bar END_ACTION ACTION fuz.bar END_ACTION";
    let result = parse(src).0;

    let prg = &result.implementations[0];
    assert_eq!(prg.name, "foo.bar");
    assert_eq!(prg.type_name, "foo");

    let prg2 = &result.implementations[1];
    assert_eq!(prg2.name, "fuz.bar");
    assert_eq!(prg2.type_name, "fuz");
}