Code refactoring.

src/linkediting.js

@@ -81,15 +81,12 @@ export default class LinkEditing extends Plugin {
 	 */
 	_setupLinkHighlight() {
 		const editor = this.editor;
-		const model = this.editor.model;
-		const modelDocument = model.document;
 		const view = editor.editing.view;
-		const viewDocument = view.document;
 		const highlightedLinks = new Set();
 
 		// Adding the class.
-		viewDocument.registerPostFixer( writer => {
-			const selection = modelDocument.selection;
+		view.document.registerPostFixer( writer => {
+			const selection = editor.model.document.selection;
 
 			if ( selection.hasAttribute( 'linkHref' ) ) {
 				const modelRange = findLinkRange( selection.getFirstPosition(), selection.getAttribute( 'linkHref' ) );
@@ -114,9 +111,9 @@ export default class LinkEditing extends Plugin {
 		// Removing the class.
 		function removeHighlight() {
 			view.change( writer => {
-				for ( const linkElement of highlightedLinks.values() ) {
-					writer.removeClass( HIGHLIGHT_CLASSES, linkElement );
-					highlightedLinks.delete( linkElement );
+				for ( const item of highlightedLinks.values() ) {
+					writer.removeClass( HIGHLIGHT_CLASSES, item );
+					highlightedLinks.delete( item );
 				}
 			} );
 		}

tests/linkediting.js

@@ -146,8 +146,8 @@ describe( 'LinkEditing', () => {
 		} );
 	} );
 
-	describe( 'highlight link boundaries', () => {
-		it( 'should convert link boundaries marker to proper view', () => {
+	describe( 'link highlighting', () => {
+		it( 'should convert the highlight to a proper view classes', () => {
 			setModelData( model,
 				'<paragraph>foo <$text linkHref="url">b{}ar</$text> baz</paragraph>'
 			);
@@ -201,7 +201,7 @@ describe( 'LinkEditing', () => {
 			expect( model.document.selection.hasAttribute( 'linkHref' ) ).to.be.true;
 		} );
 
-		it( 'should remove marker when selection is moved out from the link', () => {
+		it( 'should remove classes when selection is moved out from the link', () => {
 			setModelData( model,
 				'<paragraph>foo <$text linkHref="url">li{}nk</$text> baz</paragraph>'
 			);