Editor: Use getRegistry control for next change subscription

WordPress · Jul 10, 2019 · 468f908 · 468f908
1 parent 37a7ba9
commit 468f908
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/packages/editor/src/store/actions.js b/packages/editor/src/store/actions.js
@@ -206,7 +206,7 @@ export function __experimentalTearDownEditor() {
  */
 export function* __experimentalSubscribeSources() {
 	while ( true ) {
-		const registry = yield awaitNextStateChange();
+		yield awaitNextStateChange();
 
 		// The bailout case: If the editor becomes unmounted, it will flag
 		// itself as non-ready. Effectively unsubscribes from the registry.
@@ -215,6 +215,8 @@ export function* __experimentalSubscribeSources() {
 			break;
 		}
 
+		const registry = yield getRegistry();
+
 		let reset = false;
 		for ( const source of Object.values( sources ) ) {
 			if ( ! source.getDependencies ) {

diff --git a/packages/editor/src/store/controls.js b/packages/editor/src/store/controls.js
@@ -28,7 +28,7 @@ const controls = {
 		( registry ) => () => new Promise( ( resolve ) => {
 			const unsubscribe = registry.subscribe( () => {
 				unsubscribe();
-				resolve( registry );
+				resolve();
 			} );
 		} )
 	),