core/src/test/java/org/apache/brooklyn/core/workflow/WorkflowOperandsTest.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
package org.apache.brooklyn.core.workflow;

import org.apache.brooklyn.api.entity.EntityLocal;
import org.apache.brooklyn.api.entity.EntitySpec;
import org.apache.brooklyn.core.test.BrooklynMgmtUnitTestSupport;
import org.apache.brooklyn.entity.stock.BasicApplication;
import org.apache.brooklyn.test.Asserts;
import org.apache.brooklyn.util.collections.MutableList;
import org.apache.brooklyn.util.core.config.ConfigBag;
import org.apache.commons.lang3.tuple.Pair;
import org.testng.annotations.Test;

import java.util.List;

public class WorkflowOperandsTest extends BrooklynMgmtUnitTestSupport {

    private BasicApplication app;

    Object runSteps(List<?> steps) {
        Pair<BasicApplication, Object> result = WorkflowBasicTest.runStepsInNewApp(mgmt(), steps);
        this.app = result.getLeft();
        return result.getRight();
    }

    public Object evaluate(String expression, String type) {
        return runSteps(MutableList.of(
                "let "+(type!=null ? type+" " : "") + "result = "+expression,
                "return ${result}"
        ));
    }

    public void assertEvaluated(String expression, Object value) {
        Asserts.assertEquals(evaluate(expression, null), value);
    }

    @Test
    public void testBooleanTrue() {
        Asserts.assertEquals(evaluate("true", "boolean"), true);
    }

    @Test
    public void testBooleanFalse() {
        Object result = runSteps(MutableList.of(
                "let boolean foo = false",
                "return ${foo}"
        ));
        Asserts.assertEquals(result, false);
    }

    @Test
    public void testBooleanAndTrue() {
        Object result = runSteps(MutableList.of(
                "let boolean pass = true && true",
                "return ${pass}"
        ));
        Asserts.assertEquals(result, true);
    }

    @Test
    public void testBooleanAndFalse() {
        Object result = runSteps(MutableList.of(
                "let boolean pass = false && true",
                "return ${pass}"
        ));
        Asserts.assertEquals(result, false);
    }

    @Test
    public void testBooleanOrTrue() {
        Object result = runSteps(MutableList.of(
                "let boolean pass = false || true",
                "return ${pass}"
        ));
        Asserts.assertEquals(result, true);
    }

    @Test
    public void testBooleanOrFalse() {
        Object result = runSteps(MutableList.of(
                "let boolean pass = false || false",
                "return ${pass}"
        ));
        Asserts.assertEquals(result, false);
    }

    @Test
    public void testIntegerGreaterThanTrue() {
        Object result = runSteps(MutableList.of(
                "let boolean pass = 4 > 3",
                "return ${pass}"
        ));
        Asserts.assertEquals(result, true);
    }

    @Test
    public void testIntegerGreaterThanFalse() {
        Object result = runSteps(MutableList.of(
                "let boolean pass = 4 > 4",
                "return ${pass}"
        ));
        Asserts.assertEquals(result, false);
    }

    @Test
    public void testIntegerGreaterThanOrEqualTrue() {
        Object result = runSteps(MutableList.of(
                "let boolean pass = 4 >= 4",
                "return ${pass}"
        ));
        Asserts.assertEquals(result, true);
    }

    @Test
    public void testIntegerGreaterThanOrEqualFalse() {
        Object result = runSteps(MutableList.of(
                "let boolean pass = 3 >= 4",
                "return ${pass}"
        ));
        Asserts.assertEquals(result, false);
    }

    @Test
    public void testIntegerLessThanTrue() {
        Object result = runSteps(MutableList.of(
                "let boolean pass = 3 < 4",
                "return ${pass}"
        ));
        Asserts.assertEquals(result, true);
    }

    @Test
    public void testIntegerLessThanFalse() {
        Object result = runSteps(MutableList.of(
                "let boolean pass = 4 < 4",
                "return ${pass}"
        ));
        Asserts.assertEquals(result, false);
    }

    @Test
    public void testIntegerLessThanOrEqualTrue() {
        Object result = runSteps(MutableList.of(
                "let boolean pass = 4 <= 4",
                "return ${pass}"
        ));
        Asserts.assertEquals(result, true);
    }

    @Test
    public void testIntegerLessThanOrEqualFalse() {
        Object result = runSteps(MutableList.of(
                "let boolean pass = 4 <= 3",
                "return ${pass}"
        ));
        Asserts.assertEquals(result, false);
    }

    @Test
    public void testTernary() {
        // simple
        assertEvaluated("true ? \"left\" : \"right\"", "left");
        assertEvaluated("false ? \"left\" : \"right\"", "right");

        // nested
        assertEvaluated("true ? true ? \"a\" : \"b\" : \"c\"", "a");
        assertEvaluated("true ? false ? \"a\" : \"b\" : \"c\"", "b");
        assertEvaluated("false ? ignored ? \"a\" : \"b\" : \"c\"", "c");

        // chained
        assertEvaluated("false ? ignored : true ? \"b\" : \"c\"", "b");

        assertEvaluated("false ? ignored : true ? \"b\" : \"c\"", "b");

        Asserts.assertEquals(runSteps(MutableList.of(
                "let integer x = 4",
                "let integer y = ${x} == 4 ? 5 : 6",
                "return ${y}"
        )), 5);
    }
}