Search in sources :

Example 1 with NamedParameterNode

use of org.kie.dmn.feel.lang.ast.NamedParameterNode in project drools by kiegroup.

the class FEELParserTest method testFunctionDecisionTableInvocation.

@Test
public void testFunctionDecisionTableInvocation() {
    String inputExpression = "decision table( " + "    outputs: \"Applicant Risk Rating\"," + "    input expression list: [\"Applicant Age\", \"Medical History\"]," + "    rule list: [" + "        [ >60      , \"good\" , \"Medium\" ]," + "        [ >60      , \"bad\"  , \"High\"   ]," + "        [ [25..60] , -        , \"Medium\" ]," + "        [ <25      , \"good\" , \"Low\"    ]," + "        [ <25      , \"bad\"  , \"Medium\" ] ]," + "    hit policy: \"Unique\" )";
    // need to call parse passing in the input variables
    BaseNode functionBase = parse(inputExpression);
    assertThat(functionBase, is(instanceOf(FunctionInvocationNode.class)));
    assertThat(functionBase.getText(), is(inputExpression));
    FunctionInvocationNode function = (FunctionInvocationNode) functionBase;
    assertThat(function.getName(), is(instanceOf(NameRefNode.class)));
    assertThat(function.getName().getText(), is("decision table"));
    assertThat(function.getParams(), is(instanceOf(ListNode.class)));
    assertThat(function.getParams().getElements().size(), is(4));
    assertThat(function.getParams().getElements().get(0), is(instanceOf(NamedParameterNode.class)));
    assertThat(function.getParams().getElements().get(1), is(instanceOf(NamedParameterNode.class)));
    assertThat(function.getParams().getElements().get(2), is(instanceOf(NamedParameterNode.class)));
    assertThat(function.getParams().getElements().get(3), is(instanceOf(NamedParameterNode.class)));
    NamedParameterNode named = (NamedParameterNode) function.getParams().getElements().get(0);
    assertThat(named.getText(), is("outputs: \"Applicant Risk Rating\""));
    assertThat(named.getName().getText(), is("outputs"));
    assertThat(named.getExpression(), is(instanceOf(StringNode.class)));
    assertThat(named.getExpression().getText(), is("\"Applicant Risk Rating\""));
    named = (NamedParameterNode) function.getParams().getElements().get(1);
    assertThat(named.getName().getText(), is("input expression list"));
    assertThat(named.getExpression(), is(instanceOf(ListNode.class)));
    ListNode list = (ListNode) named.getExpression();
    assertThat(list.getElements().size(), is(2));
    assertThat(list.getElements().get(0), is(instanceOf(StringNode.class)));
    assertThat(list.getElements().get(0).getText(), is("\"Applicant Age\""));
    assertThat(list.getElements().get(1), is(instanceOf(StringNode.class)));
    assertThat(list.getElements().get(1).getText(), is("\"Medical History\""));
    named = (NamedParameterNode) function.getParams().getElements().get(2);
    assertThat(named.getName().getText(), is("rule list"));
    assertThat(named.getExpression(), is(instanceOf(ListNode.class)));
    list = (ListNode) named.getExpression();
    assertThat(list.getElements().size(), is(5));
    assertThat(list.getElements().get(0), is(instanceOf(ListNode.class)));
    ListNode rule = (ListNode) list.getElements().get(0);
    assertThat(rule.getElements().size(), is(3));
    assertThat(rule.getElements().get(0), is(instanceOf(UnaryTestNode.class)));
    assertThat(rule.getElements().get(0).getText(), is(">60"));
    assertThat(rule.getElements().get(1), is(instanceOf(StringNode.class)));
    assertThat(rule.getElements().get(1).getText(), is("\"good\""));
    assertThat(rule.getElements().get(2), is(instanceOf(StringNode.class)));
    assertThat(rule.getElements().get(2).getText(), is("\"Medium\""));
    named = (NamedParameterNode) function.getParams().getElements().get(3);
    assertThat(named.getName().getText(), is("hit policy"));
    assertThat(named.getExpression(), is(instanceOf(StringNode.class)));
    assertThat(named.getExpression().getText(), is("\"Unique\""));
}
Also used : BaseNode(org.kie.dmn.feel.lang.ast.BaseNode) NamedParameterNode(org.kie.dmn.feel.lang.ast.NamedParameterNode) ListNode(org.kie.dmn.feel.lang.ast.ListNode) FunctionInvocationNode(org.kie.dmn.feel.lang.ast.FunctionInvocationNode) Test(org.junit.Test)

Example 2 with NamedParameterNode

use of org.kie.dmn.feel.lang.ast.NamedParameterNode in project drools by kiegroup.

the class FEELParserTest method testFunctionInvocationNamedParams.

@Test
public void testFunctionInvocationNamedParams() {
    String inputExpression = "my.test.Function( named parameter 1 : x+10, named parameter 2 : \"foo\" )";
    BaseNode functionBase = parse(inputExpression);
    assertThat(functionBase, is(instanceOf(FunctionInvocationNode.class)));
    assertThat(functionBase.getText(), is(inputExpression));
    FunctionInvocationNode function = (FunctionInvocationNode) functionBase;
    assertThat(function.getName(), is(instanceOf(QualifiedNameNode.class)));
    assertThat(function.getName().getText(), is("my.test.Function"));
    assertThat(function.getParams(), is(instanceOf(ListNode.class)));
    assertThat(function.getParams().getElements().size(), is(2));
    assertThat(function.getParams().getElements().get(0), is(instanceOf(NamedParameterNode.class)));
    assertThat(function.getParams().getElements().get(1), is(instanceOf(NamedParameterNode.class)));
    NamedParameterNode named = (NamedParameterNode) function.getParams().getElements().get(0);
    assertThat(named.getText(), is("named parameter 1 : x+10"));
    assertThat(named.getName().getText(), is("named parameter 1"));
    assertThat(named.getExpression(), is(instanceOf(InfixOpNode.class)));
    assertThat(named.getExpression().getText(), is("x+10"));
    named = (NamedParameterNode) function.getParams().getElements().get(1);
    assertThat(named.getText(), is("named parameter 2 : \"foo\""));
    assertThat(named.getName().getText(), is("named parameter 2"));
    assertThat(named.getExpression(), is(instanceOf(StringNode.class)));
    assertThat(named.getExpression().getText(), is("\"foo\""));
}
Also used : BaseNode(org.kie.dmn.feel.lang.ast.BaseNode) NamedParameterNode(org.kie.dmn.feel.lang.ast.NamedParameterNode) FunctionInvocationNode(org.kie.dmn.feel.lang.ast.FunctionInvocationNode) Test(org.junit.Test)

Aggregations

Test (org.junit.Test)2 BaseNode (org.kie.dmn.feel.lang.ast.BaseNode)2 FunctionInvocationNode (org.kie.dmn.feel.lang.ast.FunctionInvocationNode)2 NamedParameterNode (org.kie.dmn.feel.lang.ast.NamedParameterNode)2 ListNode (org.kie.dmn.feel.lang.ast.ListNode)1