fix(autocomplete): restore focus to input when click on option (#4702)

Restores focus to the autocomplete input when an option is clicked, instead of dumping it into the body.

Fixes #4645.

Author: crisbeto

src/lib/autocomplete/autocomplete-trigger.ts

@@ -339,6 +339,7 @@ export class MdAutocompleteTrigger implements ControlValueAccessor, OnDestroy {
       this._clearPreviousSelectedOption(event.source);
       this._setTriggerValue(event.source.value);
       this._onChange(event.source.value);
+      this._element.nativeElement.focus();
     }
 
     this.closePanel();

src/lib/autocomplete/autocomplete.spec.ts

@@ -878,6 +878,22 @@ describe('MdAutocomplete', () => {
 
     });
 
+    it('should restore focus to the input when clicking to select a value', async(() => {
+      fixture.componentInstance.trigger.openPanel();
+      fixture.detectChanges();
+
+      fixture.whenStable().then(() => {
+        const option = overlayContainerElement.querySelector('md-option') as HTMLElement;
+
+        // Focus the option manually since the synthetic click may not do it.
+        option.focus();
+        option.click();
+        fixture.detectChanges();
+
+        expect(document.activeElement).toBe(input, 'Expected focus to be restored to the input.');
+      });
+    }));
+
   });
 
   describe('Fallback positions', () => {