fix(input-number): NzAutoFocus doesn't work or work as expected (#1833)

close #1706
NG-ZORRO · Jul 22, 2018 · 739b353 · 739b353
1 parent 5305723
commit 739b353
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/components/input-number/nz-input-number.component.ts b/components/input-number/nz-input-number.component.ts
@@ -104,12 +104,10 @@ export class NzInputNumberComponent implements ControlValueAccessor, AfterViewIn
   }
 
   updateAutoFocus(): void {
-    if (this.isInit) {
-      if (this.nzAutoFocus) {
-        this.renderer.setAttribute(this.inputElement.nativeElement, 'autofocus', 'autofocus');
-      } else {
-        this.renderer.removeAttribute(this.inputElement.nativeElement, 'autofocus');
-      }
+    if (this.nzAutoFocus) {
+      this.renderer.setAttribute(this.inputElement.nativeElement, 'autofocus', 'autofocus');
+    } else {
+      this.renderer.removeAttribute(this.inputElement.nativeElement, 'autofocus');
     }
   }
 
@@ -365,5 +363,8 @@ export class NzInputNumberComponent implements ControlValueAccessor, AfterViewIn
 
   ngAfterViewInit(): void {
     this.isInit = true;
+    if (this._autoFocus) {
+      this.focus();
+    }
   }
 }
diff --git a/components/input-number/nz-input-number.spec.ts b/components/input-number/nz-input-number.spec.ts
@@ -57,7 +57,10 @@ describe('input number', () => {
     it('should autofocus work', () => {
       fixture.detectChanges();
       testComponent.autofocus = true;
+      testComponent.nzInputNumberComponent._autoFocus = true;
+      testComponent.nzInputNumberComponent.ngAfterViewInit();
       fixture.detectChanges();
+      expect(inputElement === document.activeElement).toBe(true);
       expect(inputElement.attributes.getNamedItem('autofocus').name).toBe('autofocus');
       testComponent.autofocus = false;
       fixture.detectChanges();