chore: Copy results to the clipboard

Author: astandrik

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

@@ -262,6 +262,12 @@ export class QueryEditor {
         }
     }
 
+    async clickCopyResultButton() {
+        const copyButton = this.resultsControls.locator('button[title="Copy result"]');
+        await copyButton.waitFor({state: 'visible', timeout: VISIBILITY_TIMEOUT});
+        await copyButton.click();
+    }
+
     async isRunButtonEnabled() {
         return this.runButton.isEnabled({timeout: VISIBILITY_TIMEOUT});
     }

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

@@ -1,6 +1,7 @@
 import {expect, test} from '@playwright/test';
 
 import {QUERY_MODES, STATISTICS_MODES} from '../../../../src/utils/query';
+import {getClipboardContent} from '../../../utils/clipboard';
 import {tenantName} from '../../../utils/constants';
 import {NavigationTabs, TenantPage, VISIBILITY_TIMEOUT} from '../TenantPage';
 import {createTableQuery, longRunningQuery, longTableSelect} from '../constants';
@@ -317,4 +318,33 @@ test.describe('Test Query Editor', async () => {
         await queryEditor.expandResultsControls();
         await expect(queryEditor.isResultsControlsCollapsed()).resolves.toBe(false);
     });
+
+    test('Copy result button copies to clipboard', async ({page}) => {
+        const queryEditor = new QueryEditor(page);
+        const query = 'SELECT 42 as answer;';
+
+        // Run query to get results
+        await queryEditor.setQuery(query);
+        await queryEditor.clickRunButton();
+        await queryEditor.waitForStatus('Completed');
+
+        // Click copy button
+        await queryEditor.clickCopyResultButton();
+
+        // Wait for clipboard operation to complete
+        await page.waitForTimeout(2000);
+
+        // Retry clipboard read a few times if needed
+        let clipboardContent = '';
+        for (let i = 0; i < 3; i++) {
+            clipboardContent = await getClipboardContent(page);
+            if (clipboardContent) {
+                break;
+            }
+            await page.waitForTimeout(500);
+        }
+
+        // Verify clipboard contains the query result
+        expect(clipboardContent).toContain('42');
+    });
 });