update the testcases after the kafka 11 changes

Issue-ID: DMAAP-526
Change-Id: I477a8ee05fb3cdd76af726b6ca0d1a69aa9eef93
Signed-off-by: sunil unnava <su622b@att.com>

12 files changed, 81 insertions, 470 deletions

diff --git a/src/test/java/com/att/nsa/cambria/backends/kafka/JUnitTestSuite.java b/src/test/java/com/att/nsa/cambria/backends/kafka/JUnitTestSuite.java
index 939c0d3..54ff469 100644
--- a/src/test/java/com/att/nsa/cambria/backends/kafka/JUnitTestSuite.java
+++ b/src/test/java/com/att/nsa/cambria/backends/kafka/JUnitTestSuite.java
@@ -28,7 +28,7 @@ import org.junit.runners.Suite.SuiteClasses;
 import org.apache.log4j.Logger;

 

 @RunWith(Suite.class)

-@SuiteClasses({ KafkaConsumerTest.class, KafkaConsumerCacheTest.class, KafkaPublisherTest.class, })

+@SuiteClasses({ KafkaConsumerCacheTest.class, KafkaPublisherTest.class, })

 public class JUnitTestSuite {

 	private static final Logger LOGGER = Logger.getLogger(JUnitTestSuite.class);

 

diff --git a/src/test/java/com/att/nsa/cambria/backends/kafka/KafkaConsumerCacheTest.java b/src/test/java/com/att/nsa/cambria/backends/kafka/KafkaConsumerCacheTest.java
index 2489110..06d7b58 100644
--- a/src/test/java/com/att/nsa/cambria/backends/kafka/KafkaConsumerCacheTest.java
+++ b/src/test/java/com/att/nsa/cambria/backends/kafka/KafkaConsumerCacheTest.java
@@ -22,24 +22,58 @@ package com.att.nsa.cambria.backends.kafka;
 

 import static org.junit.Assert.*;

 

+import java.util.concurrent.ConcurrentHashMap;

+

 import org.apache.curator.framework.CuratorFramework;

 import org.junit.After;

 import org.junit.Before;

 import org.junit.Test;

-

-import com.att.nsa.cambria.backends.MetricsSet;

-import com.att.nsa.cambria.backends.kafka.KafkaConsumerCache.KafkaConsumerCacheException;

-

+import org.junit.runner.RunWith;

+import org.mockito.Mock;

+import org.mockito.MockitoAnnotations;

+import org.powermock.api.mockito.PowerMockito;

+import org.powermock.modules.junit4.PowerMockRunner;

+import org.powermock.core.classloader.annotations.PrepareForTest;

+

+import com.att.ajsc.filemonitor.AJSCPropertiesMap;

+

+import com.att.dmf.mr.backends.MetricsSet;

+import com.att.dmf.mr.backends.kafka.Kafka011Consumer;

+import com.att.dmf.mr.backends.kafka.KafkaConsumerCache;

+import com.att.dmf.mr.backends.kafka.KafkaConsumerCache.KafkaConsumerCacheException;

+import com.att.dmf.mr.constants.CambriaConstants;

+import com.att.dmf.mr.security.DMaaPAuthenticatorImpl;

+

+@RunWith(PowerMockRunner.class)

+@PrepareForTest({ AJSCPropertiesMap.class })

 public class KafkaConsumerCacheTest {

+	private KafkaConsumerCache kafkaConsumerCache =null;

+	@Mock

+	private ConcurrentHashMap<String, Kafka011Consumer> fConsumers;

+	@Mock

+	private MetricsSet fMetrics;

 

 	@Before

 	public void setUp() throws Exception {

+		MockitoAnnotations.initMocks(this);

+		

 	}

 

 	@After

 	public void tearDown() throws Exception {

 	}

 

+

+	@Test

+	public void testSweep() {

+		kafkaConsumerCache = new KafkaConsumerCache();

+		PowerMockito.mockStatic(AJSCPropertiesMap.class);

+		PowerMockito.when(AJSCPropertiesMap.getProperty(CambriaConstants.msgRtr_prop, "kSetting_TouchEveryMs")).thenReturn("100");

+		kafkaConsumerCache.sweep();

+

+	}

+	

+

 	// DOES NOT WORK

 	@Test

 	public void testStartCache() {

@@ -55,11 +89,13 @@ public class KafkaConsumerCacheTest {
 		 * e1) { // TODO Auto-generated catch block e1.printStackTrace(); } }

 		 */

 

-		KafkaConsumerCache kafka = null;

+		

 		new CuratorFrameworkImpl();

 		new MetricsSetImpl();

+		KafkaConsumerCache kafka=null;

 		try {

-			kafka = new KafkaConsumerCache("123", null);

+			kafka = new KafkaConsumerCache();

+			kafka.setfApiId("1");

 			kafka.startCache("DMAAP", null);

 		} catch (NoClassDefFoundError e) {

 

@@ -84,20 +120,17 @@ public class KafkaConsumerCacheTest {
 

 	}

 

-	/*@Test

-	public void testStopCache() {

-

-		KafkaConsumerCache kafka = null;

-		new CuratorFrameworkImpl();

-		new MetricsSetImpl();

-		try {

-			kafka = new KafkaConsumerCache("123", null);

-			kafka.stopCache();

-		} catch (NoClassDefFoundError e) {

-

-		}

-

-	}*/

+	/*

+	 * @Test public void testStopCache() {

+	 * 

+	 * KafkaConsumerCache kafka = null; new CuratorFrameworkImpl(); new

+	 * MetricsSetImpl(); try { kafka = new KafkaConsumerCache("123", null);

+	 * kafka.stopCache(); } catch (NoClassDefFoundError e) {

+	 * 

+	 * }

+	 * 

+	 * }

+	 */

 

 	@Test

 	public void testGetConsumerFor() {

@@ -105,7 +138,7 @@ public class KafkaConsumerCacheTest {
 		KafkaConsumerCache kafka = null;

 

 		try {

-			kafka = new KafkaConsumerCache("123", null);

+			kafka = new KafkaConsumerCache();

 			kafka.getConsumerFor("testTopic", "CG1", "23");

 		} catch (NoClassDefFoundError e) {

 

@@ -119,11 +152,11 @@ public class KafkaConsumerCacheTest {
 	@Test

 	public void testPutConsumerFor() {

 

-		KafkaConsumer consumer = null;

+		Kafka011Consumer consumer = null;

 		KafkaConsumerCache kafka = null;

 

 		try {

-			kafka = new KafkaConsumerCache("123", null);

+			kafka = new KafkaConsumerCache();

 

 		} catch (NoClassDefFoundError e) {

 			try {

@@ -145,7 +178,7 @@ public class KafkaConsumerCacheTest {
 		KafkaConsumerCache kafka = null;

 

 		try {

-			kafka = new KafkaConsumerCache("123", null);

+			kafka = new KafkaConsumerCache();

 

 		} catch (NoClassDefFoundError e) {

 			try {

@@ -163,7 +196,7 @@ public class KafkaConsumerCacheTest {
 

 		KafkaConsumerCache kafka = null;

 		try {

-			kafka = new KafkaConsumerCache("123", null);

+			kafka = new KafkaConsumerCache();

 

 		} catch (NoClassDefFoundError e) {

 			try {

@@ -182,7 +215,7 @@ public class KafkaConsumerCacheTest {
 		KafkaConsumerCache kafka = null;

 

 		try {

-			kafka = new KafkaConsumerCache("123", null);

+			kafka = new KafkaConsumerCache();

 			// kafka.signalOwnership("testTopic", "CG1", "23");

 		} catch (NoClassDefFoundError e) {

 			try {

@@ -192,13 +225,13 @@ public class KafkaConsumerCacheTest {
 				e.printStackTrace();

 			} catch (NullPointerException e1) {

 				// TODO Auto-generated catch block

-				//assertTrue(true);

+				// assertTrue(true);

 				e1.printStackTrace();

 			}

 

 		}

-		

-		//assertTrue(true);

+

+		// assertTrue(true);

 	}

 

 	@Test

@@ -207,7 +240,7 @@ public class KafkaConsumerCacheTest {
 		KafkaConsumerCache kafka = null;

 

 		try {

-			kafka = new KafkaConsumerCache("123", null);

+			kafka = new KafkaConsumerCache();

 			// kafka.dropConsumer("testTopic", "CG1", "23");

 		} catch (NoClassDefFoundError e) {

 			try {

diff --git a/src/test/java/com/att/nsa/cambria/backends/kafka/KafkaConsumerTest.java b/src/test/java/com/att/nsa/cambria/backends/kafka/KafkaConsumerTest.java
deleted file mode 100644
index 37e3604..0000000
--- a/src/test/java/com/att/nsa/cambria/backends/kafka/KafkaConsumerTest.java
+++ /dev/null
@@ -1,334 +0,0 @@
-/*-

- * ============LICENSE_START=======================================================

- * ONAP Policy Engine

- * ================================================================================

- * Copyright (C) 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=========================================================

- */

-

-package com.att.nsa.cambria.backends.kafka;

-

-import static org.junit.Assert.*;

-

-import java.io.IOException;

-

-import org.junit.After;

-import org.junit.Before;

-import org.junit.Test;

-

-import kafka.javaapi.consumer.ConsumerConnector;

-

-public class KafkaConsumerTest {

-

-	@Before

-	public void setUp() throws Exception {

-	}

-

-	@After

-	public void tearDown() throws Exception {

-	}

-

-	// DOES NOT WORK

-	@Test

-	public void testGetName() {

-

-		String topic = "testTopic";

-		String group = "group1";

-		String id = "1";

-		ConsumerConnector cc = null;

-		KafkaConsumer kafka = null;

-

-		try {

-			kafka = new KafkaConsumer(topic, group, id, cc);

-			

-		} catch (NullPointerException e) {

-			try {

-				kafka.getName();

-			} catch (NullPointerException e1) {

-				// TODO Auto-generated catch block

-				assertTrue(true);

-			}

-		}

-

-	}

-

-	@Test

-	public void testGetCreateTimeMs() {

-

-		String topic = "testTopic";

-		String group = "group1";

-		String id = "1";

-		ConsumerConnector cc = null;

-		

-		KafkaConsumer kafka = null;

-

-		try {

-			kafka = new KafkaConsumer(topic, group, id, cc);

-

-			

-		} catch (NullPointerException e) {

-			try {

-				kafka.getCreateTimeMs();

-			} catch (NullPointerException e1) {

-				// TODO Auto-generated catch block

-				assertTrue(true);

-			}

-		}

-

-		String trueValue = "True";

-		assertTrue(trueValue.equalsIgnoreCase("True"));

-

-	}

-

-	@Test

-	public void testGetLastAccessMs() {

-

-		String topic = "testTopic";

-		String group = "group1";

-		String id = "1";

-		ConsumerConnector cc = null;

-		

-		KafkaConsumer kafka = null;

-

-		try {

-			kafka = new KafkaConsumer(topic, group, id, cc);

-

-			

-		} catch (NullPointerException e) {

-			try {

-				kafka.getLastAccessMs();

-			} catch (NullPointerException e1) {

-				// TODO Auto-generated catch block

-				assertTrue(true);

-			}

-		}

-

-	}

-

-	@Test

-	public void testNextMessage() {

-

-		String topic = "testTopic";

-		String group = "group1";

-		String id = "1";

-		ConsumerConnector cc = null;

-		

-		KafkaConsumer kafka = null;

-

-		try {

-			kafka = new KafkaConsumer(topic, group, id, cc);

-

-			

-		} catch (NullPointerException e) {

-			try {

-				kafka.nextMessage();

-			} catch (NullPointerException e1) {

-				// TODO Auto-generated catch block

-				assertTrue(true);

-			}

-		}

-

-		String trueValue = "True";

-		assertTrue(trueValue.equalsIgnoreCase("True"));

-

-	}

-

-	@Test

-	public void testGetOffset() {

-

-		String topic = "testTopic";

-		String group = "group1";

-		String id = "1";

-		ConsumerConnector cc = null;

-		

-		KafkaConsumer kafka = null;

-

-		try {

-

-			kafka = new KafkaConsumer(topic, group, id, cc);

-

-		} catch (NullPointerException e) {

-			try {

-

-				kafka.getOffset();

-			} catch (NullPointerException e1) {

-				// TODO Auto-generated catch block

-				assertTrue(true);

-			}

-		}

-

-	}

-

-	@Test

-	public void testCommitOffsets() {

-

-		String topic = "testTopic";

-		String group = "group1";

-		String id = "1";

-		ConsumerConnector cc = null;

-		

-		KafkaConsumer kafka = null;

-

-		try {

-

-			kafka = new KafkaConsumer(topic, group, id, cc);

-

-			

-		} catch (NullPointerException e) {

-			try {

-				kafka.commitOffsets();

-			} catch (NullPointerException e1) {

-				// TODO Auto-generated catch block

-				assertTrue(true);

-			}

-		}

-

-	}

-

-	@Test

-	public void testTouch() {

-

-		String topic = "testTopic";

-		String group = "group1";

-		String id = "1";

-		ConsumerConnector cc = null;

-		

-		KafkaConsumer kafka = null;

-

-		try {

-

-			kafka = new KafkaConsumer(topic, group, id, cc);

-

-			

-		} catch (NullPointerException e) {

-			try {

-				kafka.touch();

-			} catch (NullPointerException e1) {

-				// TODO Auto-generated catch block

-				assertTrue(true);

-			}

-		}

-

-		String trueValue = "True";

-		assertTrue(trueValue.equalsIgnoreCase("True"));

-

-	}

-

-	@Test

-	public void testGetLastTouch() {

-

-		String topic = "testTopic";

-		String group = "group1";

-		String id = "1";

-		ConsumerConnector cc = null;

-		

-		KafkaConsumer kafka = null;

-

-		try {

-

-			kafka = new KafkaConsumer(topic, group, id, cc);

-

-			

-		} catch (NullPointerException e) {

-			try {

-				kafka.getLastTouch();

-			} catch (NullPointerException e1) {

-				// TODO Auto-generated catch block

-				assertTrue(true);

-			}

-		}

-

-	}

-

-	@Test

-	public void testClose() {

-

-		String topic = "testTopic";

-		String group = "group1";

-		String id = "1";

-		ConsumerConnector cc = null;

-		

-		KafkaConsumer kafka = null;

-

-		try {

-

-			kafka = new KafkaConsumer(topic, group, id, cc);

-

-			

-		} catch (NullPointerException e) {

-			try {

-				kafka.close();

-			} catch (NullPointerException e1) {

-				// TODO Auto-generated catch block

-				assertTrue(true);

-			}

-		}

-

-	}

-

-	@Test

-	public void testGetConsumerGroup() {

-

-		String topic = "testTopic";

-		String group = "group1";

-		String id = "1";

-		ConsumerConnector cc = null;

-		

-		KafkaConsumer kafka = null;

-

-		try {

-

-			kafka = new KafkaConsumer(topic, group, id, cc);

-

-			

-		} catch (NullPointerException e) {

-			try {

-				kafka.getConsumerGroup();

-			} catch (NullPointerException e1) {

-				// TODO Auto-generated catch block

-				assertTrue(true);

-			}

-		}

-

-	}

-

-	@Test

-	public void testGetConsumerId() {

-

-		String topic = "testTopic";

-		String group = "group1";

-		String id = "1";

-		ConsumerConnector cc = null;

-		

-		KafkaConsumer kafka = null;

-

-		try {

-

-			kafka = new KafkaConsumer(topic, group, id, cc);

-

-			

-

-		} catch (NullPointerException e) {

-			try {

-				kafka.getConsumerId();

-			} catch (NullPointerException e1) {

-				// TODO Auto-generated catch block

-				assertTrue(true);

-			}

-		}

-

-	}

-

-}

diff --git a/src/test/java/com/att/nsa/cambria/backends/kafka/KafkaPublisherTest.java b/src/test/java/com/att/nsa/cambria/backends/kafka/KafkaPublisherTest.java
index 465c66a..3673845 100644
--- a/src/test/java/com/att/nsa/cambria/backends/kafka/KafkaPublisherTest.java
+++ b/src/test/java/com/att/nsa/cambria/backends/kafka/KafkaPublisherTest.java
@@ -33,7 +33,7 @@ import org.junit.ClassRule;
 import org.junit.Test;

 

 import com.att.ajsc.filemonitor.AJSCPropertiesMap;

-import com.att.nsa.cambria.backends.Publisher.message;

+import com.att.dmf.mr.backends.Publisher.message;

 import com.att.nsa.drumlin.till.nv.rrNvReadable.missingReqdSetting;

 

 import kafka.common.FailedToSendMessageException;

diff --git a/src/test/java/com/att/nsa/cambria/backends/kafka/MetricsSetImpl.java b/src/test/java/com/att/nsa/cambria/backends/kafka/MetricsSetImpl.java
index a32ee98..b5f7b74 100644
--- a/src/test/java/com/att/nsa/cambria/backends/kafka/MetricsSetImpl.java
+++ b/src/test/java/com/att/nsa/cambria/backends/kafka/MetricsSetImpl.java
@@ -25,7 +25,7 @@ import java.util.Map;
 

 import org.json.JSONObject;

 

-import com.att.nsa.cambria.backends.MetricsSet;

+import com.att.dmf.mr.backends.MetricsSet;

 import com.att.nsa.metrics.CdmMeasuredItem;

 

 public class MetricsSetImpl implements MetricsSet {

diff --git a/src/test/java/com/att/nsa/cambria/backends/memory/JUnitTestSuite.java b/src/test/java/com/att/nsa/cambria/backends/memory/JUnitTestSuite.java
index ba5a7c6..fc11552 100644
--- a/src/test/java/com/att/nsa/cambria/backends/memory/JUnitTestSuite.java
+++ b/src/test/java/com/att/nsa/cambria/backends/memory/JUnitTestSuite.java
@@ -29,7 +29,7 @@ import org.apache.log4j.Logger;
 

 @RunWith(Suite.class)

 @SuiteClasses({ MemoryConsumerFactoryTest.class, MemoryMetaBrokerTest.class, MemoryQueueTest.class, 

-	MemoryQueuePublisherTest.class, MessageDropperTest.class, MessageLoggerTest.class, })

+	MemoryQueuePublisherTest.class,  MessageLoggerTest.class, })

 public class JUnitTestSuite {

 	private static final Logger LOGGER = Logger.getLogger(JUnitTestSuite.class);

 

diff --git a/src/test/java/com/att/nsa/cambria/backends/memory/MemoryConsumerFactoryTest.java b/src/test/java/com/att/nsa/cambria/backends/memory/MemoryConsumerFactoryTest.java
index 7b470bc..d6fe6e6 100644
--- a/src/test/java/com/att/nsa/cambria/backends/memory/MemoryConsumerFactoryTest.java
+++ b/src/test/java/com/att/nsa/cambria/backends/memory/MemoryConsumerFactoryTest.java
@@ -28,6 +28,8 @@ import org.junit.After;
 import org.junit.Before;

 import org.junit.Test;

 

+import com.att.dmf.mr.backends.memory.MemoryConsumerFactory;

+

 public class MemoryConsumerFactoryTest {

 

 	@Before

@@ -46,8 +48,9 @@ public class MemoryConsumerFactoryTest {
 		String topic = "testTopic";

 		String consumerGroupId = "CG1";

 		String clientId = "C1";

+		String remoteHost="remoteHost";

 		int timeoutMs = 1000; 

-		factory.getConsumerFor(topic, consumerGroupId, clientId, timeoutMs);

+		factory.getConsumerFor(topic, consumerGroupId, clientId, timeoutMs,remoteHost);

 		

 		String trueValue = "True";

 		assertTrue(trueValue.equalsIgnoreCase("True"));

diff --git a/src/test/java/com/att/nsa/cambria/backends/memory/MemoryMetaBrokerTest.java b/src/test/java/com/att/nsa/cambria/backends/memory/MemoryMetaBrokerTest.java
index b15cb2f..2c41068 100644
--- a/src/test/java/com/att/nsa/cambria/backends/memory/MemoryMetaBrokerTest.java
+++ b/src/test/java/com/att/nsa/cambria/backends/memory/MemoryMetaBrokerTest.java
@@ -28,7 +28,8 @@ import org.junit.After;
 import org.junit.Before;

 import org.junit.Test;

 

-import com.att.nsa.cambria.metabroker.Broker.TopicExistsException;

+import com.att.dmf.mr.backends.memory.MemoryMetaBroker;

+import com.att.dmf.mr.metabroker.Broker.TopicExistsException;

 

 public class MemoryMetaBrokerTest {

 

diff --git a/src/test/java/com/att/nsa/cambria/backends/memory/MemoryQueuePublisherTest.java b/src/test/java/com/att/nsa/cambria/backends/memory/MemoryQueuePublisherTest.java
index 8ee4bd1..4522734 100644
--- a/src/test/java/com/att/nsa/cambria/backends/memory/MemoryQueuePublisherTest.java
+++ b/src/test/java/com/att/nsa/cambria/backends/memory/MemoryQueuePublisherTest.java
@@ -28,6 +28,8 @@ import org.junit.After;
 import org.junit.Before;

 import org.junit.Test;

 

+import com.att.dmf.mr.backends.memory.MemoryQueuePublisher;

+

 

 public class MemoryQueuePublisherTest {

 

@@ -44,7 +46,7 @@ public class MemoryQueuePublisherTest {
 		MemoryQueuePublisher publisher = new MemoryQueuePublisher(null, null);

 		

 		try {

-			publisher.sendBatchMessage("testTopic", null);

+			publisher.sendBatchMessageNew("testTopic", null);

 		} catch (IOException e) {

 			// TODO Auto-generated catch block

 			e.printStackTrace();

diff --git a/src/test/java/com/att/nsa/cambria/backends/memory/MemoryQueueTest.java b/src/test/java/com/att/nsa/cambria/backends/memory/MemoryQueueTest.java
index fdb84d0..637add2 100644
--- a/src/test/java/com/att/nsa/cambria/backends/memory/MemoryQueueTest.java
+++ b/src/test/java/com/att/nsa/cambria/backends/memory/MemoryQueueTest.java
@@ -27,6 +27,8 @@ import org.junit.After;
 import org.junit.Before;

 import org.junit.Test;

 

+import com.att.dmf.mr.backends.memory.MemoryQueue;

+

 

 public class MemoryQueueTest {

 

diff --git a/src/test/java/com/att/nsa/cambria/backends/memory/MessageDropperTest.java b/src/test/java/com/att/nsa/cambria/backends/memory/MessageDropperTest.java
deleted file mode 100644
index c27621e..0000000
--- a/src/test/java/com/att/nsa/cambria/backends/memory/MessageDropperTest.java
+++ /dev/null
@@ -1,98 +0,0 @@
-/*-

- * ============LICENSE_START=======================================================

- * ONAP Policy Engine

- * ================================================================================

- * Copyright (C) 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=========================================================

- */

-

-package com.att.nsa.cambria.backends.memory;

-

-import static org.junit.Assert.*;

-

-import java.io.IOException;

-import java.util.ArrayList;

-import java.util.List;

-

-import org.junit.After;

-import org.junit.Before;

-import org.junit.Test;

-

-import com.att.nsa.cambria.backends.Publisher.message;

-import com.att.nsa.cambria.metabroker.Broker.TopicExistsException;

-

-import kafka.producer.KeyedMessage;

-

-public class MessageDropperTest {

-

-	@Before

-	public void setUp() throws Exception {

-	}

-

-	@After

-	public void tearDown() throws Exception {

-	}

-

-	@Test

-	public void testSendMessage() {

-		MessageDropper dropper = new MessageDropper();

-		

-		try {

-			dropper.sendMessage("testTopic", null);

-		} catch (IOException e) {

-			// TODO Auto-generated catch block

-			e.printStackTrace();

-		}

-		

-		String trueValue = "True";

-		assertTrue(trueValue.equalsIgnoreCase("True"));

-		

-	}

-	

-	@Test

-	public void testSendMessages() {

-		MessageDropper dropper = new MessageDropper();

-		 

-		try {

-			dropper.sendMessages("testTopic", null);

-		} catch (IOException e) {

-			// TODO Auto-generated catch block

-			e.printStackTrace();

-		}

-		

-		String trueValue = "True";

-		assertTrue(trueValue.equalsIgnoreCase("True"));

-		

-	}

-	

-	

-	@Test

-	public void testSendBatchMessage() {

-		MessageDropper dropper = new MessageDropper();

-		

-		try {

-			dropper.sendBatchMessage("testTopic", null);

-		} catch (IOException e) {

-			// TODO Auto-generated catch block

-			e.printStackTrace();

-		}

-		

-		String trueValue = "True";

-		assertTrue(trueValue.equalsIgnoreCase("True"));

-		

-	}

-	

-

-}

diff --git a/src/test/java/com/att/nsa/cambria/backends/memory/MessageLoggerTest.java b/src/test/java/com/att/nsa/cambria/backends/memory/MessageLoggerTest.java
index d2ae163..81e620c 100644
--- a/src/test/java/com/att/nsa/cambria/backends/memory/MessageLoggerTest.java
+++ b/src/test/java/com/att/nsa/cambria/backends/memory/MessageLoggerTest.java
@@ -28,6 +28,8 @@ import org.junit.After;
 import org.junit.Before;

 import org.junit.Test;

 

+import com.att.dmf.mr.backends.memory.MessageLogger;

+

 

 public class MessageLoggerTest {

 

@@ -82,7 +84,7 @@ public class MessageLoggerTest {
 		MessageLogger dropper = new MessageLogger();

 		

 		try {

-			dropper.sendBatchMessage("testTopic", null);

+			dropper.sendBatchMessageNew("testTopic", null);

 		} catch (IOException e) {

 			// TODO Auto-generated catch block

 			e.printStackTrace();