@wordpress/core-data 4.0.1 → 4.0.5

package/src/test/entities.js

@@ -1,3 +1,9 @@
+/**
+ * WordPress dependencies
+ */
+import triggerFetch from '@wordpress/api-fetch';
+jest.mock( '@wordpress/api-fetch' );
+
 /**
  * Internal dependencies
  */
@@ -7,7 +13,6 @@ import {
 	getKindEntities,
 	prePersistPostType,
 } from '../entities';
-import { addEntities } from '../actions';
 
 describe( 'getMethodName', () => {
 	it( 'should return the right method name for an entity with the root kind', () => {
@@ -45,43 +50,52 @@ describe( 'getMethodName', () => {
 } );
 
 describe( 'getKindEntities', () => {
+	beforeEach( async () => {
+		triggerFetch.mockReset();
+		jest.useFakeTimers();
+	} );
+
 	it( 'shouldn’t do anything if the entities have already been resolved', async () => {
+		const dispatch = jest.fn();
+		const select = {
+			getEntitiesByKind: jest.fn( () => entities ),
+		};
 		const entities = [ { kind: 'postType' } ];
-		const fulfillment = getKindEntities( 'postType' );
-		// Start the generator
-		fulfillment.next();
-		// Provide the entities
-		const end = fulfillment.next( entities );
-		expect( end.done ).toBe( true );
+		await getKindEntities( 'postType' )( { dispatch, select } );
+		expect( dispatch ).not.toHaveBeenCalled();
 	} );
 
 	it( 'shouldn’t do anything if there no defined kind config', async () => {
-		const fulfillment = getKindEntities( 'unknownKind' );
-		// Start the generator
-		fulfillment.next();
-		// Provide no entities to continue
-		const end = fulfillment.next( [] );
-		expect( end.done ).toBe( true );
+		const dispatch = jest.fn();
+		const select = {
+			getEntitiesByKind: jest.fn( () => [] ),
+		};
+		await getKindEntities( 'unknownKind' )( { dispatch, select } );
+		expect( dispatch ).not.toHaveBeenCalled();
 	} );
 
 	it( 'should fetch and add the entities', async () => {
 		const fetchedEntities = [
 			{
-				baseURL: '/wp/v2/posts',
-				kind: 'postType',
-				name: 'post',
+				rest_base: 'posts',
+				labels: {
+					singular_name: 'post',
+				},
 			},
 		];
-		const fulfillment = getKindEntities( 'postType' );
-		// Start the generator
-		fulfillment.next();
-		// Provide no entities to continue
-		fulfillment.next( [] );
-		// Fetch entities and trigger action
-		const { value: action } = fulfillment.next( fetchedEntities );
-		expect( action ).toEqual( addEntities( fetchedEntities ) );
-		const end = fulfillment.next();
-		expect( end ).toEqual( { done: true, value: fetchedEntities } );
+		const dispatch = jest.fn();
+		const select = {
+			getEntitiesByKind: jest.fn( () => [] ),
+		};
+		triggerFetch.mockImplementation( () => fetchedEntities );
+
+		await getKindEntities( 'postType' )( { dispatch, select } );
+		expect( dispatch ).toHaveBeenCalledTimes( 1 );
+		expect( dispatch.mock.calls[ 0 ][ 0 ].type ).toBe( 'ADD_ENTITIES' );
+		expect( dispatch.mock.calls[ 0 ][ 0 ].entities.length ).toBe( 1 );
+		expect( dispatch.mock.calls[ 0 ][ 0 ].entities[ 0 ].baseURL ).toBe(
+			'/wp/v2/posts'
+		);
 	} );
 } );