1 files changed, 130 insertions, 0 deletions
diff --git a/common-logging/src/test/java/org/onap/policy/common/logging/eelf/EventTrackInfoTest.java b/common-logging/src/test/java/org/onap/policy/common/logging/eelf/EventTrackInfoTest.java
new file mode 100644
index 00000000..3690bc5c
--- /dev/null
+++ b/common-logging/src/test/java/org/onap/policy/common/logging/eelf/EventTrackInfoTest.java
@@ -0,0 +1,130 @@
+/*
+ * ============LICENSE_START=======================================================
+ * ONAP Policy Engine - Common Modules
+ * ================================================================================
+ * Copyright (C) 2018 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=========================================================
+ */
+
+package org.onap.policy.common.logging.eelf;
+
+import static org.junit.Assert.*;
+
+import java.time.Instant;
+
+import org.junit.Before;
+import org.junit.Test;
+
+/**
+ * 
+ */
+public class EventTrackInfoTest {
+	
+	private static final Instant istart = Instant.ofEpochMilli(100000l);
+	private static final Instant iend = Instant.ofEpochMilli(200000l);
+	
+	private static final EventData data1 = new EventData("abc", istart, iend);
+	private static final EventData data2 = new EventData("def", iend, istart);
+	
+	private EventTrackInfo info;
+	
+	@Before
+	public void setUp() {
+		info = new EventTrackInfo();
+		
+	}
+
+	/**
+	 * Test method for {@link EventTrackInfo#EventTrackInfo()}.
+	 */
+	@Test
+	public void testEventTrackInfo() {
+		assertNotNull(info.getEventInfo());
+	}
+
+	/**
+	 * Test method for {@link EventTrackInfo#getEventDataByRequestID(String)}.
+	 */
+	@Test
+	public void testGetEventDataByRequestID() {
+		info.storeEventData(data1);
+		info.storeEventData(data2);
+
+		assertTrue(data1 == info.getEventDataByRequestID("abc"));
+		assertTrue(data2 == info.getEventDataByRequestID("def"));
+		assertNull(info.getEventDataByRequestID("hello"));
+	}
+
+	/**
+	 * Test method for {@link EventTrackInfo#storeEventData(EventData)}.
+	 */
+	@Test
+	public void testStoreEventData() {
+		// should ignore null
+		info.storeEventData(null);
+		assertTrue(info.getEventInfo().isEmpty());
+		
+		// should ignore if request id is null or empty
+		info.storeEventData(new EventData());
+		info.storeEventData(new EventData("", istart, iend));
+		assertTrue(info.getEventInfo().isEmpty());
+		
+		info.storeEventData(data1);
+		info.storeEventData(data2);
+		assertEquals(2, info.getEventInfo().size());
+
+		// look-up by request id
+		assertTrue(data1 == info.getEventDataByRequestID("abc"));
+		assertTrue(data2 == info.getEventDataByRequestID("def"));
+		
+		// doesn't replace existing value
+		info.storeEventData(new EventData("abc", iend, istart));
+		assertEquals(2, info.getEventInfo().size());
+		assertTrue(data1 == info.getEventDataByRequestID("abc"));
+		assertTrue(data2 == info.getEventDataByRequestID("def"));
+	}
+
+	/**
+	 * Test method for {@link EventTrackInfo#remove(String)}.
+	 */
+	@Test
+	public void testRemove() {
+		info.storeEventData(data1);
+		info.storeEventData(data2);
+		
+		info.remove("abc");
+		
+		// ensure only that item was removed
+		assertEquals(1, info.getEventInfo().size());
+
+		// look-up by request id
+		assertNull(info.getEventDataByRequestID("abc"));
+		assertTrue(data2 == info.getEventDataByRequestID("def"));
+	}
+
+	/**
+	 * Test method for {@link EventTrackInfo#getEventInfo()}.
+	 */
+	@Test
+	public void testGetEventInfo() {		
+		info.storeEventData(data1);
+		info.storeEventData(data2);
+		
+		assertEquals(2, info.getEventInfo().size());
+		assertTrue(data1 == info.getEventInfo().get("abc"));
+		assertTrue(data2 == info.getEventInfo().get("def"));
+	}
+
+}