commit 537f6e31f00f3406a6eb1a03a0f1223ce74ba9a4
author nixonrodrigues <nixon.rodrigues@freestoneinfotech.com> Sat Mar 18 17:17:13 2017 +0530
committer Madhan Neethiraj <madhan@apache.org> Sat Mar 18 09:21:21 2017 -0700
tree fe04be7e11213d362fa7eb052615d753f3d43d94
parent bec9370e5807f73e7dfc8e72da42d06557a745b4

ATLAS-1666: fix for IT failures caused by earlier commit

Signed-off-by: Madhan Neethiraj <madhan@apache.org>

webapp/src/test/java/org/apache/atlas/web/service/ActiveInstanceElectorServiceTest.java

diff --git a/webapp/src/test/java/org/apache/atlas/web/service/ActiveInstanceElectorServiceTest.java b/webapp/src/test/java/org/apache/atlas/web/service/ActiveInstanceElectorServiceTest.java
index c0ffc6e..d1b0318 100644
--- a/webapp/src/test/java/org/apache/atlas/web/service/ActiveInstanceElectorServiceTest.java
+++ b/webapp/src/test/java/org/apache/atlas/web/service/ActiveInstanceElectorServiceTest.java
@@ -34,7 +34,7 @@
 import java.io.IOException;
 import java.util.ArrayList;
 import java.util.Collection;
-
+import org.apache.atlas.exception.AtlasBaseException;
 import static org.mockito.Mockito.doThrow;
 import static org.mockito.Mockito.inOrder;
 import static org.mockito.Mockito.mock;
@@ -268,7 +268,7 @@
             }
         });
 
-        doThrow(new Exception()).when(activeInstanceState).update("id1");
+        doThrow(new AtlasBaseException()).when(activeInstanceState).update("id1");
 
         ActiveInstanceElectorService activeInstanceElectorService =
                 new ActiveInstanceElectorService(configuration, changeHandlers, curatorFactory,
@@ -294,7 +294,7 @@
         LeaderLatch leaderLatch = mock(LeaderLatch.class);
         when(curatorFactory.leaderLatchInstance("id1", HAConfiguration.ATLAS_SERVER_ZK_ROOT_DEFAULT)).thenReturn(leaderLatch);
 
-        doThrow(new Exception()).when(activeInstanceState).update("id1");
+        doThrow(new AtlasBaseException()).when(activeInstanceState).update("id1");
 
         ActiveInstanceElectorService activeInstanceElectorService =
                 new ActiveInstanceElectorService(configuration, new ArrayList(), curatorFactory,
@@ -392,7 +392,7 @@
         LeaderLatch leaderLatch = mock(LeaderLatch.class);
         when(curatorFactory.leaderLatchInstance("id1", HAConfiguration.ATLAS_SERVER_ZK_ROOT_DEFAULT)).thenReturn(leaderLatch);
 
-        doThrow(new Exception()).when(activeInstanceState).update("id1");
+        doThrow(new AtlasBaseException()).when(activeInstanceState).update("id1");
 
         ActiveInstanceElectorService activeInstanceElectorService =
                 new ActiveInstanceElectorService(configuration, new ArrayList(),