fix: add couple of template tests

Author: Anton Standrik

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

@@ -16,6 +16,11 @@ export enum AsyncReplicationTemplates {
     Drop = 'Drop async replication',
 }
 
+export enum TablesTemplates {
+    UpdateTable = 'Update table',
+    CreateRowTable = 'Create row table',
+}
+
 export class NewSqlDropdownMenu {
     private dropdownButton: Locator;
     private menu: Locator;
@@ -40,7 +45,7 @@ export class NewSqlDropdownMenu {
         await categoryItem.hover();
     }
 
-    async selectTemplate(template: AsyncReplicationTemplates) {
+    async selectTemplate(template: AsyncReplicationTemplates | TablesTemplates) {
         const templateItem = this.subMenu.getByRole('menuitem').filter({hasText: template});
         await templateItem.waitFor({state: 'visible', timeout: VISIBILITY_TIMEOUT});
         await templateItem.click();

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

@@ -109,11 +109,13 @@ export class QueryEditor {
     async clickRunButton() {
         await this.runButton.waitFor({state: 'visible', timeout: VISIBILITY_TIMEOUT});
         await this.runButton.click();
+        await this.page.waitForTimeout(1000);
     }
 
     async clickExplainButton() {
         await this.explainButton.waitFor({state: 'visible', timeout: VISIBILITY_TIMEOUT});
         await this.explainButton.click();
+        await this.page.waitForTimeout(1000);
     }
 
     async getExplainResult(type: ExplainResultType) {

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

@@ -8,6 +8,7 @@ import {RowTableAction} from '../summary/types';
 import {
     AsyncReplicationTemplates,
     NewSqlDropdownMenu,
+    TablesTemplates,
     TemplateCategory,
 } from './models/NewSqlDropdownMenu';
 import {QueryEditor, QueryTabs} from './models/QueryEditor';
@@ -26,6 +27,60 @@ test.describe('Query Templates', () => {
         await tenantPage.goto(pageQueryParams);
     });
 
+    test('Update table template should not run successfully', async ({page}) => {
+        const newSqlDropdown = new NewSqlDropdownMenu(page);
+        const queryEditor = new QueryEditor(page);
+
+        // Open dropdown and select Update table template
+        await newSqlDropdown.clickNewSqlButton();
+        await newSqlDropdown.hoverCategory(TemplateCategory.Tables);
+        await newSqlDropdown.selectTemplate(TablesTemplates.UpdateTable);
+
+        // Try to run the query
+        await queryEditor.clickRunButton();
+
+        // Verify that execution fails
+        try {
+            await queryEditor.waitForStatus('Failed');
+            // If we reach here, the test passed because execution failed as expected
+        } catch (error) {
+            throw new Error('Update table template should not have executed successfully');
+        }
+    });
+
+    test('Create row table template should handle both success and failure cases', async ({
+        page,
+    }) => {
+        const newSqlDropdown = new NewSqlDropdownMenu(page);
+        const queryEditor = new QueryEditor(page);
+
+        // Open dropdown and select Create row table template
+        await newSqlDropdown.clickNewSqlButton();
+        await newSqlDropdown.hoverCategory(TemplateCategory.Tables);
+        await newSqlDropdown.selectTemplate(TablesTemplates.CreateRowTable);
+
+        // Try to run the query
+        await queryEditor.clickRunButton();
+
+        try {
+            // Wait for either Completed or Failed status
+            const status = await queryEditor.getExecutionStatus();
+
+            if (status === 'Failed') {
+                // If failed, verify it's the expected "path exists" error
+                const errorMessage = await queryEditor.getErrorMessage();
+                expect(errorMessage).toContain('path exist, request accepts it');
+            } else {
+                // If not failed, verify it completed successfully
+                expect(status).toBe('Completed');
+            }
+        } catch (error) {
+            throw new Error(
+                'Query execution neither completed successfully nor failed with expected error',
+            );
+        }
+    });
+
     test('Unsaved changes modal appears when switching between templates', async ({page}) => {
         const objectSummary = new ObjectSummary(page);
         const unsavedChangesModal = new UnsavedChangesModal(page);