resolves #2

OpenPOWER-BigData · Oct 26, 2016 · 1b99a64 · 1b99a64
1 parent 7e621ed
commit 1b99a64
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/...rver-extensions/src/test/java/org/apache/hive/hcatalog/listener/TestMsgBusConnection.java b/...rver-extensions/src/test/java/org/apache/hive/hcatalog/listener/TestMsgBusConnection.java
@@ -48,7 +48,7 @@ public class TestMsgBusConnection extends TestCase {
   private Driver driver;
   private BrokerService broker;
   private MessageConsumer consumer;
-
+  private static final int TIMEOUT = 2000;
   @Override
   protected void setUp() throws Exception {
 
@@ -88,7 +88,7 @@ public void testConnection() throws Exception {
 
     try {
       driver.run("create database testconndb");
-      Message msg = consumer.receive();
+      Message msg = consumer.receive(TIMEOUT);
       assertTrue("Expected TextMessage", msg instanceof TextMessage);
       assertEquals(HCatConstants.HCAT_CREATE_DATABASE_EVENT,
           msg.getStringProperty(HCatConstants.HCAT_EVENT));
@@ -100,13 +100,13 @@ public void testConnection() throws Exception {
       broker.start(true);
       connectClient();
       driver.run("create database testconndb");
-      msg = consumer.receive();
+      msg = consumer.receive(TIMEOUT);
       assertEquals(HCatConstants.HCAT_CREATE_DATABASE_EVENT,
           msg.getStringProperty(HCatConstants.HCAT_EVENT));
       assertEquals("topic://planetlab.hcat", msg.getJMSDestination().toString());
       assertEquals("testconndb", messageObject.getDB());
       driver.run("drop database testconndb cascade");
-      msg = consumer.receive();
+      msg = consumer.receive(TIMEOUT);
       assertEquals(HCatConstants.HCAT_DROP_DATABASE_EVENT,
           msg.getStringProperty(HCatConstants.HCAT_EVENT));
       assertEquals("topic://planetlab.hcat", msg.getJMSDestination().toString());