Add comprehensive tests for tab persistence functionality

Author: Copilot

src/store/reducers/query/__test__/resultTab.test.ts

@@ -0,0 +1,55 @@
+import queryReducer, {setResultTab} from '../query';
+import type {QueryState} from '../types';
+
+describe('resultTab functionality', () => {
+    const initialState: QueryState = {
+        input: '',
+        history: {
+            queries: [],
+            currentIndex: -1,
+        },
+    };
+
+    test('should set result tab for explain query type', () => {
+        const action = setResultTab({queryType: 'explain', tabId: 'json'});
+        const newState = queryReducer(initialState, action);
+
+        expect(newState.selectedResultTab).toEqual({
+            explain: 'json',
+        });
+    });
+
+    test('should set result tab for execute query type', () => {
+        const action = setResultTab({queryType: 'execute', tabId: 'stats'});
+        const newState = queryReducer(initialState, action);
+
+        expect(newState.selectedResultTab).toEqual({
+            execute: 'stats',
+        });
+    });
+
+    test('should maintain separate tabs for different query types', () => {
+        const state1 = queryReducer(
+            initialState,
+            setResultTab({queryType: 'explain', tabId: 'json'}),
+        );
+        const state2 = queryReducer(state1, setResultTab({queryType: 'execute', tabId: 'stats'}));
+
+        expect(state2.selectedResultTab).toEqual({
+            explain: 'json',
+            execute: 'stats',
+        });
+    });
+
+    test('should update existing tab preference', () => {
+        const state1 = queryReducer(
+            initialState,
+            setResultTab({queryType: 'explain', tabId: 'json'}),
+        );
+        const state2 = queryReducer(state1, setResultTab({queryType: 'explain', tabId: 'ast'}));
+
+        expect(state2.selectedResultTab).toEqual({
+            explain: 'ast',
+        });
+    });
+});

src/store/reducers/query/__test__/tabPersistence.test.tsx

@@ -0,0 +1,78 @@
+import queryReducer, {setResultTab} from '../query';
+
+describe('QueryResultViewer tab persistence integration', () => {
+    const initialState = {
+        input: '',
+        history: {
+            queries: [],
+            currentIndex: -1,
+        },
+    };
+
+    it('should save and retrieve tab selection for explain queries', () => {
+        // Test that we can set and get the tab preference
+        let state = queryReducer(initialState, setResultTab({queryType: 'explain', tabId: 'json'}));
+
+        expect(state.selectedResultTab).toEqual({
+            explain: 'json',
+        });
+
+        // Test updating the same query type
+        state = queryReducer(state, setResultTab({queryType: 'explain', tabId: 'ast'}));
+
+        expect(state.selectedResultTab).toEqual({
+            explain: 'ast',
+        });
+    });
+
+    it('should save and retrieve tab selection for execute queries', () => {
+        const state = queryReducer(
+            initialState,
+            setResultTab({queryType: 'execute', tabId: 'stats'}),
+        );
+
+        expect(state.selectedResultTab).toEqual({
+            execute: 'stats',
+        });
+    });
+
+    it('should maintain separate preferences for different query types', () => {
+        let state = initialState;
+
+        // Set explain tab
+        state = queryReducer(state, setResultTab({queryType: 'explain', tabId: 'json'}));
+        expect(state.selectedResultTab).toEqual({
+            explain: 'json',
+        });
+
+        // Set execute tab - should not override explain
+        state = queryReducer(state, setResultTab({queryType: 'execute', tabId: 'stats'}));
+        expect(state.selectedResultTab).toEqual({
+            explain: 'json',
+            execute: 'stats',
+        });
+
+        // Update explain tab - should not affect execute
+        state = queryReducer(state, setResultTab({queryType: 'explain', tabId: 'ast'}));
+        expect(state.selectedResultTab).toEqual({
+            explain: 'ast',
+            execute: 'stats',
+        });
+    });
+
+    it('should handle multiple updates to the same query type', () => {
+        let state = initialState;
+
+        // Set initial value
+        state = queryReducer(state, setResultTab({queryType: 'explain', tabId: 'schema'}));
+        expect(state.selectedResultTab?.explain).toBe('schema');
+
+        // Update to different tab
+        state = queryReducer(state, setResultTab({queryType: 'explain', tabId: 'json'}));
+        expect(state.selectedResultTab?.explain).toBe('json');
+
+        // Update again
+        state = queryReducer(state, setResultTab({queryType: 'explain', tabId: 'ast'}));
+        expect(state.selectedResultTab?.explain).toBe('ast');
+    });
+});