http://git-wip-us.apache.org/repos/asf/atlas/blob/2a89b089/webapp/src/test/java/org/apache/atlas/web/resources/AdminExportImportTestIT.java
----------------------------------------------------------------------
diff --git 
a/webapp/src/test/java/org/apache/atlas/web/resources/AdminExportImportTestIT.java
 
b/webapp/src/test/java/org/apache/atlas/web/resources/AdminExportImportTestIT.java
index d156054..babf275 100644
--- 
a/webapp/src/test/java/org/apache/atlas/web/resources/AdminExportImportTestIT.java
+++ 
b/webapp/src/test/java/org/apache/atlas/web/resources/AdminExportImportTestIT.java
@@ -21,7 +21,7 @@ package org.apache.atlas.web.resources;
 
 import org.apache.atlas.AtlasServiceException;
 import org.apache.atlas.exception.AtlasBaseException;
-import org.apache.atlas.model.impexp.AtlasCluster;
+import org.apache.atlas.model.impexp.AtlasServer;
 import org.apache.atlas.model.impexp.AtlasExportRequest;
 import org.apache.atlas.model.impexp.AtlasImportRequest;
 import org.apache.atlas.model.impexp.AtlasImportResult;
@@ -44,10 +44,10 @@ import static org.testng.Assert.assertTrue;
 public class AdminExportImportTestIT extends BaseResourceIT {
     private final String FILE_TO_IMPORT = "stocks-base.zip";
     private final String EXPORT_REQUEST_FILE = "export-incremental";
-    private final String SOURCE_CLUSTER_NAME = "cl1";
+    private final String SOURCE_SERVER_NAME = "cl1";
 
     static final String IMPORT_TRANSFORM_CLEAR_ATTRS =
-            "{ \"Asset\": { \"*\":[ 
\"clearAttrValue:replicatedToCluster,replicatedFromCluster\" ] } }";
+            "{ \"Asset\": { \"*\":[ 
\"clearAttrValue:replicatedTo,replicatedFrom\" ] } }";
     static final String IMPORT_TRANSFORM_SET_DELETED =
             "{ \"Asset\": { \"*\":[ \"setDeleted\" ] } }";
 
@@ -77,7 +77,7 @@ public class AdminExportImportTestIT extends BaseResourceIT {
 
     private void performImport(String fileToImport) throws 
AtlasServiceException {
         AtlasImportRequest request = new AtlasImportRequest();
-        
request.getOptions().put(AtlasImportRequest.OPTION_KEY_REPLICATED_FROM, 
SOURCE_CLUSTER_NAME);
+        
request.getOptions().put(AtlasImportRequest.OPTION_KEY_REPLICATED_FROM, 
SOURCE_SERVER_NAME);
         request.getOptions().put(AtlasImportRequest.TRANSFORMS_KEY, 
IMPORT_TRANSFORM_CLEAR_ATTRS);
 
         performImport(fileToImport, request);
@@ -100,11 +100,11 @@ public class AdminExportImportTestIT extends 
BaseResourceIT {
         assertEquals(result.getProcessedEntities().size(), 37);
     }
 
-    private void assertReplicationData(String clusterName) throws 
AtlasServiceException {
-        AtlasCluster cluster = atlasClientV2.getCluster(clusterName);
-        assertNotNull(cluster);
-        assertNotNull(cluster.getAdditionalInfo());
-        assertTrue(cluster.getAdditionalInfo().size() > 0);
+    private void assertReplicationData(String serverName) throws 
AtlasServiceException {
+        AtlasServer server = atlasClientV2.getServer(serverName);
+        assertNotNull(server);
+        assertNotNull(server.getAdditionalInfo());
+        assertTrue(server.getAdditionalInfo().size() > 0);
     }
 
     @AfterClass

Reply via email to