1 files changed, 235 insertions, 0 deletions
diff --git a/aai-core/src/test/java/org/onap/aai/util/StoreNotificationEventTest.java b/aai-core/src/test/java/org/onap/aai/util/StoreNotificationEventTest.java
new file mode 100644
index 00000000..5c87b29e
--- /dev/null
+++ b/aai-core/src/test/java/org/onap/aai/util/StoreNotificationEventTest.java
@@ -0,0 +1,235 @@
+/**
+ * ============LICENSE_START=======================================================
+ * org.onap.aai
+ * ================================================================================
+ * Copyright © 2017 AT&T Intellectual Property. 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.
+ * ============LICENSE_END=========================================================
+ *
+ * ECOMP is a trademark and service mark of AT&T Intellectual Property.
+ */
+package org.onap.aai.util;
+
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertTrue;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.StringWriter;
+import java.util.HashMap;
+import java.util.Map;
+
+import javax.json.Json;
+import javax.json.JsonObject;
+
+import org.eclipse.persistence.dynamic.DynamicEntity;
+import org.eclipse.persistence.dynamic.DynamicType;
+import org.eclipse.persistence.jaxb.JAXBContextProperties;
+import org.eclipse.persistence.jaxb.JAXBMarshaller;
+import org.eclipse.persistence.jaxb.dynamic.DynamicJAXBContext;
+import org.eclipse.persistence.jaxb.dynamic.DynamicJAXBContextFactory;
+import org.junit.BeforeClass;
+import org.junit.Test;
+import org.mockito.Mockito;
+import org.onap.aai.AAISetup;
+import org.onap.aai.dmaap.AAIDmaapEventJMSProducer;
+import org.onap.aai.domain.notificationEvent.NotificationEvent.EventHeader;
+import org.onap.aai.exceptions.AAIException;
+import org.onap.aai.introspection.Introspector;
+import org.onap.aai.introspection.Loader;
+import org.onap.aai.introspection.LoaderFactory;
+import org.onap.aai.introspection.ModelType;
+import org.onap.aai.introspection.Version;
+
+import com.fasterxml.jackson.core.JsonGenerationException;
+import com.fasterxml.jackson.databind.JsonMappingException;
+
+public class StoreNotificationEventTest extends AAISetup {
+
+	private static AAIDmaapEventJMSProducer producer;
+	private static StoreNotificationEvent sne;
+
+	@BeforeClass
+	public static void setUp() throws Exception {
+		producer = Mockito.mock(AAIDmaapEventJMSProducer.class);
+		sne = new StoreNotificationEvent(producer, "transiationId", "sourceOfTruth");
+	}
+
+	@Test(expected = AAIException.class)
+	public void testStoreEventNullObj() throws AAIException {
+		sne.storeEvent(new EventHeader(), null);
+	}
+
+	@Test(expected = AAIException.class)
+	public void testStoreEventInvalidObjForPojoUtils() throws AAIException {
+		sne.storeEvent(new EventHeader(), new Object());
+	}
+
+	@Test
+	public void testStoreEventEmptyEventHeader() throws AAIException, JsonGenerationException, JsonMappingException, IOException {
+		JsonObject object = Json.createObjectBuilder().add("hello", "world").build();
+		String res = sne.storeEvent(new EventHeader(), object);
+
+		assertNotNull(res);
+		assertTrue(res.contains("\"cambria.partition\" : \"" + AAIConstants.UEB_PUB_PARTITION_AAI + "\""));
+		assertTrue(res.contains("\"event-header\""));
+		assertTrue(res.contains("\"id\""));
+		assertTrue(res.contains("\"timestamp\""));
+		assertTrue(res.contains("\"source-name\" : \"" + AAIConfig.get("aai.notificationEvent.default.sourceName") + "\""));
+		assertTrue(res.contains("\"domain\" : \"" + AAIConfig.get("aai.notificationEvent.default.domain") + "\""));
+		assertTrue(res.contains("\"sequence-number\" : \"" + AAIConfig.get("aai.notificationEvent.default.sequenceNumber") + "\""));
+		assertTrue(res.contains("\"severity\" : \"" + AAIConfig.get("aai.notificationEvent.default.severity") + "\""));
+		assertTrue(res.contains("\"event-type\" : \"" + AAIConfig.get("aai.notificationEvent.default.eventType") + "\""));
+		assertTrue(res.contains("\"version\" : \"" + AAIConfig.get("aai.notificationEvent.default.version") + "\""));
+		assertTrue(res.contains("\"action\" : \"UNK\""));
+		assertTrue(res.contains("\"entity-link\" : \"UNK\""));
+		assertTrue(res.contains("\"entity\""));
+		assertTrue(res.contains("\"hello\""));
+		assertTrue(res.contains("\"chars\" : \"world\""));
+		assertTrue(res.contains("\"string\" : \"world\""));
+		assertTrue(res.contains("\"valueType\" : \"STRING\""));
+	}
+
+	@Test
+	public void testStoreEvent() throws AAIException, JsonGenerationException, JsonMappingException, IOException {
+		JsonObject object = Json.createObjectBuilder().add("hello", "world").build();
+		EventHeader eh = new EventHeader();
+		eh.setId("123");
+		eh.setTimestamp("current-time");
+		eh.setEntityLink("entity-link");
+		eh.setAction("action!");
+		eh.setEventType("surprise");
+		eh.setDomain("PROD");
+		eh.setSourceName("source");
+		eh.setSequenceNumber("23");
+		eh.setSeverity("ALERT");
+		eh.setVersion("v11");
+
+		String res = sne.storeEvent(eh, object);
+
+		assertNotNull(res);
+		assertTrue(res.contains("\"cambria.partition\" : \"" + AAIConstants.UEB_PUB_PARTITION_AAI + "\""));
+		assertTrue(res.contains("\"event-header\""));
+		assertTrue(res.contains("\"id\" : \"123\""));
+		assertTrue(res.contains("\"timestamp\" : \"current-time\""));
+		assertTrue(res.contains("\"source-name\" : \"source\""));
+		assertTrue(res.contains("\"domain\" : \"PROD\""));
+		assertTrue(res.contains("\"sequence-number\" : \"23\""));
+		assertTrue(res.contains("\"severity\" : \"ALERT\""));
+		assertTrue(res.contains("\"event-type\" : \"surprise\""));
+		assertTrue(res.contains("\"version\" : \"v11\""));
+		assertTrue(res.contains("\"action\" : \"action!\""));
+		assertTrue(res.contains("\"entity-link\" : \"entity-link\""));
+		assertTrue(res.contains("\"entity\""));
+		assertTrue(res.contains("\"hello\""));
+		assertTrue(res.contains("\"chars\" : \"world\""));
+		assertTrue(res.contains("\"string\" : \"world\""));
+		assertTrue(res.contains("\"valueType\" : \"STRING\""));
+	}
+
+	@Test(expected=AAIException.class)
+	public void testStoreDynamicEventNullObj() throws AAIException {
+		DynamicEntity eventHeader = Mockito.mock(DynamicEntity.class);
+		DynamicJAXBContext notificationJaxbContext = Mockito.mock(DynamicJAXBContext.class);
+		sne.storeDynamicEvent(notificationJaxbContext, "v11", eventHeader, null);
+	}
+
+	@Test(expected = AAIException.class)
+	public void testStoreDynamicEventAAIException() throws Exception {
+		DynamicEntity eventHeader = Mockito.mock(DynamicEntity.class);
+		DynamicEntity obj = Mockito.mock(DynamicEntity.class);
+		DynamicJAXBContext notificationJaxbContext = Mockito.mock(DynamicJAXBContext.class);
+		ClassLoader cl = getClass().getClassLoader();
+		InputStream is = cl.getResourceAsStream("bundleconfig-local/etc/oxm/aai_oxm_v11.xml");
+		Map<String, Object> properties = new HashMap<String, Object>();
+		properties.put(JAXBContextProperties.OXM_METADATA_SOURCE, is);
+		DynamicJAXBContext notificationJaxbContextReal = DynamicJAXBContextFactory.createContextFromOXM(cl, properties);
+		DynamicType dtReal = notificationJaxbContextReal.getDynamicType("inventory.aai.onap.org.v11.NotificationEvent");
+		DynamicType dt = Mockito.mock(DynamicType.class);
+		DynamicEntity notificationEventReal = dtReal.newDynamicEntity();
+		JAXBMarshaller marshaller = Mockito.mock(JAXBMarshaller.class);
+		
+		Mockito.when(notificationJaxbContext.getDynamicType(Mockito.anyString())).thenReturn(dt);
+		Mockito.when(dt.newDynamicEntity()).thenReturn(notificationEventReal);
+		Mockito.when(notificationJaxbContext.createMarshaller()).thenReturn(marshaller);
+		Mockito.doNothing().when(marshaller).marshal(Mockito.any(DynamicJAXBContext.class), Mockito.any(StringWriter.class));
+
+		sne.storeDynamicEvent(notificationJaxbContext, "v11", eventHeader, obj);
+	}
+	
+	@Test(expected = AAIException.class)
+	public void testStoreEventIntrospectorNullObj() throws Exception {
+		Loader loader = Mockito.mock(Loader.class);
+		sne.storeEvent(loader, null, null);
+	}
+	
+	@Test
+	public void testStoreEventIntrospector() throws Exception {
+		Loader loader = LoaderFactory.createLoaderForVersion(ModelType.MOXY, Version.v11);
+		Introspector eventHeader = loader.introspectorFromName("notification-event-header");
+		eventHeader.setValue("id", "123");
+		eventHeader.setValue("timestamp", "current-time");
+		eventHeader.setValue("entity-link", "entity-link");
+		eventHeader.setValue("action", "action!");
+		eventHeader.setValue("event-type", "surprise");
+		eventHeader.setValue("domain", "PROD");
+		eventHeader.setValue("source-name", "source");
+		eventHeader.setValue("sequence-number", "23");
+		eventHeader.setValue("severity", "ALERT");
+		eventHeader.setValue("version", "v11");
+		
+		Introspector obj = loader.introspectorFromName("notification-event");
+		
+		String res = sne.storeEvent(loader, eventHeader, obj);
+		
+		assertNotNull(res);
+		assertTrue(res.contains("\"cambria.partition\":\"" + AAIConstants.UEB_PUB_PARTITION_AAI + "\""));
+		assertTrue(res.contains("\"event-header\""));
+		assertTrue(res.contains("\"id\":\"123\""));
+		assertTrue(res.contains("\"timestamp\":\"current-time\""));
+		assertTrue(res.contains("\"source-name\":\"source\""));
+		assertTrue(res.contains("\"domain\":\"PROD\""));
+		assertTrue(res.contains("\"sequence-number\":\"23\""));
+		assertTrue(res.contains("\"severity\":\"ALERT\""));
+		assertTrue(res.contains("\"event-type\":\"surprise\""));
+		assertTrue(res.contains("\"version\":\"v11\""));
+		assertTrue(res.contains("\"action\":\"action!\""));
+		assertTrue(res.contains("\"entity-link\":\"entity-link\""));
+		assertTrue(res.contains("\"notification-event\""));
+	}
+	
+	@Test
+	public void testStoreEventIntrospectorEmptyEventHeader() throws Exception {
+		Loader loader = LoaderFactory.createLoaderForVersion(ModelType.MOXY, Version.v11);
+		Introspector eventHeader = loader.introspectorFromName("notification-event-header");
+		Introspector obj = loader.introspectorFromName("notification-event");
+		
+		String res = sne.storeEvent(loader, eventHeader, obj);
+		
+		assertNotNull(res);
+		assertTrue(res.contains("\"cambria.partition\":\"" + AAIConstants.UEB_PUB_PARTITION_AAI + "\""));
+		assertTrue(res.contains("\"event-header\""));
+		assertTrue(res.contains("\"id\""));
+		assertTrue(res.contains("\"timestamp\""));
+		assertTrue(res.contains("\"source-name\":\"" + AAIConfig.get("aai.notificationEvent.default.sourceName") + "\""));
+		assertTrue(res.contains("\"domain\":\"" + AAIConfig.get("aai.notificationEvent.default.domain") + "\""));
+		assertTrue(res.contains("\"sequence-number\":\"" + AAIConfig.get("aai.notificationEvent.default.sequenceNumber") + "\""));
+		assertTrue(res.contains("\"severity\":\"" + AAIConfig.get("aai.notificationEvent.default.severity") + "\""));
+		assertTrue(res.contains("\"event-type\":\"" + AAIConfig.get("aai.notificationEvent.default.eventType") + "\""));
+		assertTrue(res.contains("\"version\":\"" + AAIConfig.get("aai.notificationEvent.default.version") + "\""));
+		assertTrue(res.contains("\"action\":\"UNK\""));
+		assertTrue(res.contains("\"entity-link\":\"UNK\""));
+		assertTrue(res.contains("\"notification-event\""));
+	}
+}