Fix a single entity fetch

WordPress · Aug 3, 2018 · 895463b · 895463b
1 parent 0408943
commit 895463b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/packages/core-data/src/resolvers.js b/packages/core-data/src/resolvers.js
@@ -52,7 +52,9 @@ export async function* getEntityRecord( state, kind, name, key ) {
 		return;
 	}
 	const record = await apiFetch( { path: `${ entity.baseURL }/${ key }?context=edit` } );
-	yield receiveEntityRecords( kind, name, record );
+	yield receiveEntityRecords( kind, name, record, {
+		[ entity.key || 'id' ]: key,
+	} );
 }
 
 /**

diff --git a/packages/core-data/src/test/resolvers.js b/packages/core-data/src/test/resolvers.js
@@ -62,7 +62,7 @@ describe( 'getEntityRecord', () => {
 		};
 		const fulfillment = getEntityRecord( state, 'root', 'postType', 'post' );
 		const received = ( await fulfillment.next() ).value;
-		expect( received ).toEqual( receiveEntityRecords( 'root', 'postType', POST_TYPE ) );
+		expect( received ).toEqual( receiveEntityRecords( 'root', 'postType', POST_TYPE, { id: 'post' } ) );
 	} );
 
 	it( 'loads the kind entities and yields with requested post type', async () => {
@@ -74,7 +74,7 @@ describe( 'getEntityRecord', () => {
 			name: 'post',
 		} ] ) );
 		const received = ( await fulfillment.next() ).value;
-		expect( received ).toEqual( receiveEntityRecords( 'postType', 'post', POST ) );
+		expect( received ).toEqual( receiveEntityRecords( 'postType', 'post', POST, { id: 10 } ) );
 	} );
 } );