UIPQB-164: Columns are reset when user modifies query

CHANGELOG.md

@@ -6,6 +6,7 @@
 * [UIPQB-141](https://folio-org.atlassian.net/browse/UIPQB-141) Modal dialog focus inconsistencies across screenreaders.
 * [UIPQB-162](https://folio-org.atlassian.net/browse/UIPQB-162) Errors when query includes a modified custom field.
 * [UIPQB-175](https://folio-org.atlassian.net/browse/UIPQB-175) Displays the "Smth went wrong" page, when the user clicks on "Select operator" dropdown and selects any of them, if there are deleted custom fields.
+* [UIPQB-164](https://folio-org.atlassian.net/browse/UIPQB-164) Columns are reset when user modifies query.
 
 
 ## [1.2.6](https://github.com/folio-org/ui-plugin-query-builder/tree/v1.2.6) (2024-12-11)

src/QueryBuilder/ResultViewer/ResultViewer.js

@@ -82,6 +82,7 @@
     onPreviewShown,
     defaultLimit,
     forcedVisibleValues,
+    visibleColumns
   });
 
   // refresh functionality

src/QueryBuilder/ResultViewer/ResultViewer.test.js

@@ -76,8 +76,8 @@ describe('ResultViewer', () => {
   });
 
   describe('Initial and visible columns setters', () => {
-    it.each([[], undefined])('should call both when no initial fields are provided (recordColumns=%s)', async (visibleColumns) => {
-      render(renderResultViewer({ visibleColumns }));
+    it.each([[], undefined])('should call both when no initial fields are provided (recordColumns=%s)', async () => {
+      render(renderResultViewer({ visibleColumns: ['user_id'] }));
 
       await waitFor(() => {
         expect(screen.queryByText('ui-plugin-query-builder.viewer.retrieving')).not.toBeInTheDocument();

src/hooks/useViewerCallbacks.js

@@ -8,12 +8,17 @@
   currentRecordsCount,
   onPreviewShown,
   defaultLimit,
+                                     visibleColumns
 }) => {
   useEffect(() => {
     if (defaultColumns.length !== 0) {
       onSetDefaultColumns?.(defaultColumns);
 
-      onSetDefaultVisibleColumns?.(defaultVisibleColumns);
+      const uniqueVisibleColumns = Array.from(
+          new Set([...defaultVisibleColumns, ...visibleColumns])
+      );
+
+      onSetDefaultVisibleColumns?.(uniqueVisibleColumns);
     }
   }, [currentRecordsCount]);