fix: add truncated test

Author: astandrik

tests/suites/tenant/constants.ts

@@ -1,7 +1,8 @@
 // Long running query for tests
 // May cause Memory exceed on real database
 
-const simpleQuery = 'SELECT 1;';
+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('');

tests/suites/tenant/queryEditor/QueryEditor.ts

@@ -104,6 +104,12 @@ export class SettingsDialog {
         await this.page.waitForTimeout(1000);
     }
 
+    async changeLimitRows(limitRows: number) {
+        const limitRowsInput = this.dialog.locator('.ydb-query-settings-dialog__limit-rows input');
+        await limitRowsInput.fill(limitRows.toString());
+        await this.page.waitForTimeout(1000);
+    }
+
     async clickButton(buttonName: ButtonNames) {
         const button = this.dialog.getByRole('button', {name: buttonName});
         await button.waitFor({state: 'visible', timeout: VISIBILITY_TIMEOUT});

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

@@ -2,7 +2,7 @@ import {expect, test} from '@playwright/test';
 
 import {tenantName} from '../../../utils/constants';
 import {NavigationTabs, TenantPage, VISIBILITY_TIMEOUT} from '../TenantPage';
-import {createTableQuery, longRunningQuery} from '../constants';
+import {createTableQuery, longRunningQuery, longTableSelect} from '../constants';
 
 import {
     ButtonNames,
@@ -392,4 +392,14 @@ test.describe('Test Query Editor', async () => {
         await page.waitForTimeout(1000);
         await expect(queryEditor.resultTable.isResultHeaderHidden()).resolves.toBe(true);
     });
+
+    test('Truncated head value is 1 for 1 row truncated result', async ({page}) => {
+        const queryEditor = new QueryEditor(page);
+        await queryEditor.setQuery(longTableSelect);
+        await queryEditor.clickGearButton();
+        await queryEditor.settingsDialog.changeLimitRows(1);
+        await queryEditor.settingsDialog.clickButton(ButtonNames.Save);
+        await queryEditor.clickRunButton();
+        await expect(queryEditor.resultTable.getResultHeadText()).resolves.toBe('Truncated(1)');
+    });
 });