diff --git a/superset-frontend/src/explore/components/ExportToCSVDropdown/ExportToCSVDropdown.test.tsx b/superset-frontend/src/explore/components/ExportToCSVDropdown/ExportToCSVDropdown.test.tsx index 5207b5f5ffbb4..144f03aa6f165 100644 --- a/superset-frontend/src/explore/components/ExportToCSVDropdown/ExportToCSVDropdown.test.tsx +++ b/superset-frontend/src/explore/components/ExportToCSVDropdown/ExportToCSVDropdown.test.tsx @@ -18,21 +18,27 @@ */ import React from 'react'; import userEvent from '@testing-library/user-event'; -import { render, screen } from 'spec/helpers/testing-library'; +import { render, screen, waitFor } from 'spec/helpers/testing-library'; import { ExportToCSVDropdown } from './index'; const exportCSVOriginal = jest.fn(); const exportCSVPivoted = jest.fn(); -test('Dropdown button with menu renders', () => { - render( - -
.CSV
-
, +const waitForRender = () => { + waitFor(() => + render( + +
.CSV
+
, + ), ); +}; + +test('Dropdown button with menu renders', () => { + waitForRender(); expect(screen.getByText('.CSV')).toBeVisible(); @@ -43,14 +49,7 @@ test('Dropdown button with menu renders', () => { }); test('Call export csv original on click', () => { - render( - -
.CSV
-
, - ); + waitForRender(); userEvent.click(screen.getByText('.CSV')); userEvent.click(screen.getByText('Original')); @@ -59,14 +58,7 @@ test('Call export csv original on click', () => { }); test('Call export csv pivoted on click', () => { - render( - -
.CSV
-
, - ); + waitForRender(); userEvent.click(screen.getByText('.CSV')); userEvent.click(screen.getByText('Pivoted'));