From 0e00abf5b3bfaab7cc2afb8ef6c0eb34f35e7e45 Mon Sep 17 00:00:00 2001
From: yuboluo <yubonluo@amazon.com>
Date: Fri, 18 Oct 2024 20:52:22 +0800
Subject: [PATCH] [Workspace] Workspace assets page crashed after library_write
 permission revoked (#8648)

* fix copy issue

Signed-off-by: yubonluo <yubonluo@amazon.com>

* Changeset file for PR #8648 created/updated

* fix: snapshot

Signed-off-by: SuZhou-Joe <suzhou@amazon.com>

---------

Signed-off-by: yubonluo <yubonluo@amazon.com>
Signed-off-by: SuZhou-Joe <suzhou@amazon.com>
Co-authored-by: opensearch-changeset-bot[bot] <154024398+opensearch-changeset-bot[bot]@users.noreply.github.com>
Co-authored-by: SuZhou-Joe <suzhou@amazon.com>
---
 changelogs/fragments/8648.yml                 |  2 ++
 .../duplicate_result_flyout.test.tsx.snap     | 20 +++++++----
 .../components/duplicate_result_flyout.tsx    | 12 ++++---
 .../saved_objects_table.test.tsx              | 33 ++++++++++++++++++-
 .../objects_table/saved_objects_table.tsx     | 25 ++++++++------
 5 files changed, 70 insertions(+), 22 deletions(-)
 create mode 100644 changelogs/fragments/8648.yml

diff --git a/changelogs/fragments/8648.yml b/changelogs/fragments/8648.yml
new file mode 100644
index 000000000000..84cc02108e2d
--- /dev/null
+++ b/changelogs/fragments/8648.yml
@@ -0,0 +1,2 @@
+fix:
+- [Workspace] Workspace assets page crashed after library_write permission revoked ([#8648](https://github.com/opensearch-project/OpenSearch-Dashboards/pull/8648))
\ No newline at end of file
diff --git a/src/plugins/saved_objects_management/public/management_section/objects_table/components/__snapshots__/duplicate_result_flyout.test.tsx.snap b/src/plugins/saved_objects_management/public/management_section/objects_table/components/__snapshots__/duplicate_result_flyout.test.tsx.snap
index 39270a4557ed..9501523a3755 100644
--- a/src/plugins/saved_objects_management/public/management_section/objects_table/components/__snapshots__/duplicate_result_flyout.test.tsx.snap
+++ b/src/plugins/saved_objects_management/public/management_section/objects_table/components/__snapshots__/duplicate_result_flyout.test.tsx.snap
@@ -142,9 +142,13 @@ HTMLCollection [
           <div
             class="euiFlyoutHeader euiFlyoutHeader--hasBorder"
           >
-            <span>
-              Copy saved objects to targetWorkspace
-            </span>
+            <h2
+              class="euiTitle euiTitle--small"
+            >
+              <span>
+                Copy saved objects to targetWorkspace
+              </span>
+            </h2>
           </div>
           <div
             class="euiFlyoutBody"
@@ -342,9 +346,13 @@ HTMLCollection [
           <div
             class="euiFlyoutHeader euiFlyoutHeader--hasBorder"
           >
-            <span>
-              Copy saved objects to targetWorkspace
-            </span>
+            <h2
+              class="euiTitle euiTitle--small"
+            >
+              <span>
+                Copy saved objects to targetWorkspace
+              </span>
+            </h2>
           </div>
           <div
             class="euiFlyoutBody"
diff --git a/src/plugins/saved_objects_management/public/management_section/objects_table/components/duplicate_result_flyout.tsx b/src/plugins/saved_objects_management/public/management_section/objects_table/components/duplicate_result_flyout.tsx
index 2bca3d33ffe8..0698a55b6d52 100644
--- a/src/plugins/saved_objects_management/public/management_section/objects_table/components/duplicate_result_flyout.tsx
+++ b/src/plugins/saved_objects_management/public/management_section/objects_table/components/duplicate_result_flyout.tsx
@@ -220,11 +220,13 @@ export class DuplicateResultFlyout extends React.Component<DuplicateResultFlyout
       <EuiFlyout ownFocus onClose={onClose} size="s">
         <EuiFlyoutHeader hasBorder>
           <EuiTitle size="s">
-            <FormattedMessage
-              id="savedObjectsManagement.copyResult.title"
-              defaultMessage="Copy saved objects to {workspaceName}"
-              values={{ workspaceName }}
-            />
+            <h2>
+              <FormattedMessage
+                id="savedObjectsManagement.copyResult.title"
+                defaultMessage="Copy saved objects to {workspaceName}"
+                values={{ workspaceName }}
+              />
+            </h2>
           </EuiTitle>
         </EuiFlyoutHeader>
         <EuiFlyoutBody>{this.copyResult({ failedCopies, successfulCopies })}</EuiFlyoutBody>
diff --git a/src/plugins/saved_objects_management/public/management_section/objects_table/saved_objects_table.test.tsx b/src/plugins/saved_objects_management/public/management_section/objects_table/saved_objects_table.test.tsx
index df7bec839a90..b3e440460428 100644
--- a/src/plugins/saved_objects_management/public/management_section/objects_table/saved_objects_table.test.tsx
+++ b/src/plugins/saved_objects_management/public/management_section/objects_table/saved_objects_table.test.tsx
@@ -1021,7 +1021,38 @@ describe('SavedObjectsTable', () => {
       });
     });
 
-    it('should catch error when duplicating selected object is fail', async () => {
+    it('should catch error when duplicating selected object is failed', async () => {
+      const component = shallowRender({ applications, workspaces });
+      component.setState({ isShowingDuplicateModal: true });
+
+      const mockCopy = jest.fn().mockResolvedValue({ error: 'error' });
+      workspaces.client$.next({ copy: mockCopy });
+      const client = workspaces.client$.getValue();
+
+      // Ensure all promises resolve
+      await new Promise((resolve) => process.nextTick(resolve));
+      // Ensure the state changes are reflected
+      component.update();
+
+      await component.instance().onDuplicate(mockSelectedSavedObjects, false, 'workspace2', 'bar');
+
+      expect(client?.copy).toHaveBeenCalledWith(
+        [
+          { id: '1', type: 'dashboard' },
+          { id: '2', type: 'dashboard' },
+        ],
+        'workspace2',
+        false
+      );
+      component.update();
+
+      expect(notifications.toasts.addDanger).toHaveBeenCalledWith({
+        title: 'Unable to copy 2 saved objects.',
+        text: 'error',
+      });
+    });
+
+    it('should show error toast when copy is fail', async () => {
       const component = shallowRender({ applications, workspaces });
       component.setState({ isShowingDuplicateModal: true });
 
diff --git a/src/plugins/saved_objects_management/public/management_section/objects_table/saved_objects_table.tsx b/src/plugins/saved_objects_management/public/management_section/objects_table/saved_objects_table.tsx
index 3ecfd103488b..8c77f9dd3eff 100644
--- a/src/plugins/saved_objects_management/public/management_section/objects_table/saved_objects_table.tsx
+++ b/src/plugins/saved_objects_management/public/management_section/objects_table/saved_objects_table.tsx
@@ -761,13 +761,14 @@ export class SavedObjectsTable extends Component<SavedObjectsTableProps, SavedOb
     const { notifications, workspaces, useUpdatedUX } = this.props;
     const workspaceClient = workspaces.client$.getValue();
 
-    const showErrorNotification = () => {
+    const showErrorNotification = (text?: string) => {
       notifications.toasts.addDanger({
         title: i18n.translate('savedObjectsManagement.objectsTable.duplicate.dangerNotification', {
           defaultMessage:
             'Unable to copy {useUpdatedUX, select, true {{errorCount, plural, one {# asset} other {# assets}}} other {{errorCount, plural, one {# saved object} other {# saved objects}}}}.',
           values: { errorCount: savedObjects.length, useUpdatedUX },
         }),
+        ...(text && { text }),
       });
     };
     if (!workspaceClient) {
@@ -781,18 +782,22 @@ export class SavedObjectsTable extends Component<SavedObjectsTableProps, SavedOb
         targetWorkspace,
         includeReferencesDeep
       );
-
-      this.setState({
-        isShowingDuplicateResultFlyout: true,
-        failedCopies: result.success ? [] : result.errors,
-        successfulCopies: result.successCount > 0 ? result.successResults : [],
-        targetWorkspaceName,
-      });
+      if (result?.error) {
+        showErrorNotification(result.error);
+      } else {
+        this.setState({
+          isShowingDuplicateResultFlyout: true,
+          failedCopies: result?.errors || [],
+          successfulCopies: result?.successCount > 0 ? result.successResults : [],
+          targetWorkspaceName,
+        });
+      }
     } catch (e) {
       showErrorNotification();
+    } finally {
+      this.hideDuplicateModal();
+      await this.refreshObjects();
     }
-    this.hideDuplicateModal();
-    await this.refreshObjects();
   };
 
   renderDuplicateModal() {