2 files changed, 249 insertions, 0 deletions
diff --git a/auth/cli-codegen/src/test/java/org/onap/policy/apex/auth/clicodegen/StErrorListener.java b/auth/cli-codegen/src/test/java/org/onap/policy/apex/auth/clicodegen/StErrorListener.java
new file mode 100644
index 000000000..ed703da82
--- /dev/null
+++ b/auth/cli-codegen/src/test/java/org/onap/policy/apex/auth/clicodegen/StErrorListener.java
@@ -0,0 +1,111 @@
+/*-
+ * ============LICENSE_START=======================================================
+ *  Copyright (C) 2016-2018 Ericsson. All rights reserved.
+ * ================================================================================
+ * Licensed 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.
+ * 
+ * SPDX-License-Identifier: Apache-2.0
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.policy.apex.auth.clicodegen;
+
+import org.stringtemplate.v4.STErrorListener;
+import org.stringtemplate.v4.misc.STMessage;
+
+/**
+ * Customized ST error listener.
+ *
+ * @author Sven van der Meer (sven.van.der.meer@ericsson.com)
+ */
+public class StErrorListener implements STErrorListener {
+
+    /** Counts errors of the listener. */
+    private int errorCount;
+
+    /* (non-Javadoc)
+     * @see org.stringtemplate.v4.STErrorListener#IOError(org.stringtemplate.v4.misc.STMessage)
+     */
+    @Override
+    public void IOError(final STMessage msg) {
+        switch (msg.error) {
+        default:
+            this.registerErrors(msg);
+        }
+    }
+
+    /* (non-Javadoc)
+     * @see org.stringtemplate.v4.STErrorListener#compileTimeError(org.stringtemplate.v4.misc.STMessage)
+     */
+    @Override
+    public void compileTimeError(final STMessage msg) {
+        switch (msg.error) {
+        default:
+            this.registerErrors(msg);
+        }
+    }
+
+    /* (non-Javadoc)
+     * @see org.stringtemplate.v4.STErrorListener#internalError(org.stringtemplate.v4.misc.STMessage)
+     */
+    @Override
+    public void internalError(final STMessage msg) {
+        switch (msg.error) {
+        default:
+            this.registerErrors(msg);
+        }
+    }
+
+    /* (non-Javadoc)
+     * @see org.stringtemplate.v4.STErrorListener#runTimeError(org.stringtemplate.v4.misc.STMessage)
+     */
+    @Override
+    public void runTimeError(final STMessage msg) {
+        switch (msg.error) {
+        case NO_SUCH_PROPERTY:
+        case ARGUMENT_COUNT_MISMATCH:
+        case ANON_ARGUMENT_MISMATCH:
+            break;
+        default:
+            this.registerErrors(msg);
+        }
+    }
+
+    /**
+     * Registers an error with the local error listener and increases the error count.
+     * 
+     * @param msg error message
+     */
+    protected void registerErrors(final STMessage msg) {
+        setErrorCount(getErrorCount() + 1);
+        System.err.println("STG/ST (" + msg.error + ") " + msg.arg + " -> " + msg.cause);
+    }
+
+    /**
+     * Gets the error count.
+     *
+     * @return the error count
+     */
+    protected int getErrorCount() {
+        return errorCount;
+    }
+
+    /**
+     * Sets the error count.
+     *
+     * @param errorCount the new error count
+     */
+    protected void setErrorCount(final int errorCount) {
+        this.errorCount = errorCount;
+    }
+}
diff --git a/auth/cli-codegen/src/test/java/org/onap/policy/apex/auth/clicodegen/TestSTG.java b/auth/cli-codegen/src/test/java/org/onap/policy/apex/auth/clicodegen/TestSTG.java
new file mode 100644
index 000000000..1fbc78de9
--- /dev/null
+++ b/auth/cli-codegen/src/test/java/org/onap/policy/apex/auth/clicodegen/TestSTG.java
@@ -0,0 +1,138 @@
+/*-
+ * ============LICENSE_START=======================================================
+ *  Copyright (C) 2016-2018 Ericsson. All rights reserved.
+ * ================================================================================
+ * Licensed 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.
+ * 
+ * SPDX-License-Identifier: Apache-2.0
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.policy.apex.auth.clicodegen;
+
+import static org.junit.Assert.assertEquals;
+
+import java.util.Arrays;
+import java.util.LinkedHashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+import java.util.TreeSet;
+
+import org.junit.Test;
+import org.onap.policy.apex.auth.clicodegen.CGCliEditor;
+import org.stringtemplate.v4.STGroup;
+import org.stringtemplate.v4.STGroupFile;
+
+/**
+ * Test for the CG CLI Editor STG file.
+ *
+ * @author Sven van der Meer (sven.van.der.meer@ericsson.com)
+ */
+public class TestSTG {
+
+    /**
+     * Get the chunks for the codegen.
+     *
+     * @return the chunks
+     */
+    private static Map<String, List<String>> getCodeGenChunks() {
+        // CHECKSTYLE:OFF: LineLength
+
+        final Map<String, List<String>> chunks = new LinkedHashMap<>();
+
+        chunks.put("/policyModel",
+                Arrays.asList("name", "version", "uuid", "description", "declarations", "definitions"));
+        chunks.put("/schemaDecl", Arrays.asList("name", "version", "uuid", "description", "flavour", "schema"));
+        chunks.put("/ctxAlbumDecl", Arrays.asList("name", "version", "uuid", "description", "scope", "writable",
+                "schemaName", "schemaVersion"));
+        chunks.put("/eventDecl",
+                Arrays.asList("name", "version", "uuid", "description", "nameSpace", "source", "target", "fields"));
+        chunks.put("/eventDefField",
+                Arrays.asList("eventName", "version", "fieldName", "fieldSchema", "fieldSchemaVersion", "optional"));
+        chunks.put("/taskDecl",
+                Arrays.asList("name", "version", "uuid", "description", "infields", "outfields", "logic"));
+        chunks.put("/taskDefInputFields",
+                Arrays.asList("taskName", "version", "fieldName", "fieldSchema", "fieldSchemaVersion"));
+        chunks.put("/taskDefOutputFields",
+                Arrays.asList("taskName", "version", "fieldName", "fieldSchema", "fieldSchemaVersion"));
+        chunks.put("/taskDefLogic", Arrays.asList("taskName", "version", "flavour", "logic"));
+        chunks.put("/taskDefParameter", Arrays.asList("name", "version", "parName", "defaultValue"));
+        chunks.put("/taskDefCtxRef", Arrays.asList("name", "version", "albumName", "albumVersion"));
+        chunks.put("/policyDef", Arrays.asList("name", "version", "uuid", "description", "template", "firstState"));
+        chunks.put("/policyStateDef", Arrays.asList("policyName", "version", "stateName", "triggerName",
+                "triggerVersion", "defaultTask", "defaultTaskVersion", "outputs", "tasks"));
+        chunks.put("/policyStateOutput", Arrays.asList("policyName", "version", "stateName", "outputName", "eventName",
+                "eventVersion", "nextState"));
+        chunks.put("/policyStateTaskSelectionLogic",
+                Arrays.asList("name", "version", "stateName", "logicFlavour", "logic"));
+        chunks.put("/policyStateTask", Arrays.asList("policyName", "version", "stateName", "taskLocalName", "taskName",
+                "taskVersion", "outputType", "outputName"));
+        chunks.put("/policyStateFinalizerLogic",
+                Arrays.asList("name", "version", "stateName", "finalizerLogicName", "logicFlavour", "logic"));
+        chunks.put("/policyStateContextRef",
+                Arrays.asList("name", "version", "stateName", "albumName", "albumVersion"));
+
+        return chunks;
+        // CHECKSTYLE:ON: LineLength
+    }
+
+    /** Test STG load. */
+    @Test
+    public void testSTGLoad() {
+        final StErrorListener errListener = new StErrorListener();
+        final STGroupFile stg = new STGroupFile(CGCliEditor.STG_FILE);
+        stg.setListener(errListener);
+
+        stg.getTemplateNames(); // dummy to compile group and get errors
+        assertEquals(0, errListener.getErrorCount());
+    }
+
+    /** Test STG chunks. */
+    @Test
+    public void testSTGChunks() {
+        final StErrorListener errListener = new StErrorListener();
+        final STGroupFile stg = new STGroupFile(CGCliEditor.STG_FILE);
+        stg.setListener(errListener);
+
+        stg.getTemplateNames(); // dummy to compile group and get errors
+        final Map<String, List<String>> chunks = getCodeGenChunks();
+        String error = "";
+        final Set<String> definedNames = stg.getTemplateNames();
+        for (final STGroup group : stg.getImportedGroups()) {
+            definedNames.addAll(group.getTemplateNames());
+        }
+        final Set<String> requiredNames = chunks.keySet();
+
+        for (final String required : requiredNames) {
+            if (!definedNames.contains(required)) {
+                error += " - target STG does not define template for <" + required + ">\n";
+            } else {
+                final Set<String> definedParams = ((stg.getInstanceOf(required).getAttributes() == null)
+                        ? new TreeSet<String>() : stg.getInstanceOf(required).getAttributes().keySet());
+                final List<String> requiredParams = chunks.get(required);
+                for (final String reqArg : requiredParams) {
+                    if (!definedParams.contains(reqArg)) {
+                        error += " - target STG with template <" + required + "> does not define argument <" + reqArg
+                                + ">\n";
+                    }
+                }
+            }
+        }
+
+        if (!("".equals(error))) {
+            System.err.println(error);
+        }
+        assertEquals(0, error.length());
+    }
+}