update tests

WordPress · May 25, 2019 · 71984ee · 71984ee
1 parent 5565d5c
commit 71984ee
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git a/packages/data/src/components/with-dispatch/test/index.js b/packages/data/src/components/with-dispatch/test/index.js
@@ -8,7 +8,7 @@ import TestRenderer from 'react-test-renderer';
  */
 import withDispatch from '../';
 import { createRegistry } from '../../../registry';
-import RegistryProvider from '../../registry-provider';
+import { RegistryProvider } from '../../registry-provider';
 
 describe( 'withDispatch', () => {
 	let registry;

diff --git a/packages/viewport/src/test/if-viewport-matches.js b/packages/viewport/src/test/if-viewport-matches.js
@@ -1,7 +1,7 @@
 /**
  * External dependencies
  */
-import TestRenderer from 'react-test-renderer';
+import TestRenderer, { act } from 'react-test-renderer';
 
 /**
  * WordPress dependencies
@@ -20,16 +20,24 @@ describe( 'ifViewportMatches()', () => {
 	it( 'should not render if query does not match', () => {
 		dispatch( 'core/viewport' ).setIsMatching( { '> wide': false } );
 		const EnhancedComponent = ifViewportMatches( '> wide' )( Component );
-		const testRenderer = TestRenderer.create( <EnhancedComponent /> );
+
+		let testRenderer;
+		act( () => {
+			testRenderer = TestRenderer.create( <EnhancedComponent /> );
+		} );
 
 		expect( testRenderer.root.findAllByType( Component ) ).toHaveLength( 0 );
 	} );
 
 	it( 'should render if query does match', () => {
-		dispatch( 'core/viewport' ).setIsMatching( { '> wide': true } );
+		act( () => {
+			dispatch( 'core/viewport' ).setIsMatching( { '> wide': true } );
+		} );
 		const EnhancedComponent = ifViewportMatches( '> wide' )( Component );
-		const testRenderer = TestRenderer.create( <EnhancedComponent /> );
-
+		let testRenderer;
+		act( () => {
+			testRenderer = TestRenderer.create( <EnhancedComponent /> );
+		} );
 		expect( testRenderer.root.findAllByType( Component ) ).toHaveLength( 1 );
 	} );
 } );