Fix: UnlinkCommand should update its state upon editor load. Closes #93.

ckeditor · May 15, 2017 · 80aad4e · 80aad4e
1 parent 6d18c55
commit 80aad4e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/unlinkcommand.js b/src/unlinkcommand.js
@@ -24,7 +24,9 @@ export default class UnlinkCommand extends Command {
 		super( editor );
 
 		// Checks when command should be enabled or disabled.
-		this.listenTo( editor.document.selection, 'change:attribute', () => this.refreshState() );
+		this.listenTo( editor.document, 'changesDone', () => {
+			this.refreshState();
+		} );
 	}
 
 	/**

diff --git a/tests/unlinkcommand.js b/tests/unlinkcommand.js
@@ -34,10 +34,10 @@ describe( 'UnlinkCommand', () => {
 	} );
 
 	describe( 'constructor()', () => {
-		it( 'should listen on selection attribute change and refresh state', () => {
+		it( 'should listen on document#changesDone and refresh the state', () => {
 			const refreshStateSpy = testUtils.sinon.spy( command, 'refreshState' );
 
-			document.selection.fire( 'change:attribute' );
+			document.fire( 'changesDone' );
 
 			expect( refreshStateSpy.calledOnce ).to.true;
 		} );