commit 96d9a0ec283aacaf05ddbfa504ef4f06d133d92e
author Yussuf Shaikh <yussuf@us.ibm.com> Tue May 09 14:53:27 2017 -0700
committer Sarath Subramanian <ssubramanian@hortonworks.com> Tue May 09 14:53:27 2017 -0700
tree 2b3f59e1cd4908af44cbd1d7071b56f0c5338e5a
parent 235f4e9e2d761812976515a0853169d5cb934494

ATLAS-1770: Fix HardDeleteHandlerV1Test and SoftDeleteHandlerV1Test failures during setup

Signed-off-by: Sarath Subramanian <ssubramanian@hortonworks.com>

repository/src/test/java/org/apache/atlas/repository/impexp/ExportServiceTest.java

diff --git a/repository/src/test/java/org/apache/atlas/repository/impexp/ExportServiceTest.java b/repository/src/test/java/org/apache/atlas/repository/impexp/ExportServiceTest.java
index cd355c3..13452ca 100644
--- a/repository/src/test/java/org/apache/atlas/repository/impexp/ExportServiceTest.java
+++ b/repository/src/test/java/org/apache/atlas/repository/impexp/ExportServiceTest.java
@@ -27,6 +27,7 @@
 import org.apache.atlas.model.instance.AtlasEntity;
 import org.apache.atlas.model.instance.AtlasObjectId;
 import org.apache.atlas.model.typedef.AtlasTypesDef;
+import org.apache.atlas.repository.graph.AtlasGraphProvider;
 import org.apache.atlas.repository.store.bootstrap.AtlasTypeDefStoreInitializer;
 import org.apache.atlas.repository.store.graph.v1.AtlasEntityChangeNotifier;
 import org.apache.atlas.repository.store.graph.v1.AtlasEntityStoreV1;
@@ -39,6 +40,7 @@
 import org.slf4j.LoggerFactory;
 import org.testng.Assert;
 import org.powermock.reflect.Whitebox;
+import org.testng.annotations.AfterClass;
 import org.testng.annotations.BeforeClass;
 import org.testng.annotations.BeforeTest;
 import org.testng.annotations.Guice;
@@ -95,6 +97,11 @@
         exportService = new ExportService(typeRegistry);
     }
 
+    @AfterClass
+    public void clear() {
+        AtlasGraphProvider.cleanup();
+    }
+
     private AtlasExportRequest getRequestForFullFetch() {
         AtlasExportRequest request = new AtlasExportRequest();