@@ -393,7 +393,7 @@ describe('ReactScope', () => {
393
393
} ) ;
394
394
395
395
// @gate www
396
- it ( 'DO_NOT_USE_queryAllNodes() works as intended' , ( ) => {
396
+ it ( 'DO_NOT_USE_queryAllNodes() works as intended' , async ( ) => {
397
397
const testScopeQuery = ( type , props ) => true ;
398
398
const TestScope = React . unstable_Scope ;
399
399
const scopeRef = React . createRef ( ) ;
@@ -417,20 +417,25 @@ describe('ReactScope', () => {
417
417
) ;
418
418
}
419
419
420
- const renderer = ReactTestRenderer . create ( < Test toggle = { true } /> , {
421
- createNodeMock : element => {
422
- return element ;
423
- } ,
424
- } ) ;
420
+ let renderer ;
421
+ await act (
422
+ ( ) =>
423
+ ( renderer = ReactTestRenderer . create ( < Test toggle = { true } /> , {
424
+ createNodeMock : element => {
425
+ return element ;
426
+ } ,
427
+ unstable_isConcurrent : true ,
428
+ } ) ) ,
429
+ ) ;
425
430
let nodes = scopeRef . current . DO_NOT_USE_queryAllNodes ( testScopeQuery ) ;
426
431
expect ( nodes ) . toEqual ( [ divRef . current , spanRef . current , aRef . current ] ) ;
427
- renderer . update ( < Test toggle = { false } /> ) ;
432
+ await act ( ( ) => renderer . update ( < Test toggle = { false } /> ) ) ;
428
433
nodes = scopeRef . current . DO_NOT_USE_queryAllNodes ( testScopeQuery ) ;
429
434
expect ( nodes ) . toEqual ( [ aRef . current , divRef . current , spanRef . current ] ) ;
430
435
} ) ;
431
436
432
437
// @gate www
433
- it ( 'DO_NOT_USE_queryFirstNode() works as intended' , ( ) => {
438
+ it ( 'DO_NOT_USE_queryFirstNode() works as intended' , async ( ) => {
434
439
const testScopeQuery = ( type , props ) => true ;
435
440
const TestScope = React . unstable_Scope ;
436
441
const scopeRef = React . createRef ( ) ;
@@ -454,20 +459,26 @@ describe('ReactScope', () => {
454
459
) ;
455
460
}
456
461
457
- const renderer = ReactTestRenderer . create ( < Test toggle = { true } /> , {
458
- createNodeMock : element => {
459
- return element ;
460
- } ,
461
- } ) ;
462
+ let renderer ;
463
+ await act (
464
+ ( ) =>
465
+ ( renderer = ReactTestRenderer . create ( < Test toggle = { true } /> , {
466
+ createNodeMock : element => {
467
+ return element ;
468
+ } ,
469
+ unstable_isConcurrent : true ,
470
+ } ) ) ,
471
+ ) ;
462
472
let node = scopeRef . current . DO_NOT_USE_queryFirstNode ( testScopeQuery ) ;
463
473
expect ( node ) . toEqual ( divRef . current ) ;
464
- renderer . update ( < Test toggle = { false } /> ) ;
474
+ await act ( ( ) => renderer . update ( < Test toggle = { false } /> ) ) ;
475
+
465
476
node = scopeRef . current . DO_NOT_USE_queryFirstNode ( testScopeQuery ) ;
466
477
expect ( node ) . toEqual ( aRef . current ) ;
467
478
} ) ;
468
479
469
480
// @gate www
470
- it ( 'containsNode() works as intended' , ( ) => {
481
+ it ( 'containsNode() works as intended' , async ( ) => {
471
482
const TestScope = React . unstable_Scope ;
472
483
const scopeRef = React . createRef ( ) ;
473
484
const divRef = React . createRef ( ) ;
@@ -500,23 +511,28 @@ describe('ReactScope', () => {
500
511
) ;
501
512
}
502
513
503
- const renderer = ReactTestRenderer . create ( < Test toggle = { true } /> , {
504
- createNodeMock : element => {
505
- return element ;
506
- } ,
507
- } ) ;
514
+ let renderer ;
515
+ await act (
516
+ ( ) =>
517
+ ( renderer = ReactTestRenderer . create ( < Test toggle = { true } /> , {
518
+ createNodeMock : element => {
519
+ return element ;
520
+ } ,
521
+ unstable_isConcurrent : true ,
522
+ } ) ) ,
523
+ ) ;
508
524
expect ( scopeRef . current . containsNode ( divRef . current ) ) . toBe ( true ) ;
509
525
expect ( scopeRef . current . containsNode ( spanRef . current ) ) . toBe ( true ) ;
510
526
expect ( scopeRef . current . containsNode ( aRef . current ) ) . toBe ( true ) ;
511
527
expect ( scopeRef . current . containsNode ( outerSpan . current ) ) . toBe ( false ) ;
512
528
expect ( scopeRef . current . containsNode ( emRef . current ) ) . toBe ( false ) ;
513
- renderer . update ( < Test toggle = { false } /> ) ;
529
+ await act ( ( ) => renderer . update ( < Test toggle = { false } /> ) ) ;
514
530
expect ( scopeRef . current . containsNode ( divRef . current ) ) . toBe ( true ) ;
515
531
expect ( scopeRef . current . containsNode ( spanRef . current ) ) . toBe ( true ) ;
516
532
expect ( scopeRef . current . containsNode ( aRef . current ) ) . toBe ( true ) ;
517
533
expect ( scopeRef . current . containsNode ( outerSpan . current ) ) . toBe ( false ) ;
518
534
expect ( scopeRef . current . containsNode ( emRef . current ) ) . toBe ( true ) ;
519
- renderer . update ( < Test toggle = { true } /> ) ;
535
+ await act ( ( ) => renderer . update ( < Test toggle = { true } /> ) ) ;
520
536
expect ( scopeRef . current . containsNode ( emRef . current ) ) . toBe ( false ) ;
521
537
} ) ;
522
538
} ) ;
0 commit comments