@@ -9,7 +9,7 @@ import type { Container } from '@deephaven/golden-layout';
99import { Workspace } from '@deephaven/redux' ;
1010import { IrisGridPanel } from './IrisGridPanel' ;
1111
12- const MockIrisGrid = jest . fn ( ( ) => null ) ;
12+ const MockIrisGrid : React . FC & jest . Mock = jest . fn ( ( ) => null ) ;
1313
1414jest . mock ( '@deephaven/iris-grid' , ( ) => {
1515 const { forwardRef } = jest . requireActual ( 'react' ) ;
@@ -84,10 +84,9 @@ function makeIrisGridPanelWrapper(
8484async function expectLoading ( container ) {
8585 await waitFor ( ( ) =>
8686 expect (
87- container . querySelector ( "[data-icon='circle- large']" )
87+ container . querySelector ( '[role=progressbar].loading-spinner- large')
8888 ) . toBeInTheDocument ( )
8989 ) ;
90- expect ( container . querySelector ( "[data-icon='loading']" ) ) . toBeInTheDocument ( ) ;
9190}
9291
9392async function expectNotLoading ( container ) {
@@ -97,7 +96,7 @@ async function expectNotLoading(container) {
9796 ) . not . toBeInTheDocument ( )
9897 ) ;
9998 expect (
100- container . querySelector ( "[data-icon=' loading']" )
99+ container . querySelector ( '[role=progressbar]. loading-spinner-large' )
101100 ) . not . toBeInTheDocument ( ) ;
102101}
103102
@@ -127,7 +126,7 @@ it('shows the loading spinner until grid is ready', async () => {
127126 const tablePromise = Promise . resolve ( table ) ;
128127 const makeModel = makeMakeModel ( tablePromise ) ;
129128
130- expect . assertions ( 6 ) ;
129+ expect . assertions ( 4 ) ;
131130 const { container } = makeIrisGridPanelWrapper ( makeModel ) ;
132131
133132 await expectLoading ( container ) ;
0 commit comments