From b03c9379b8190646dac4cf01f47b1bedb5ccde56 Mon Sep 17 00:00:00 2001
From: cptran777 <catran@linkedin.com>
Date: Fri, 31 Aug 2018 01:37:17 -0700
Subject: [PATCH] Change export policy api POST to return updated information
 upon success

---
 .../src/main/java/wherehows/dao/table/ExportPolicyDao.java    | 2 +-
 wherehows-frontend/app/controllers/api/v2/Dataset.java        | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/wherehows-dao/src/main/java/wherehows/dao/table/ExportPolicyDao.java b/wherehows-dao/src/main/java/wherehows/dao/table/ExportPolicyDao.java
index 35db42b32c..451e762bec 100644
--- a/wherehows-dao/src/main/java/wherehows/dao/table/ExportPolicyDao.java
+++ b/wherehows-dao/src/main/java/wherehows/dao/table/ExportPolicyDao.java
@@ -25,7 +25,7 @@ public class ExportPolicyDao {
     throw new UnsupportedOperationException("Not implemented yet");
   }
 
-  public void updateDatasetExportPolicy(@Nonnull DatasetExportPolicy record, @Nonnull String user) throws Exception {
+  public void updateDatasetExportPolicy(@Nonnull datasetUrn, @Nonnull DatasetExportPolicy record, @Nonnull String user) throws Exception {
     throw new UnsupportedOperationException("Not implemented yet");
 
   }
diff --git a/wherehows-frontend/app/controllers/api/v2/Dataset.java b/wherehows-frontend/app/controllers/api/v2/Dataset.java
index dfe1ef42c0..60b9de94b2 100644
--- a/wherehows-frontend/app/controllers/api/v2/Dataset.java
+++ b/wherehows-frontend/app/controllers/api/v2/Dataset.java
@@ -280,13 +280,13 @@ public class Dataset extends Controller {
     try {
       JsonNode exportPolicy = request().body().asJson();
 
-      EXPORT_POLICY_DAO.updateDatasetExportPolicy(exportPolicy, username);
+      EXPORT_POLICY_DAO.updateDatasetExportPolicy(datasetUrn, exportPolicy, username);
     } catch (Exception e) {
       Logger.error("Update dataset export policy fail", e);
       return Promise.promise(() -> internalServerError(errorResponse(e)));
     }
 
-    return Promise.promise(() -> ok(_EMPTY_RESPONSE));
+    return Promise.promise(() -> ok(Json.newObject().set("exportPolicy", Json.toJson(exportPolicy))));
   }
 
   public static Promise<Result> getDatasetOwners(String datasetUrn) {