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'));