aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--plugins/plugins-context/plugins-context-schema/plugins-context-schema-avro/src/main/java/org/onap/policy/apex/plugins/context/schema/avro/AvroSchemaKeyTranslationUtilities.java10
-rw-r--r--plugins/plugins-context/plugins-context-schema/plugins-context-schema-avro/src/test/java/org/onap/policy/apex/plugins/context/schema/avro/AvroSchemaMapTest.java107
-rw-r--r--plugins/plugins-context/plugins-context-schema/plugins-context-schema-avro/src/test/resources/avsc/MapExampleAddressInvalidFields.avsc3
-rw-r--r--plugins/plugins-context/plugins-context-schema/plugins-context-schema-avro/src/test/resources/data/MapExampleAddressInvalidFields.json14
4 files changed, 124 insertions, 10 deletions
diff --git a/plugins/plugins-context/plugins-context-schema/plugins-context-schema-avro/src/main/java/org/onap/policy/apex/plugins/context/schema/avro/AvroSchemaKeyTranslationUtilities.java b/plugins/plugins-context/plugins-context-schema/plugins-context-schema-avro/src/main/java/org/onap/policy/apex/plugins/context/schema/avro/AvroSchemaKeyTranslationUtilities.java
index 965457206..6229c066e 100644
--- a/plugins/plugins-context/plugins-context-schema/plugins-context-schema-avro/src/main/java/org/onap/policy/apex/plugins/context/schema/avro/AvroSchemaKeyTranslationUtilities.java
+++ b/plugins/plugins-context/plugins-context-schema/plugins-context-schema-avro/src/main/java/org/onap/policy/apex/plugins/context/schema/avro/AvroSchemaKeyTranslationUtilities.java
@@ -38,6 +38,8 @@ public final class AvroSchemaKeyTranslationUtilities {
private static final String DOT_STRING_REPLACEMENT = "_DoT_";
private static final String DASH_STRING = "-";
private static final String DASH_STRING_REPLACEMENT = "_DasH_";
+ private static final String COLON_STRING = ":";
+ private static final String COLON_STRING_REPLACEMENT = "_ColoN_";
/**
* Default constructor to avoid subclassing.
@@ -134,9 +136,13 @@ public final class AvroSchemaKeyTranslationUtilities {
*/
private static String translateIllegalKey(final String key, final boolean revert) {
if (revert) {
- return key.replace(DOT_STRING_REPLACEMENT, DOT_STRING).replace(DASH_STRING_REPLACEMENT, DASH_STRING);
+ return key.replace(DOT_STRING_REPLACEMENT, DOT_STRING)
+ .replace(DASH_STRING_REPLACEMENT, DASH_STRING)
+ .replace(COLON_STRING_REPLACEMENT, COLON_STRING);
} else {
- return key.replace(DOT_STRING, DOT_STRING_REPLACEMENT).replace(DASH_STRING, DASH_STRING_REPLACEMENT);
+ return key.replace(DOT_STRING, DOT_STRING_REPLACEMENT)
+ .replace(DASH_STRING, DASH_STRING_REPLACEMENT)
+ .replace(COLON_STRING, COLON_STRING_REPLACEMENT);
}
}
}
diff --git a/plugins/plugins-context/plugins-context-schema/plugins-context-schema-avro/src/test/java/org/onap/policy/apex/plugins/context/schema/avro/AvroSchemaMapTest.java b/plugins/plugins-context/plugins-context-schema/plugins-context-schema-avro/src/test/java/org/onap/policy/apex/plugins/context/schema/avro/AvroSchemaMapTest.java
index 690234b47..abc9335ae 100644
--- a/plugins/plugins-context/plugins-context-schema/plugins-context-schema-avro/src/test/java/org/onap/policy/apex/plugins/context/schema/avro/AvroSchemaMapTest.java
+++ b/plugins/plugins-context/plugins-context-schema/plugins-context-schema-avro/src/test/java/org/onap/policy/apex/plugins/context/schema/avro/AvroSchemaMapTest.java
@@ -22,6 +22,9 @@
package org.onap.policy.apex.plugins.context.schema.avro;
import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertNull;
+import static org.junit.Assert.assertThrows;
import java.io.File;
import java.io.IOException;
@@ -31,6 +34,7 @@ import org.apache.avro.util.Utf8;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
+import org.onap.policy.apex.context.ContextRuntimeException;
import org.onap.policy.apex.context.SchemaHelper;
import org.onap.policy.apex.context.impl.schema.SchemaHelperFactory;
import org.onap.policy.apex.context.parameters.ContextParameterConstants;
@@ -80,7 +84,6 @@ public class AvroSchemaMapTest {
schemaParameters.setName(ContextParameterConstants.SCHEMA_GROUP_NAME);
schemaParameters.getSchemaHelperParameterMap().put("AVRO", new AvroSchemaHelperParameters());
ParameterService.register(schemaParameters);
-
}
/**
@@ -92,6 +95,100 @@ public class AvroSchemaMapTest {
}
/**
+ * Test valid schemas with substitutions.
+ *
+ * @throws IOException Signals that an I/O exception has occurred.
+ */
+ @Test
+ public void testValidSubstitutions() throws IOException {
+ final String subst1 = "{\"type\":\"record\",\"name\":\"Subst1\","
+ + "\"fields\":[{\"name\": \"A_DasH_B\",\"type\":\"string\"}]}";
+ final AxContextSchema avroSubstSchema1 = new AxContextSchema(
+ new AxArtifactKey("AvroSubst1", "0.0.1"), "AVRO", subst1);
+ schemas.getSchemasMap().put(avroSubstSchema1.getKey(), avroSubstSchema1);
+
+ SchemaHelper schemaHelperSubst1 = new SchemaHelperFactory()
+ .createSchemaHelper(testKey, avroSubstSchema1.getKey());
+ final GenericRecord subst1A = (GenericRecord) schemaHelperSubst1.unmarshal("{\"A-B\":\"foo\"}");
+ assertEquals(new Utf8("foo"), subst1A.get("A_DasH_B"));
+ assertNull(subst1A.get("A-B"));
+ final Throwable exception1 = assertThrows(ContextRuntimeException.class,
+ () -> schemaHelperSubst1.unmarshal("{\"A-B\":123}"));
+ assertNotNull(exception1.getCause());
+ assertEquals("Expected string. Got VALUE_NUMBER_INT", exception1.getCause().getMessage());
+
+ final String subst2 = "{\"type\":\"record\",\"name\":\"Subst2\","
+ + "\"fields\":[{\"name\": \"C_DoT_D\",\"type\":\"int\"}]}";
+ final AxContextSchema avroSubstSchema2 = new AxContextSchema(
+ new AxArtifactKey("AvroSubst2", "0.0.1"), "AVRO", subst2);
+ schemas.getSchemasMap().put(avroSubstSchema2.getKey(), avroSubstSchema2);
+
+ final SchemaHelper schemaHelperSubst2 = new SchemaHelperFactory()
+ .createSchemaHelper(testKey, avroSubstSchema2.getKey());
+ final GenericRecord subst2A = (GenericRecord) schemaHelperSubst2.unmarshal("{\"C.D\":123}");
+ assertEquals(123, subst2A.get("C_DoT_D"));
+ assertNull(subst2A.get("C.D"));
+ final Throwable exception2 = assertThrows(ContextRuntimeException.class,
+ () -> schemaHelperSubst2.unmarshal("{\"C_DoT_D\":\"bar\"}"));
+ assertNotNull(exception2.getCause());
+ assertEquals("Expected int. Got VALUE_STRING", exception2.getCause().getMessage());
+
+ final String subst3 = "{\"type\":\"record\",\"name\":\"Subst3\","
+ + "\"fields\":[{\"name\": \"E_ColoN_F\",\"type\":\"boolean\"}]}";
+ final AxContextSchema avroSubstSchema3 = new AxContextSchema(
+ new AxArtifactKey("AvroSubst3", "0.0.1"), "AVRO", subst3);
+ schemas.getSchemasMap().put(avroSubstSchema3.getKey(), avroSubstSchema3);
+
+ final SchemaHelper schemaHelperSubst3 = new SchemaHelperFactory()
+ .createSchemaHelper(testKey, avroSubstSchema3.getKey());
+ final GenericRecord subst3A = (GenericRecord) schemaHelperSubst3.unmarshal("{\"E:F\":true}");
+ assertEquals(true, subst3A.get("E_ColoN_F"));
+ assertNull(subst3A.get("E:F"));
+ final Throwable exception3 = assertThrows(ContextRuntimeException.class,
+ () -> schemaHelperSubst3.unmarshal("{\"E_ColoN_F\":\"gaz\"}"));
+ assertNotNull(exception3.getCause());
+ assertEquals("Expected boolean. Got VALUE_STRING", exception3.getCause().getMessage());
+ }
+
+ /**
+ * Test invalid schemas without substitutions.
+ *
+ * @throws IOException Signals that an I/O exception has occurred.
+ */
+ @Test
+ public void testInValidSubstitutions() throws IOException {
+ final String fail1 = "{\"type\":\"record\",\"name\":\"Fail1\","
+ + "\"fields\":[{\"name\": \"A-B\",\"type\":\"string\"}]}";
+ final AxContextSchema avroFailSchema1 = new AxContextSchema(
+ new AxArtifactKey("AvroFail1", "0.0.1"), "AVRO", fail1);
+ schemas.getSchemasMap().put(avroFailSchema1.getKey(), avroFailSchema1);
+ final Throwable exception1 = assertThrows(ContextRuntimeException.class,
+ () -> new SchemaHelperFactory().createSchemaHelper(testKey, avroFailSchema1.getKey()));
+ assertNotNull(exception1.getCause());
+ assertEquals("Illegal character in: A-B", exception1.getCause().getMessage());
+
+ final String fail2 = "{\"type\":\"record\",\"name\":\"Fail2\","
+ + "\"fields\":[{\"name\": \"C.D\",\"type\":\"int\"}]}";
+ final AxContextSchema avroFailSchema2 = new AxContextSchema(
+ new AxArtifactKey("AvroFail2", "0.0.1"), "AVRO", fail2);
+ schemas.getSchemasMap().put(avroFailSchema2.getKey(), avroFailSchema2);
+ final Throwable exception2 = assertThrows(ContextRuntimeException.class,
+ () -> new SchemaHelperFactory().createSchemaHelper(testKey, avroFailSchema2.getKey()));
+ assertNotNull(exception2.getCause());
+ assertEquals("Illegal character in: C.D", exception2.getCause().getMessage());
+
+ final String fail3 = "{\"type\":\"record\",\"name\":\"Fail3\","
+ + "\"fields\":[{\"name\": \"E:F\",\"type\":\"boolean\"}]}";
+ final AxContextSchema avroFailSchema3 = new AxContextSchema(
+ new AxArtifactKey("AvroFail3", "0.0.1"), "AVRO", fail3);
+ schemas.getSchemasMap().put(avroFailSchema3.getKey(), avroFailSchema3);
+ final Throwable exception3 = assertThrows(ContextRuntimeException.class,
+ () -> new SchemaHelperFactory().createSchemaHelper(testKey, avroFailSchema3.getKey()));
+ assertNotNull(exception3.getCause());
+ assertEquals("Illegal character in: E:F", exception3.getCause().getMessage());
+ }
+
+ /**
* Test map init.
*
* @throws IOException Signals that an I/O exception has occurred.
@@ -162,7 +259,7 @@ public class AvroSchemaMapTest {
final SchemaHelper schemaHelper = new SchemaHelperFactory().createSchemaHelper(testKey, avroSchema.getKey());
GenericRecord subRecord = (GenericRecord) schemaHelper.createNewSubInstance("AddressUSRecord");
- assertEquals(null, subRecord.get("streetAddress"));
+ assertNull(subRecord.get("streetAddress"));
}
/**
@@ -179,6 +276,12 @@ public class AvroSchemaMapTest {
final SchemaHelper schemaHelper = new SchemaHelperFactory().createSchemaHelper(testKey, avroSchema.getKey());
testUnmarshalMarshal(schemaHelper, "src/test/resources/data/MapExampleAddressInvalidFields.json");
+
+ String vals = TextFileUtils.getTextFileAsString("src/test/resources/data/MapExampleAddressInvalidFields.json");
+ final HashMap<?, ?> newMapFull = (HashMap<?, ?>) schemaHelper.createNewInstance(vals);
+ final String expect = "{\"street_DasH_address\": \"Wayne Manor\", \"the_DoT_city\": \"Gotham City\", "
+ + "\"the_ColoN_code\": \"BatCave7\"}";
+ assertEquals(expect, newMapFull.get(new Utf8("address_DoT_3")).toString());
}
/**
diff --git a/plugins/plugins-context/plugins-context-schema/plugins-context-schema-avro/src/test/resources/avsc/MapExampleAddressInvalidFields.avsc b/plugins/plugins-context/plugins-context-schema/plugins-context-schema-avro/src/test/resources/avsc/MapExampleAddressInvalidFields.avsc
index ddffb4bbc..86c145bfc 100644
--- a/plugins/plugins-context/plugins-context-schema/plugins-context-schema-avro/src/test/resources/avsc/MapExampleAddressInvalidFields.avsc
+++ b/plugins/plugins-context/plugins-context-schema/plugins-context-schema-avro/src/test/resources/avsc/MapExampleAddressInvalidFields.avsc
@@ -4,7 +4,8 @@
"name" : "AddressUSRecord",
"fields" : [
{"name": "street_DasH_address", "type": "string"},
- {"name": "the_DoT_city", "type": "string"}
+ {"name": "the_DoT_city", "type": "string"},
+ {"name": "the_ColoN_code", "type": "string"}
]
}
} \ No newline at end of file
diff --git a/plugins/plugins-context/plugins-context-schema/plugins-context-schema-avro/src/test/resources/data/MapExampleAddressInvalidFields.json b/plugins/plugins-context/plugins-context-schema/plugins-context-schema-avro/src/test/resources/data/MapExampleAddressInvalidFields.json
index 6116db4f3..a1a70e95a 100644
--- a/plugins/plugins-context/plugins-context-schema/plugins-context-schema-avro/src/test/resources/data/MapExampleAddressInvalidFields.json
+++ b/plugins/plugins-context/plugins-context-schema/plugins-context-schema-avro/src/test/resources/data/MapExampleAddressInvalidFields.json
@@ -1,18 +1,22 @@
{
"address.0" : {
"street-address" : "1600 Pennsylvania Avenue",
- "the.city" : "Washington DC"
+ "the.city" : "Washington DC",
+ "the:code" : "12345"
},
"address.1" : {
"street-address" : "Somewhere",
- "the.city" : "Over the rainbow"
+ "the.city" : "Over the rainbow",
+ "the:code" : "BlueBird"
},
"address.2" : {
"street-address" : "221 B Baker St.",
- "the.city" : "London"
+ "the.city" : "London",
+ "the:code" : "NW1"
},
"address.3" : {
"street-address" : "Wayne Manor",
- "the.city" : "Gotham City"
+ "the.city" : "Gotham City",
+ "the:code" : "BatCave7"
}
-} \ No newline at end of file
+}