Repository: atlas Updated Branches: refs/heads/master c0a91c7e1 -> 8746b3063
http://git-wip-us.apache.org/repos/asf/atlas/blob/116fb62c/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
