From 71ee8fcb130f20f434e53c0a65078e4587bc4677 Mon Sep 17 00:00:00 2001
From: Ashwin P Chandran <ashwinpc@amazon.com>
Date: Mon, 17 Apr 2023 21:46:46 +0000
Subject: [PATCH] Fixes tests

Signed-off-by: Ashwin P Chandran <ashwinpc@amazon.com>
---
 .../public/application/utils/state_management/preload.ts  | 3 +++
 .../utils/state_management/redux_persistence.ts           | 3 +--
 .../application/utils/state_management/ui_state_slice.ts  | 8 ++++++++
 .../public/saved_visualizations/transforms.test.ts        | 4 ++--
 4 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/src/plugins/vis_builder/public/application/utils/state_management/preload.ts b/src/plugins/vis_builder/public/application/utils/state_management/preload.ts
index 43aa2e7b8ede..f7a0f6bd7ad3 100644
--- a/src/plugins/vis_builder/public/application/utils/state_management/preload.ts
+++ b/src/plugins/vis_builder/public/application/utils/state_management/preload.ts
@@ -8,6 +8,7 @@ import { VisBuilderServices } from '../../..';
 import { getPreloadedState as getPreloadedStyleState } from './style_slice';
 import { getPreloadedState as getPreloadedVisualizationState } from './visualization_slice';
 import { getPreloadedState as getPreloadedMetadataState } from './metadata_slice';
+import { getPreloadedState as getPreloadedUIState } from './ui_state_slice';
 import { RootState } from './store';
 
 export const getPreloadedState = async (
@@ -16,10 +17,12 @@ export const getPreloadedState = async (
   const styleState = await getPreloadedStyleState(services);
   const visualizationState = await getPreloadedVisualizationState(services);
   const metadataState = await getPreloadedMetadataState(services);
+  const uiState = await getPreloadedUIState(services);
 
   return {
     style: styleState,
     visualization: visualizationState,
     metadata: metadataState,
+    ui: uiState,
   };
 };
diff --git a/src/plugins/vis_builder/public/application/utils/state_management/redux_persistence.ts b/src/plugins/vis_builder/public/application/utils/state_management/redux_persistence.ts
index b58b14e6d4be..3ebfa47268ec 100644
--- a/src/plugins/vis_builder/public/application/utils/state_management/redux_persistence.ts
+++ b/src/plugins/vis_builder/public/application/utils/state_management/redux_persistence.ts
@@ -12,9 +12,8 @@ export const loadReduxState = async (services: VisBuilderServices) => {
     const serializedState = services.osdUrlStateStorage.get<RootState>('_a');
     if (serializedState !== null) return serializedState;
   } catch (err) {
-    /* eslint-disable no-console */
+    // eslint-disable-next-line no-console
     console.error(err);
-    /* eslint-enable no-console */
   }
 
   return await getPreloadedState(services);
diff --git a/src/plugins/vis_builder/public/application/utils/state_management/ui_state_slice.ts b/src/plugins/vis_builder/public/application/utils/state_management/ui_state_slice.ts
index 26ad24b28a50..826fe9d9873d 100644
--- a/src/plugins/vis_builder/public/application/utils/state_management/ui_state_slice.ts
+++ b/src/plugins/vis_builder/public/application/utils/state_management/ui_state_slice.ts
@@ -4,11 +4,19 @@
  */
 
 import { createSlice, PayloadAction } from '@reduxjs/toolkit';
+import { VisBuilderServices } from '../../../types';
 
 export type UIStateState<T = any> = T;
 
 const initialState = {} as UIStateState;
 
+export const getPreloadedState = async ({
+  types,
+  data,
+}: VisBuilderServices): Promise<UIStateState> => {
+  return initialState;
+};
+
 export const uiStateSlice = createSlice({
   name: 'ui',
   initialState,
diff --git a/src/plugins/vis_builder/public/saved_visualizations/transforms.test.ts b/src/plugins/vis_builder/public/saved_visualizations/transforms.test.ts
index 8f3b59762853..68c24dfe4af1 100644
--- a/src/plugins/vis_builder/public/saved_visualizations/transforms.test.ts
+++ b/src/plugins/vis_builder/public/saved_visualizations/transforms.test.ts
@@ -71,7 +71,7 @@ describe('transforms', () => {
       visualizationState: JSON.stringify({
         searchField: '',
       }),
-      uiState: JSON.stringify('{}'),
+      uiState: '{}',
       searchSourceFields: {
         index: 'test-index',
       },
@@ -83,11 +83,11 @@ describe('transforms', () => {
       expect(state).toMatchInlineSnapshot(`
         Object {
           "style": Object {},
+          "ui": Object {},
           "visualization": Object {
             "indexPattern": "test-index",
             "searchField": "",
           },
-          "ui": Object {},
         }
       `);
     });