From bd1abd3b6e3d3a6a8892905f7faab8d79fd1bdb0 Mon Sep 17 00:00:00 2001
From: Lyndsi Kay Williams <55605634+lyndsiWilliams@users.noreply.github.com>
Date: Fri, 9 Sep 2022 06:33:14 -0500
Subject: [PATCH] test: Fix act errors in SamplesPane test (#21406)

---
 .../DataTablesPane/test/SamplesPane.test.tsx           | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/superset-frontend/src/explore/components/DataTablesPane/test/SamplesPane.test.tsx b/superset-frontend/src/explore/components/DataTablesPane/test/SamplesPane.test.tsx
index 391540f4d8..eed6e84808 100644
--- a/superset-frontend/src/explore/components/DataTablesPane/test/SamplesPane.test.tsx
+++ b/superset-frontend/src/explore/components/DataTablesPane/test/SamplesPane.test.tsx
@@ -22,9 +22,9 @@ import userEvent from '@testing-library/user-event';
 import {
   render,
   waitForElementToBeRemoved,
+  waitFor,
 } from 'spec/helpers/testing-library';
 import { exploreActions } from 'src/explore/actions/exploreActions';
-import { promiseTimeout } from '@superset-ui/core';
 import { SamplesPane } from '../components';
 import { createSamplesPaneProps } from './fixture';
 
@@ -72,9 +72,9 @@ describe('SamplesPane', () => {
     expect(
       await findByText('No samples were returned for this dataset'),
     ).toBeVisible();
-    await promiseTimeout(() => {
+    await waitFor(() => {
       expect(setForceQuery).toHaveBeenCalledTimes(0);
-    }, 10);
+    });
   });
 
   test('error response', async () => {
@@ -100,9 +100,9 @@ describe('SamplesPane', () => {
       },
     );
 
-    await promiseTimeout(() => {
+    await waitFor(() => {
       expect(setForceQuery).toHaveBeenCalledTimes(1);
-    }, 10);
+    });
     expect(queryByText('2 rows')).toBeVisible();
     expect(queryByText('Action')).toBeVisible();
     expect(queryByText('Horror')).toBeVisible();