chore: Expand results tab

Author: astandrik

tests/suites/tenant/TenantPage.ts

@@ -3,7 +3,7 @@ import type {Locator, Page} from '@playwright/test';
 import {PageModel} from '../../models/PageModel';
 import {tenantPage} from '../../utils/constants';
 
-export const VISIBILITY_TIMEOUT = 10000;
+export const VISIBILITY_TIMEOUT = 5000;
 
 export enum NavigationTabs {
     Query = 'Query',

tests/suites/tenant/queryEditor/models/QueryEditor.ts

@@ -240,6 +240,28 @@ export class QueryEditor {
         return true;
     }
 
+    async collapseResultsControls() {
+        const collapseButton = this.resultsControls.locator('button[title="Collapse"]');
+        await collapseButton.waitFor({state: 'visible', timeout: VISIBILITY_TIMEOUT});
+        await collapseButton.click();
+    }
+
+    async expandResultsControls() {
+        const expandButton = this.resultsControls.locator('button[title="Expand"]');
+        await expandButton.waitFor({state: 'visible', timeout: VISIBILITY_TIMEOUT});
+        await expandButton.click();
+    }
+
+    async isResultsControlsCollapsed() {
+        const expandButton = this.resultsControls.locator('button[title="Expand"]');
+        try {
+            await expandButton.waitFor({state: 'visible', timeout: VISIBILITY_TIMEOUT});
+            return true;
+        } catch {
+            return false;
+        }
+    }
+
     async isRunButtonEnabled() {
         return this.runButton.isEnabled({timeout: VISIBILITY_TIMEOUT});
     }

tests/suites/tenant/queryEditor/queryEditor.test.ts

@@ -296,4 +296,25 @@ test.describe('Test Query Editor', async () => {
         await expect(queryEditor.resultTable.hasMultipleResultTabs()).resolves.toBe(false);
         await expect(queryEditor.resultTable.getResultHeadText()).resolves.toBe('Result(1)');
     });
+
+    test('Results controls collapse and expand functionality', async ({page}) => {
+        const queryEditor = new QueryEditor(page);
+
+        // Run a query to show results
+        await queryEditor.setQuery('SELECT 1;');
+        await queryEditor.clickRunButton();
+        await queryEditor.waitForStatus('Completed');
+
+        // Verify controls are initially visible
+        await expect(queryEditor.isResultsControlsVisible()).resolves.toBe(true);
+        await expect(queryEditor.isResultsControlsCollapsed()).resolves.toBe(false);
+
+        // Test collapse
+        await queryEditor.collapseResultsControls();
+        await expect(queryEditor.isResultsControlsCollapsed()).resolves.toBe(true);
+
+        // Test expand
+        await queryEditor.expandResultsControls();
+        await expect(queryEditor.isResultsControlsCollapsed()).resolves.toBe(false);
+    });
 });