fix: tests

Author: astandrik

src/containers/Tenant/Query/QueryResult/components/QueryInfoDropdown/QueryInfoDropdown.tsx

@@ -4,6 +4,7 @@ import {ActionTooltip, Button, DropdownMenu} from '@gravity-ui/uikit';
 
 import i18n from '../../i18n';
 
+import {b} from './shared';
 import type {QueryResultsInfo} from './useQueryInfoMenuItems';
 import {useQueryInfoMenuItems} from './useQueryInfoMenuItems';
 
@@ -50,6 +51,7 @@ export function QueryInfoDropdown({
             popupProps={{
                 placement: ['bottom-end', 'left'],
             }}
+            switcherWrapperClassName={b('query-info-switcher-wrapper')}
             renderSwitcher={renderSwitcher}
             items={items}
             size="xl"

src/containers/Tenant/Query/QueryResult/components/QueryInfoDropdown/shared.ts

@@ -0,0 +1,2 @@
+import {cn} from '../../../../../../utils/cn';
+export const b = cn('query-info-dropdown');

src/containers/Tenant/Query/QueryResult/components/QueryInfoDropdown/useQueryInfoMenuItems.tsx

@@ -6,14 +6,13 @@ import {Text} from '@gravity-ui/uikit';
 
 import {planToSvgApi} from '../../../../../../store/reducers/planToSvg';
 import type {QueryPlan, ScriptPlan, TKqpStatsQuery} from '../../../../../../types/api/query';
-import {cn} from '../../../../../../utils/cn';
 import createToast from '../../../../../../utils/createToast';
 import {prepareCommonErrorMessage} from '../../../../../../utils/errors';
 import {parseQueryError} from '../../../../../../utils/query';
 import i18n from '../../i18n';
 
+import {b} from './shared';
 import {downloadFile} from './utils';
-const b = cn('query-info-dropdown');
 
 export interface MenuItemContentProps {
     title: string;

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

@@ -37,18 +37,20 @@ test.describe('Test Plan to SVG functionality', async () => {
             expect(status).toBe('Completed');
         }).toPass();
 
-        // 4. Check if Execution Plan button appears and click it to open dropdown
-        const executionPlanButton = page.locator('button:has-text("Execution plan")');
-        await expect(executionPlanButton).toBeVisible();
-        await executionPlanButton.click();
+        // 4. Check if dropdown button appears and click it to open menu
+        const dropdownButton = page.locator('.query-info-switcher-wrapper');
+        await expect(dropdownButton).toBeVisible();
+        await dropdownButton.click();
 
         // 5. Verify dropdown menu items are visible
-        const openInNewTabOption = page.locator('text="Open in new tab"');
-        const downloadOption = page.locator('text="Download"');
+        const openInNewTabOption = page.locator('text="Open Execution Plan"');
+        const downloadPlanOption = page.locator('text="Download Execution Plan"');
+        const downloadDiagnosticsOption = page.locator('text="Download Diagnostics"');
         await expect(openInNewTabOption).toBeVisible();
-        await expect(downloadOption).toBeVisible();
+        await expect(downloadPlanOption).toBeVisible();
+        await expect(downloadDiagnosticsOption).toBeVisible();
 
-        // 6. Click "Open in new tab" option
+        // 6. Click "Open Execution Plan" option
         await openInNewTabOption.click();
         await page.waitForTimeout(1000); // Wait for new tab to open
 
@@ -73,16 +75,16 @@ test.describe('Test Plan to SVG functionality', async () => {
             expect(status).toBe('Completed');
         }).toPass();
 
-        // 4. Click execution plan button to open dropdown
-        const executionPlanButton = page.locator('button:has-text("Execution plan")');
-        await executionPlanButton.click();
+        // 4. Click dropdown button to open menu
+        const dropdownButton = page.locator('.query-info-switcher-wrapper');
+        await dropdownButton.click();
 
         // 5. Setup download listener before clicking download
         const downloadPromise = page.waitForEvent('download');
 
-        // 6. Click download option
-        const downloadOption = page.locator('text="Download"');
-        await downloadOption.click();
+        // 6. Click download execution plan option
+        const downloadPlanOption = page.locator('text="Download Execution Plan"');
+        await downloadPlanOption.click();
 
         // 7. Wait for download to start and verify filename
         const download = await downloadPromise;
@@ -114,32 +116,33 @@ test.describe('Test Plan to SVG functionality', async () => {
             });
         });
 
-        // 5. Click execution plan button to open dropdown
-        const executionPlanButton = page.locator('button:has-text("Execution plan")');
-        await executionPlanButton.click();
+        // 5. Click dropdown button to open menu
+        const dropdownButton = page.locator('.query-info-switcher-wrapper');
+        await dropdownButton.click();
 
-        // 6. Click "Open in new tab" option and wait for error state
-        const openInNewTabOption = page.locator('text="Open in new tab"');
-        await openInNewTabOption.click();
+        // 6. Click "Open Execution Plan" option and wait for error state
+        const openExecutionPlanOption = page.locator('text="Open Execution Plan"');
+        await openExecutionPlanOption.click();
         await page.waitForTimeout(1000); // Wait for error to be processed
 
         // 7. Close the dropdown
         await page.keyboard.press('Escape');
 
         // 8. Verify error state
-        await expect(executionPlanButton).toHaveClass(/flat-danger/);
+        await expect(dropdownButton).toHaveClass(/flat-danger/);
 
         // 9. Verify error tooltip
-        await executionPlanButton.hover();
+        await dropdownButton.hover();
         await page.waitForTimeout(500); // Wait for tooltip animation
         const tooltipText = await page.textContent('.g-tooltip');
         expect(tooltipText).toContain('Error');
         expect(tooltipText).toContain('Failed to generate SVG');
 
         // 10. Verify dropdown is disabled after error
-        await executionPlanButton.click();
-        await expect(openInNewTabOption).not.toBeVisible();
-        await expect(page.locator('text="Download"')).not.toBeVisible();
+        await dropdownButton.click();
+        await expect(openExecutionPlanOption).not.toBeVisible();
+        await expect(page.locator('text="Download Execution Plan"')).not.toBeVisible();
+        await expect(page.locator('text="Download Diagnostics"')).not.toBeVisible();
     });
 
     test('Statistics setting becomes disabled when execution plan experiment is enabled', async ({