ATLAS-3987: Atlas client export API, must pass server error code in the exception

client/common/src/main/java/org/apache/atlas/AtlasBaseClient.java

@@ -477,16 +477,16 @@ public abstract class AtlasBaseClient {
     public InputStream exportData(AtlasExportRequest request) throws AtlasServiceException {
         try {
             return (InputStream) callAPI(EXPORT, Object.class, request);
-        } catch (Exception e) {
-            LOG.error("error writing to file", e);
+        } catch (AtlasServiceException e) {
+            LOG.error("error in export API call", e);
             throw new AtlasServiceException(e);
         }
     }
 
     public void exportData(AtlasExportRequest request, String absolutePath) throws AtlasServiceException {
         OutputStream fileOutputStream = null;
+        InputStream inputStream = exportData(request);
         try {
-            InputStream inputStream = exportData(request);
             fileOutputStream = new FileOutputStream(new File(absolutePath));
             byte[] buffer = new byte[8 * 1024];
             int bytesRead;

client/common/src/main/java/org/apache/atlas/AtlasServiceException.java

@@ -41,6 +41,11 @@ public class AtlasServiceException extends Exception {
         super(e);
     }
 
+    public AtlasServiceException(AtlasServiceException e) {
+        super(e);
+        this.status = e.status;
+    }
+
     public ClientResponse.Status getStatus() {
         return status;
     }

webapp/src/test/java/org/apache/atlas/web/resources/AdminExportImportTestIT.java

@@ -19,6 +19,7 @@
 package org.apache.atlas.web.resources;
 
 
+import org.apache.atlas.AtlasClientV2;
 import org.apache.atlas.AtlasServiceException;
 import org.apache.atlas.exception.AtlasBaseException;
 import org.apache.atlas.model.impexp.AtlasExportResult;
@@ -84,6 +85,17 @@ public class AdminExportImportTestIT extends BaseResourceIT {
         assertTrue(zs.getCreationOrder().size() >= EXPECTED_CREATION_ORDER_SIZE, "expected creationOrderSize > " + EXPECTED_CREATION_ORDER_SIZE + ", but found " + zs.getCreationOrder().size());
     }
 
+    @Test
+    public void unAuthExportData() throws IOException {
+        AtlasClientV2 unAuthClient = new AtlasClientV2(atlasUrls, new String[]{"admin", "wr0ng_pa55w0rd"});
+        AtlasExportRequest request = TestResourceFileUtils.readObjectFromJson(".", EXPORT_REQUEST_FILE, AtlasExportRequest.class);
+        try {
+            InputStream exportedStream = unAuthClient.exportData(request);
+        } catch(AtlasServiceException e) {
+            assertNotNull(e.getStatus(), "expected server error code in the status");
+        }
+    }
+
     private void performImport(String fileToImport, int expectedProcessedEntitiesCount) throws AtlasServiceException {
         AtlasImportRequest request = new AtlasImportRequest();
         request.getOptions().put(AtlasImportRequest.OPTION_KEY_REPLICATED_FROM, SOURCE_SERVER_NAME);