fix: add status check test

Author: Anton Standrik

tests/suites/tenant/queryEditor/QueryEditor.ts

@@ -237,4 +237,20 @@ export class QueryEditor {
         await this.indicatorIcon.waitFor({state: 'hidden', timeout: VISIBILITY_TIMEOUT});
         return true;
     }
+
+    async waitForStatus(expectedStatus: string, timeout = VISIBILITY_TIMEOUT) {
+        await this.executionStatus.waitFor({state: 'visible', timeout});
+
+        // Keep checking status until it matches or times out
+        const startTime = Date.now();
+        while (Date.now() - startTime < timeout) {
+            const status = await this.executionStatus.innerText();
+            if (status === expectedStatus) {
+                return true;
+            }
+            await this.page.waitForTimeout(100); // Small delay between checks
+        }
+
+        throw new Error(`Status did not change to ${expectedStatus} within ${timeout}ms`);
+    }
 }

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

@@ -73,9 +73,7 @@ test.describe('Test Query Editor', async () => {
         await queryEditor.setQuery(invalidQuery);
         await queryEditor.clickRunButton();
 
-        const statusElement = await queryEditor.getExecutionStatus();
-        await expect(statusElement).toBe('Failed');
-
+        await expect(queryEditor.waitForStatus('Failed')).resolves.toBe(true);
         const errorMessage = await queryEditor.getErrorMessage();
         await expect(errorMessage).toContain('Column references are not allowed without FROM');
     });
@@ -123,13 +121,9 @@ test.describe('Test Query Editor', async () => {
         await queryEditor.clickRunButton();
 
         await expect(queryEditor.isStopButtonVisible()).resolves.toBe(true);
-
         await queryEditor.clickStopButton();
-        await page.waitForTimeout(1000); // Wait for the editor to initialize
 
-        // Check for a message or indicator that the query was stopped
-        const statusElement = await queryEditor.getExecutionStatus();
-        await expect(statusElement).toBe('Stopped');
+        await expect(queryEditor.waitForStatus('Stopped')).resolves.toBe(true);
     });
 
     test('Stop button is not visible for quick queries', async ({page}) => {
@@ -239,4 +233,12 @@ test.describe('Test Query Editor', async () => {
         await queryEditor.clickRunButton();
         await expect(queryEditor.resultTable.getResultHeadText()).resolves.toBe('Truncated(1)');
     });
+
+    test('Query execution status changes correctly', async ({page}) => {
+        const queryEditor = new QueryEditor(page);
+        await queryEditor.setQuery(testQuery);
+        await queryEditor.clickRunButton();
+
+        await expect(queryEditor.waitForStatus('Completed')).resolves.toBe(true);
+    });
 });