Update superset-frontend/spec/javascripts/sqllab/ResultSet_spec.jsx

Co-authored-by: Michael S. Molina <70410625+michael-s-molina@users.noreply.github.com>
apache · May 21, 2021 · 1497624 · 1497624
1 parent 5da9ecf
commit 1497624
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/superset-frontend/spec/javascripts/sqllab/ResultSet_spec.jsx b/superset-frontend/spec/javascripts/sqllab/ResultSet_spec.jsx
@@ -186,13 +186,12 @@ describe('ResultSet', () => {
 describe('RTL ResultSet tests', () => {
   it('renders if there is no limit in query.results but has queryLimit', () => {
     render(<ResultSet {...mockedProps} />, { useRedux: true });
-    const grid = screen.getAllByRole('grid');
-    expect(grid.length).toBe(1);
+    expect(screen.getByRole('grid')).toBeInTheDocument();
   });
 
   it('renders if there is a limit in query.results but not queryLimit', () => {
     const props = { ...mockedProps, query: queryWithNoQueryLimit };
     render(<ResultSet {...props} />, { useRedux: true });
-    expect(screen.getByRole('grid')).toBeInTheDocument());
+    expect(screen.getByRole('grid')).toBeInTheDocument();
   });
 });