Tests: Mock deprecated when testing if it was called

Author: gziolo

packages/components/src/higher-order/with-context/test/index.js

@@ -5,14 +5,17 @@ import renderer from 'react-test-renderer';
 import PropTypes from 'prop-types';
 
 /**
- * Internal dependencies
+ * WordPress dependencies
  */
-import withContext from '../';
+import { Component } from '@wordpress/element';
+import deprecated from '@wordpress/deprecated';
 
 /**
- * WordPress dependencies
+ * Internal dependencies
  */
-import { Component } from '@wordpress/element';
+import withContext from '../';
+
+jest.mock( '@wordpress/deprecated', () => jest.fn() );
 
 class PassContext extends Component {
 	getChildContext() {
@@ -39,7 +42,7 @@ describe( 'withContext', () => {
 		);
 
 		expect( wrapper.root.findByType( 'div' ).children[ 0 ] ).toBe( 'ok' );
-		expect( console ).toHaveWarned();
+		expect( deprecated ).toHaveBeenCalled();
 	} );
 
 	it( 'should allow specifying a context getter mapping', () => {