improve state persistence and validation handling

Refactors state persistence logic to ensure invalid state resets
are persisted. Updates tests to reflect changes in state retrieval
behavior. Enhances reliability and consistency of state management.

Author: guidomodarelli

plugins/wazuh-fleet/public/components/common/data-grid/data-grid-state-persistence-manager/use-data-grid-state-persistence-manager.test.ts

@@ -131,7 +131,7 @@ describe('useDataGridStatePersistenceManager', () => {
     );
     const retrievedState = result.current.retrieveState();
 
-    expect(mockStateManagement.retrieveState).toHaveBeenCalledTimes(1);
+    expect(mockStateManagement.retrieveState).toHaveBeenCalledTimes(2);
     expect(retrievedState.pageSize).toEqual(DEFAULT_PAGE_SIZE);
   });
 

plugins/wazuh-fleet/public/components/common/data-grid/data-grid-state-persistence-manager/use-data-grid-state-persistence-manager.ts

@@ -146,6 +146,17 @@ const useDataGridStatePersistenceManager = ({
     stateManagement.clearState();
   };
 
+  const persistState = (payload: Partial<DataGridState>): void => {
+    stateManagement.persistState(payload);
+  };
+
+  const updateState = (payload: Partial<DataGridState>): void => {
+    const currentState = stateManagement.retrieveState();
+    const newState = { ...currentState, ...payload };
+
+    persistState(newState);
+  };
+
   const retrieveState = (): DataGridState => {
     const state: DataGridState = Object.assign(
       {},
@@ -170,20 +181,23 @@ const useDataGridStatePersistenceManager = ({
       validateColumnsState(state.columns ?? []);
     } catch {
       state.columns = [];
+      updateState({ columns: [] });
       console.warn('Columns state was invalid, resetting to default.');
     }
 
     try {
       validateColumnWidthsState(state.columnsWidth ?? {});
     } catch {
       state.columnsWidth = {};
+      updateState({ columnsWidth: {} });
       console.warn('Columns width state was invalid, resetting to default.');
     }
 
     try {
       validatePageSizeState(state.pageSize ?? 0);
     } catch {
       state.pageSize = DEFAULT_PAGE_SIZE;
+      updateState({ pageSize: DEFAULT_PAGE_SIZE });
       console.warn('Page size state was invalid, resetting to default.');
     }
 
@@ -194,17 +208,6 @@ const useDataGridStatePersistenceManager = ({
     return defaultState;
   };
 
-  const persistState = (payload: Partial<DataGridState>): void => {
-    stateManagement.persistState(payload);
-  };
-
-  const updateState = (payload: Partial<DataGridState>): void => {
-    const currentState = retrieveState();
-    const newState = { ...currentState, ...payload };
-
-    persistState(newState);
-  };
-
   return {
     retrieveState,
     persistState,