From d5e48539246c41481a617c59473010f9cb5ddecc Mon Sep 17 00:00:00 2001
From: tygao <tygao@amazon.com>
Date: Thu, 29 Aug 2024 17:24:55 +0800
Subject: [PATCH] test: add mock for test cases

Signed-off-by: tygao <tygao@amazon.com>
---
 .../__snapshots__/header.test.tsx.snap        | 255 ++++++++++++++++++
 .../object_view/components/header.test.tsx    |  30 ++-
 2 files changed, 284 insertions(+), 1 deletion(-)

diff --git a/src/plugins/saved_objects_management/public/management_section/object_view/components/__snapshots__/header.test.tsx.snap b/src/plugins/saved_objects_management/public/management_section/object_view/components/__snapshots__/header.test.tsx.snap
index 323031ad6547..84d7c8a6c078 100644
--- a/src/plugins/saved_objects_management/public/management_section/object_view/components/__snapshots__/header.test.tsx.snap
+++ b/src/plugins/saved_objects_management/public/management_section/object_view/components/__snapshots__/header.test.tsx.snap
@@ -273,3 +273,258 @@ exports[`Intro component renders correctly 1`] = `
   </EuiPageContentHeader>
 </Header>
 `;
+
+exports[`Intro component renders correctly when use new UX 1`] = `
+<Header
+  application={
+    Object {
+      "applications$": BehaviorSubject {
+        "_isScalar": false,
+        "_value": Map {},
+        "closed": false,
+        "hasError": false,
+        "isStopped": false,
+        "observers": Array [],
+        "thrownError": null,
+      },
+      "capabilities": Object {
+        "catalogue": Object {},
+        "management": Object {},
+        "navLinks": Object {},
+        "workspaces": Object {},
+      },
+      "currentAppId$": Observable {
+        "_isScalar": false,
+        "source": Subject {
+          "_isScalar": false,
+          "closed": false,
+          "hasError": false,
+          "isStopped": false,
+          "observers": Array [],
+          "thrownError": null,
+        },
+      },
+      "getUrlForApp": [MockFunction],
+      "navigateToApp": [MockFunction],
+      "navigateToUrl": [MockFunction],
+      "registerMountContext": [MockFunction],
+      "setAppBadgeControls": [MockFunction],
+      "setAppBottomControls": [MockFunction],
+      "setAppCenterControls": [MockFunction],
+      "setAppDescriptionControls": [MockFunction],
+      "setAppLeftControls": [MockFunction],
+      "setAppRightControls": [MockFunction],
+    }
+  }
+  canDelete={true}
+  canEdit={true}
+  canViewInApp={true}
+  navigationUI={
+    Object {
+      "HeaderControl": [Function],
+    }
+  }
+  onDeleteClick={[Function]}
+  type="search"
+  useUpdatedUX={true}
+  viewUrl="/some-url"
+>
+  <HeaderControl
+    controls={
+      Array [
+        Object {
+          "ariaLabel": "Delete search",
+          "color": "danger",
+          "controlType": "icon",
+          "display": "base",
+          "iconType": "trash",
+          "run": [Function],
+          "testId": "savedObjectEditDelete",
+        },
+        Object {
+          "controlType": "button",
+          "fill": true,
+          "href": "/some-url",
+          "iconSide": "right",
+          "iconType": "popout",
+          "label": "View search",
+          "testId": "savedObjectEditViewInApp",
+        },
+      ]
+    }
+    setMountPoint={[MockFunction]}
+  >
+    Delete search
+  </HeaderControl>
+</Header>
+`;
+
+exports[`Intro component renders correctly when use new UX and canDelete is true 1`] = `
+<Header
+  application={
+    Object {
+      "applications$": BehaviorSubject {
+        "_isScalar": false,
+        "_value": Map {},
+        "closed": false,
+        "hasError": false,
+        "isStopped": false,
+        "observers": Array [],
+        "thrownError": null,
+      },
+      "capabilities": Object {
+        "catalogue": Object {},
+        "management": Object {},
+        "navLinks": Object {},
+        "workspaces": Object {},
+      },
+      "currentAppId$": Observable {
+        "_isScalar": false,
+        "source": Subject {
+          "_isScalar": false,
+          "closed": false,
+          "hasError": false,
+          "isStopped": false,
+          "observers": Array [],
+          "thrownError": null,
+        },
+      },
+      "getUrlForApp": [MockFunction],
+      "navigateToApp": [MockFunction],
+      "navigateToUrl": [MockFunction],
+      "registerMountContext": [MockFunction],
+      "setAppBadgeControls": [MockFunction],
+      "setAppBottomControls": [MockFunction],
+      "setAppCenterControls": [MockFunction],
+      "setAppDescriptionControls": [MockFunction],
+      "setAppLeftControls": [MockFunction],
+      "setAppRightControls": [MockFunction],
+    }
+  }
+  canDelete={true}
+  canEdit={true}
+  canViewInApp={true}
+  navigationUI={
+    Object {
+      "HeaderControl": [Function],
+    }
+  }
+  onDeleteClick={[Function]}
+  type="search"
+  useUpdatedUX={true}
+  viewUrl="/some-url"
+>
+  <HeaderControl
+    controls={
+      Array [
+        Object {
+          "ariaLabel": "Delete search",
+          "color": "danger",
+          "controlType": "icon",
+          "display": "base",
+          "iconType": "trash",
+          "run": [Function],
+          "testId": "savedObjectEditDelete",
+        },
+        Object {
+          "controlType": "button",
+          "fill": true,
+          "href": "/some-url",
+          "iconSide": "right",
+          "iconType": "popout",
+          "label": "View search",
+          "testId": "savedObjectEditViewInApp",
+        },
+      ]
+    }
+    setMountPoint={[MockFunction]}
+  >
+    Delete search
+  </HeaderControl>
+</Header>
+`;
+
+exports[`Intro component renders correctly when use new UX and canViewInApp is true 1`] = `
+<Header
+  application={
+    Object {
+      "applications$": BehaviorSubject {
+        "_isScalar": false,
+        "_value": Map {},
+        "closed": false,
+        "hasError": false,
+        "isStopped": false,
+        "observers": Array [],
+        "thrownError": null,
+      },
+      "capabilities": Object {
+        "catalogue": Object {},
+        "management": Object {},
+        "navLinks": Object {},
+        "workspaces": Object {},
+      },
+      "currentAppId$": Observable {
+        "_isScalar": false,
+        "source": Subject {
+          "_isScalar": false,
+          "closed": false,
+          "hasError": false,
+          "isStopped": false,
+          "observers": Array [],
+          "thrownError": null,
+        },
+      },
+      "getUrlForApp": [MockFunction],
+      "navigateToApp": [MockFunction],
+      "navigateToUrl": [MockFunction],
+      "registerMountContext": [MockFunction],
+      "setAppBadgeControls": [MockFunction],
+      "setAppBottomControls": [MockFunction],
+      "setAppCenterControls": [MockFunction],
+      "setAppDescriptionControls": [MockFunction],
+      "setAppLeftControls": [MockFunction],
+      "setAppRightControls": [MockFunction],
+    }
+  }
+  canDelete={true}
+  canEdit={true}
+  canViewInApp={true}
+  navigationUI={
+    Object {
+      "HeaderControl": [Function],
+    }
+  }
+  onDeleteClick={[Function]}
+  type="search"
+  useUpdatedUX={true}
+  viewUrl="/some-url"
+>
+  <HeaderControl
+    controls={
+      Array [
+        Object {
+          "ariaLabel": "Delete search",
+          "color": "danger",
+          "controlType": "icon",
+          "display": "base",
+          "iconType": "trash",
+          "run": [Function],
+          "testId": "savedObjectEditDelete",
+        },
+        Object {
+          "controlType": "button",
+          "fill": true,
+          "href": "/some-url",
+          "iconSide": "right",
+          "iconType": "popout",
+          "label": "View search",
+          "testId": "savedObjectEditViewInApp",
+        },
+      ]
+    }
+    setMountPoint={[MockFunction]}
+  >
+    Delete search
+  </HeaderControl>
+</Header>
+`;
diff --git a/src/plugins/saved_objects_management/public/management_section/object_view/components/header.test.tsx b/src/plugins/saved_objects_management/public/management_section/object_view/components/header.test.tsx
index a69d68637956..eb38642b631a 100644
--- a/src/plugins/saved_objects_management/public/management_section/object_view/components/header.test.tsx
+++ b/src/plugins/saved_objects_management/public/management_section/object_view/components/header.test.tsx
@@ -63,7 +63,9 @@ describe('Intro component', () => {
     onDeleteClick: () => undefined,
     useUpdatedUX: false,
     navigationUI: ({
-      HeaderControl: () => null,
+      HeaderControl: ({ controls }) => {
+        return controls?.[0].ariaLabel ?? controls?.[0].label ?? null;
+      },
     } as unknown) as NavigationPublicPluginStart['ui'],
     application: coreMock.createStart().application,
   };
@@ -144,4 +146,30 @@ describe('Intro component', () => {
     });
     expect(mounted.exists(`a[data-test-subj='savedObjectEditViewInApp']`)).toBe(false);
   });
+
+  it('renders correctly when use new UX', () => {
+    const mounted = mountHeader({
+      ...defaultProps,
+      useUpdatedUX: true,
+    });
+    expect(mounted).toMatchSnapshot();
+  });
+
+  it('renders correctly when use new UX and canViewInApp is true', () => {
+    const mounted = mountHeader({
+      ...defaultProps,
+      useUpdatedUX: true,
+      canViewInApp: true,
+    });
+    expect(mounted).toMatchSnapshot();
+  });
+
+  it('renders correctly when use new UX and canDelete is true', () => {
+    const mounted = mountHeader({
+      ...defaultProps,
+      useUpdatedUX: true,
+      canDelete: true,
+    });
+    expect(mounted).toMatchSnapshot();
+  });
 });