From 17d0fa9e467c46fee4e746e1e70b05b3e2954ca2 Mon Sep 17 00:00:00 2001
From: cptran777 <catran@linkedin.com>
Date: Fri, 31 Aug 2018 01:30:58 -0700
Subject: [PATCH] Add proper reset for dataset export policy update

---
 .../datasets/containers/dataset-compliance.ts   | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/wherehows-web/app/components/datasets/containers/dataset-compliance.ts b/wherehows-web/app/components/datasets/containers/dataset-compliance.ts
index 57c8e33923..816d175062 100644
--- a/wherehows-web/app/components/datasets/containers/dataset-compliance.ts
+++ b/wherehows-web/app/components/datasets/containers/dataset-compliance.ts
@@ -7,7 +7,8 @@ import { IDatasetColumn } from 'wherehows-web/typings/api/datasets/columns';
 import {
   IComplianceInfo,
   IComplianceSuggestion,
-  IDatasetExportPolicy
+  IDatasetExportPolicy,
+  IDatasetExportPolicyResponse
 } from 'wherehows-web/typings/api/datasets/compliance';
 import { IDatasetView } from 'wherehows-web/typings/api/datasets/dataset';
 import { IDatasetSchema } from 'wherehows-web/typings/api/datasets/schema';
@@ -248,6 +249,18 @@ export default class DatasetComplianceContainer extends Component {
     setProperties(this, { isNewComplianceInfo, complianceInfo });
   });
 
+  /**
+   * Reads the export policy properties for the dataset
+   * @type {Task<Promise<IReadComplianceResult>, (a?: any) => TaskInstance<Promise<IDatasetExportPolicy>>>}
+   */
+  getExportPolicyTask = task(function*(
+    this: DatasetComplianceContainer
+  ): IterableIterator<Promise<IDatasetExportPolicy>> {
+    const exportPolicy: IDatasetExportPolicy = yield readDatasetExportPolicyByUrn(get(this, 'urn'));
+
+    set(this, 'exportPolicy', exportPolicy);
+  });
+
   /**
    * Reads the compliance data types
    * @type {Task<Promise<Array<IComplianceDataType>>, (a?: any) => TaskInstance<Promise<Array<IComplianceDataType>>>>}
@@ -341,7 +354,7 @@ export default class DatasetComplianceContainer extends Component {
   async saveExportPolicy(this: DatasetComplianceContainer, exportPolicy: IDatasetExportPolicy): Promise<void> {
     await this.notifyOnSave<void>(saveDatasetExportPolicyByUrn(get(this, 'urn'), exportPolicy));
 
-    this.resetPrivacyCompliancePolicy.call(this);
+    this.getExportPolicyTask.perform();
   }
 
   /**