test case of model generator

Issue-ID: POLICY-1992
Change-Id: I90143d9903e0ada74dd9efba80aa6589b9f3dff4
Signed-off-by: ning.xi <ning.xi@est.tech>

1 files changed, 142 insertions, 0 deletions

diff --git a/tools/model-generator/src/test/java/org/onap/policy/apex/tools/model/generator/model2event/Model2EventSchemaTest.java b/tools/model-generator/src/test/java/org/onap/policy/apex/tools/model/generator/model2event/Model2EventSchemaTest.java
new file mode 100644
index 000000000..552101f94
--- /dev/null
+++ b/tools/model-generator/src/test/java/org/onap/policy/apex/tools/model/generator/model2event/Model2EventSchemaTest.java
@@ -0,0 +1,142 @@
+/*-
+ * ============LICENSE_START=======================================================
+ *  Copyright (C) 2019 Nordix Foundation.
+ * ================================================================================
+ * 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.tools.model.generator.model2event;
+
+import static org.assertj.core.api.Assertions.assertThatCode;
+import static org.junit.Assert.assertEquals;
+
+import java.util.Arrays;
+import org.apache.avro.Schema;
+import org.apache.avro.Schema.Field;
+import org.apache.avro.Schema.Type;
+import org.junit.AfterClass;
+import org.junit.BeforeClass;
+import org.junit.Test;
+import org.onap.policy.apex.context.impl.schema.java.JavaSchemaHelperParameters;
+import org.onap.policy.apex.context.parameters.ContextParameterConstants;
+import org.onap.policy.apex.context.parameters.SchemaParameters;
+import org.onap.policy.apex.model.basicmodel.service.ModelService;
+import org.onap.policy.common.parameters.ParameterService;
+import org.stringtemplate.v4.STGroupFile;
+
+/**
+ * Test the Model2EventSchema.
+ */
+public class Model2EventSchemaTest {
+    String modelFile = "src/test/resources/blankSchema.json";
+    String type = "stimuli";
+    /** The name of the application. */
+    public static final String APP_NAME = "gen-model2eventSchema";
+
+    /**
+     * Set ups parameterService for the test.
+     */
+    @BeforeClass
+    public static void prepareForTest() {
+        final SchemaParameters schemaParameters = new SchemaParameters();
+        schemaParameters.setName(ContextParameterConstants.SCHEMA_GROUP_NAME);
+        schemaParameters.getSchemaHelperParameterMap().put("JAVA", new JavaSchemaHelperParameters());
+        if (!ParameterService.contains(ContextParameterConstants.SCHEMA_GROUP_NAME)) {
+            ParameterService.register(schemaParameters);
+        }
+    }
+
+    @Test
+    public void testEventSchemaBadModelFile() {
+        Model2JsonEventSchema app = new Model2JsonEventSchema(modelFile, type, APP_NAME);
+        assertThatCode(() -> {
+            int ret = app.runApp();
+            assertEquals(-1, ret);
+        }).doesNotThrowAnyException();
+
+    }
+
+    @Test
+    public void testEventSchemaBadType() {
+        modelFile = "src/test/resources/SmallModel.json";
+        type = "default";
+        Model2JsonEventSchema app = new Model2JsonEventSchema(modelFile, type, APP_NAME);
+        assertThatCode(() -> {
+            int ret = app.runApp();
+            assertEquals(-1, ret);
+        }).doesNotThrowAnyException();
+    }
+
+    @Test
+    public void testEventSchemaStimuli() {
+        modelFile = "src/test/resources/SmallModel.json";
+        type = "stimuli";
+        Model2JsonEventSchema app = new Model2JsonEventSchema(modelFile, type, APP_NAME);
+        assertThatCode(() -> {
+            int ret = app.runApp();
+            assertEquals(0, ret);
+        }).doesNotThrowAnyException();
+    }
+
+    @Test
+    public void testEventSchemaResponse() {
+        modelFile = "src/test/resources/SmallModel.json";
+        type = "response";
+        Model2JsonEventSchema app = new Model2JsonEventSchema(modelFile, type, APP_NAME);
+        assertThatCode(() -> {
+            int ret = app.runApp();
+            assertEquals(0, ret);
+        }).doesNotThrowAnyException();
+    }
+
+    @Test
+    public void testEventSchemaInternal() {
+        modelFile = "src/test/resources/SmallModel.json";
+        type = "internal";
+        Model2JsonEventSchema app = new Model2JsonEventSchema(modelFile, type, APP_NAME);
+        assertThatCode(() -> {
+            int ret = app.runApp();
+            assertEquals(0, ret);
+        }).doesNotThrowAnyException();
+    }
+
+    @Test
+    public void testEventSchemaNotSimpleType() {
+        modelFile = "src/test/resources/ExecutionPropertiesRestTestPolicyModel.json";
+        type = "internal";
+        Model2JsonEventSchema app = new Model2JsonEventSchema(modelFile, type, APP_NAME);
+        final STGroupFile stg = new STGroupFile("org/onap/policy/apex/tools/model/generator/event-json.stg");
+
+        Field stringField = new Field("string", Schema.create(Type.STRING), null, null);
+        Field enumField =
+                new Field("enum", Schema.createEnum("my_enum", "doc", null, Arrays.asList("a", "b", "c")), null, null);
+        Schema schema = Schema.createRecord("my_record", "doc", "mytest", false);
+        schema.setFields(Arrays.asList(stringField, enumField));
+        Schema arrayOut = Schema.createArray(schema);
+        Schema mapOut = Schema.createMap(arrayOut);
+        app.addFieldType(mapOut, stg);
+        assertThatCode(() -> {
+            int ret = app.runApp();
+            assertEquals(0, ret);
+        }).doesNotThrowAnyException();
+    }
+
+    @AfterClass
+    public static void cleanTest() {
+        ParameterService.deregister(ContextParameterConstants.SCHEMA_GROUP_NAME);
+        ModelService.clear();
+    }
+}
\ No newline at end of file