fix all failed ut

Signed-off-by: ananzh <ananzh@amazon.com>
ananzh · Aug 30, 2023 · 26acb3e · 26acb3e
1 parent 1d981e1
commit 26acb3e
Show file tree

Hide file tree

Showing 4 changed files with 69 additions and 49 deletions.
diff --git a/...plugins/discover/public/application/components/data_grid/data_grid_table_columns.test.tsx b/...plugins/discover/public/application/components/data_grid/data_grid_table_columns.test.tsx
@@ -57,9 +57,12 @@ describe('Testing buildDataGridColumns function ', () => {
       Array [
         Object {
           "actions": Object {
-            "showHide": true,
-            "showMoveLeft": false,
-            "showMoveRight": false,
+            "showHide": Object {
+              "iconType": "cross",
+              "label": "Remove column",
+            },
+            "showMoveLeft": true,
+            "showMoveRight": true,
           },
           "cellActions": undefined,
           "display": undefined,
@@ -69,9 +72,12 @@ describe('Testing buildDataGridColumns function ', () => {
         },
         Object {
           "actions": Object {
-            "showHide": true,
-            "showMoveLeft": false,
-            "showMoveRight": false,
+            "showHide": Object {
+              "iconType": "cross",
+              "label": "Remove column",
+            },
+            "showMoveLeft": true,
+            "showMoveRight": true,
           },
           "cellActions": undefined,
           "display": undefined,
@@ -98,9 +104,9 @@ describe('Testing buildDataGridColumns function ', () => {
       Array [
         Object {
           "actions": Object {
-            "showHide": true,
-            "showMoveLeft": false,
-            "showMoveRight": false,
+            "showHide": false,
+            "showMoveLeft": true,
+            "showMoveRight": true,
           },
           "cellActions": undefined,
           "display": "Time (order_date)",
@@ -111,9 +117,12 @@ describe('Testing buildDataGridColumns function ', () => {
         },
         Object {
           "actions": Object {
-            "showHide": true,
-            "showMoveLeft": false,
-            "showMoveRight": false,
+            "showHide": Object {
+              "iconType": "cross",
+              "label": "Remove column",
+            },
+            "showMoveLeft": true,
+            "showMoveRight": true,
           },
           "cellActions": undefined,
           "display": undefined,
@@ -123,9 +132,12 @@ describe('Testing buildDataGridColumns function ', () => {
         },
         Object {
           "actions": Object {
-            "showHide": true,
-            "showMoveLeft": false,
-            "showMoveRight": false,
+            "showHide": Object {
+              "iconType": "cross",
+              "label": "Remove column",
+            },
+            "showMoveLeft": true,
+            "showMoveRight": true,
           },
           "cellActions": undefined,
           "display": undefined,
@@ -135,9 +147,9 @@ describe('Testing buildDataGridColumns function ', () => {
         },
         Object {
           "actions": Object {
-            "showHide": true,
-            "showMoveLeft": false,
-            "showMoveRight": false,
+            "showHide": false,
+            "showMoveLeft": true,
+            "showMoveRight": true,
           },
           "cellActions": undefined,
           "display": "Source",
@@ -162,9 +174,12 @@ describe('Testing buildDataGridColumns function ', () => {
       Array [
         Object {
           "actions": Object {
-            "showHide": true,
-            "showMoveLeft": false,
-            "showMoveRight": false,
+            "showHide": Object {
+              "iconType": "cross",
+              "label": "Remove column",
+            },
+            "showMoveLeft": true,
+            "showMoveRight": true,
           },
           "cellActions": undefined,
           "display": undefined,
@@ -174,9 +189,12 @@ describe('Testing buildDataGridColumns function ', () => {
         },
         Object {
           "actions": Object {
-            "showHide": true,
-            "showMoveLeft": false,
-            "showMoveRight": false,
+            "showHide": Object {
+              "iconType": "cross",
+              "label": "Remove column",
+            },
+            "showMoveLeft": true,
+            "showMoveRight": true,
           },
           "cellActions": undefined,
           "display": undefined,
@@ -186,9 +204,9 @@ describe('Testing buildDataGridColumns function ', () => {
         },
         Object {
           "actions": Object {
-            "showHide": true,
-            "showMoveLeft": false,
-            "showMoveRight": false,
+            "showHide": false,
+            "showMoveLeft": true,
+            "showMoveRight": true,
           },
           "cellActions": undefined,
           "display": "Time (order_date)",

diff --git a/...blic/application/components/doc_viewer_links/__snapshots__/doc_viewer_links.test.tsx.snap b/...blic/application/components/doc_viewer_links/__snapshots__/doc_viewer_links.test.tsx.snap
diff --git a/src/plugins/discover/public/application/utils/state_management/common.test.ts b/src/plugins/discover/public/application/utils/state_management/common.test.ts
@@ -3,7 +3,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import { addColumn, removeColumn, reorderColumn, setColumns } from './common';
+import { addColumn, removeColumn, reorderColumn } from './common';
 
 describe('commonUtils', () => {
   it('should handle addColumn', () => {
@@ -22,12 +22,4 @@ describe('commonUtils', () => {
       'column1',
     ]);
   });
-
-  it('should handle setColumns', () => {
-    expect(setColumns('timeField', ['timeField', 'column1', 'column2'])).toEqual([
-      'column1',
-      'column2',
-    ]);
-    expect(setColumns(undefined, ['column1', 'column2'])).toEqual(['column1', 'column2']);
-  });
 });
diff --git a/src/plugins/discover/public/application/utils/state_management/discover_slice.test.tsx b/src/plugins/discover/public/application/utils/state_management/discover_slice.test.tsx
@@ -58,7 +58,7 @@ describe('discoverSlice', () => {
   it('should handle setColumns', () => {
     const action = {
       type: 'discover/setColumns',
-      payload: { timeField: 'timeField', columns: ['timeField', 'column1', 'column2'] },
+      payload: { columns: ['column1', 'column2'] },
     };
     const result = discoverSlice.reducer(initialState, action);
     expect(result.columns).toEqual(['column1', 'column2']);