fix: add long running test

Author: astandrik

src/containers/Tenant/Query/QueryEditorControls/EditorButton.scss

@@ -8,8 +8,8 @@
     }
 
     &__stop-button {
+        color: var(--g-color-text-brand-contrast);
         background-color: var(--g-color-text-danger);
-
         &_error {
             @include mixins.query-buttons-animations();
         }

tests/suites/tenant/constants.ts

@@ -5,7 +5,10 @@ export const simpleQuery = 'SELECT 1;';
 export const longTableSelect = 'SELECT * FROM `.sys/pg_class`';
 
 // 400 is pretty enough
-export const longRunningQuery = new Array(400).fill(simpleQuery).join('');
+export const longRunningQuery = `
+$sample = AsList(AsStruct(ListFromRange(1, 100000) AS value, CAST(1 AS Uint32) AS id));
+
+SELECT value, id FROM as_table($sample) FLATTEN BY (value);`;
 
 export const createTableQuery = `
 CREATE TABLE \`/local/ydb_row_table\` (

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

@@ -3,6 +3,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 {toggleExperiment} from '../../../utils/toggleExperiment';
 import {NavigationTabs, TenantPage, VISIBILITY_TIMEOUT} from '../TenantPage';
 import {createTableQuery, longRunningQuery, longTableSelect} from '../constants';
 
@@ -102,6 +103,19 @@ test.describe('Test Query Editor', async () => {
         await expect(queryEditor.isElapsedTimeVisible()).resolves.toBe(true);
     });
 
+    test('Query streaming finishes in reasonable time', async ({page}) => {
+        const queryEditor = new QueryEditor(page);
+        await toggleExperiment(page, 'on', 'Query Streaming');
+
+        await queryEditor.clickGearButton();
+        await queryEditor.settingsDialog.changeLimitRows(100000);
+        await queryEditor.settingsDialog.clickButton(ButtonNames.Save);
+        await queryEditor.setQuery(longRunningQuery);
+        await queryEditor.clickRunButton();
+
+        await expect(queryEditor.waitForStatus('Completed')).resolves.toBe(true);
+    });
+
     test('Query execution is terminated when stop button is clicked', async ({page}) => {
         const queryEditor = new QueryEditor(page);