b129330ce650d83cf07d9ad184f829a664ce0bf0
[policy/apex-pdp.git] /
1 /*-
2  * ============LICENSE_START=======================================================
3  *  Copyright (C) 2022 Bell Canada. All rights reserved.
4  * Modifications Copyright (C) 2024 Nordix Foundation.
5  * ================================================================================
6  * Licensed under the Apache License, Version 2.0 (the "License");
7  * you may not use this file except in compliance with the License.
8  * You may obtain a copy of the License at
9  *
10  *      http://www.apache.org/licenses/LICENSE-2.0
11  *
12  * Unless required by applicable law or agreed to in writing, software
13  * distributed under the License is distributed on an "AS IS" BASIS,
14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15  * See the License for the specific language governing permissions and
16  * limitations under the License.
17  *
18  * SPDX-License-Identifier: Apache-2.0
19  * ============LICENSE_END=========================================================
20  */
21
22 package org.onap.policy.apex.plugins.context.schema.json;
23
24 import static org.assertj.core.api.Assertions.assertThat;
25 import static org.assertj.core.api.Assertions.assertThatThrownBy;
26
27 import com.google.gson.JsonArray;
28 import com.google.gson.JsonObject;
29 import com.worldturner.medeia.api.ValidationFailedException;
30 import java.util.ArrayList;
31 import java.util.Map;
32 import org.junit.jupiter.api.Test;
33 import org.onap.policy.common.utils.coder.CoderException;
34
35 class JsonSchemaHelperMarshalTest extends CommonTestData {
36
37     /**
38      * Test Boolean.
39      */
40     @Test
41     void testBooleanMarshal() {
42         var schemaHelper = createSchema(BOOLEAN_SCHEMA);
43         assertThat(schemaHelper.marshal2String(Boolean.TRUE)).isEqualTo("true");
44     }
45
46     /**
47      * Test null.
48      */
49     @Test
50     void testNullMarshal() {
51         var schemaHelper = createSchema(NULL_SCHEMA);
52         assertThat(schemaHelper.marshal2String(null)).isEqualTo("null");
53     }
54
55     /**
56      * Test Array.
57      *
58      * @throws CoderException the coder exception
59      */
60     @Test
61     void testArrayMarshal() throws CoderException {
62         var schemaHelper = createSchema(MEASUREMENTGROUPS_TYPE);
63         var object = coder.decode(MEASUREMENTGROUPS, Object.class);
64         assertThat(object).isInstanceOf(ArrayList.class);
65         var marshalledString = schemaHelper.marshal2String(object);
66         assertThat(marshalledString).isEqualTo(MEASUREMENTGROUPS);
67         var marshalledObject = schemaHelper.marshal2Object(object);
68         assertThat(marshalledObject).isInstanceOf(JsonArray.class);
69     }
70
71     /**
72      * Test Object marshal valid scenario using JSON Schema draft 04.
73      *
74      * @throws CoderException the coderException
75      */
76     @Test
77     void testObjectSchemaDraft04_valid() throws CoderException {
78         var dataAsObject = coder.decode(COMMONHEADER, Map.class);
79         var dataReturned = validateAndMarshal(COMMONHEADERTYPE_DRAFT04, dataAsObject, true);
80         assertThat(dataReturned).isEqualTo(COMMONHEADER);
81     }
82
83     /**
84      * Test Object marshal valid scenario using JSON Schema draft 07.
85      *
86      * @throws CoderException the coderException
87      */
88     @Test
89     void testObjectSchemaDraft07_valid() throws CoderException {
90         var dataAsObject = coder.decode(COMMONHEADER, Map.class);
91         var dataReturned = validateAndMarshal(COMMONHEADERTYPE_DRAFT07, dataAsObject, true);
92         assertThat(dataReturned).isEqualTo(COMMONHEADER);
93         assertThat(validateAndMarshal(COMMONHEADERTYPE_DRAFT07, dataAsObject, false)).isInstanceOf(JsonObject.class);
94     }
95
96     /**
97      * Test Object marshal invalid - required field missing scenario.
98      *
99      * @throws CoderException the coderException
100      */
101     @Test
102     void testObjectSchema_fieldMissing() throws CoderException {
103         var dataAsObject = coder.decode(COMMONHEADER, Map.class);
104         dataAsObject.remove(TEST_ID);
105         assertThatThrownBy(() -> validateAndMarshal(COMMONHEADERTYPE_DRAFT07, dataAsObject, true))
106             .isInstanceOf(ValidationFailedException.class)
107             .hasMessageContaining("Required property testId is missing from object");
108     }
109
110     /**
111      * Test Object marshal with optional field.
112      *
113      * @throws CoderException the coderException
114      */
115     @Test
116     void testObjectSchema_OptionalField() throws CoderException {
117         var dataAsObject = coder.decode(COMMONHEADER, Map.class);
118         var dataAsjsonObject = coder.decode(COMMONHEADER, JsonObject.class);
119         dataAsObject.remove(TEST_ID);
120         dataAsjsonObject.remove(TEST_ID);
121         var dataReturned = validateAndMarshal(COMMONHEADERTYPE_WITH_OPTIONAL, dataAsObject, false);
122         assertThat(dataReturned).isEqualTo(dataAsjsonObject);
123     }
124
125     private Object validateAndMarshal(String schemaDef, Object obj, boolean marshalAsString) {
126         var schemaHelper = createSchema(schemaDef);
127         if (marshalAsString) {
128             return schemaHelper.marshal2String(obj);
129         } else {
130             return schemaHelper.marshal2Object(obj);
131         }
132     }
133
134 }