chore: add test

NG-ZORRO · Oct 13, 2018 · a08e149 · a08e149
1 parent 4f793df
commit a08e149
Show file tree

Hide file tree

Showing 2 changed files with 61 additions and 5 deletions.
diff --git a/components/tree-select/demo/customized-icon.ts b/components/tree-select/demo/customized-icon.ts
@@ -1,5 +1,4 @@
-import { Component, OnInit } from '@angular/core';
-import { NzTreeNode } from 'ng-zorro-antd';
+import { Component } from '@angular/core';
 
 @Component({
   selector: 'nz-demo-tree-select-customized-icon',
@@ -28,7 +27,7 @@ import { NzTreeNode } from 'ng-zorro-antd';
 export class NzDemoTreeSelectCustomizedIconComponent {
   value: string;
   nodes = [
-    new NzTreeNode({
+    {
       title   : 'root1',
       key     : '1001',
       expanded: true,
@@ -64,6 +63,6 @@ export class NzDemoTreeSelectCustomizedIconComponent {
           ]
         }
       ]
-    })
+    }
   ];
 }
diff --git a/components/tree-select/nz-tree-select.spec.ts b/components/tree-select/nz-tree-select.spec.ts
@@ -22,7 +22,7 @@ describe('tree-select component', () => {
   beforeEach(async(() => {
     TestBed.configureTestingModule({
       imports     : [ NzTreeSelectModule, NoopAnimationsModule, FormsModule, ReactiveFormsModule ],
-      declarations: [ NzTestTreeSelectBasicComponent, NzTestTreeSelectCheckableComponent, NzTestTreeSelectFormComponent ]
+      declarations: [ NzTestTreeSelectBasicComponent, NzTestTreeSelectCheckableComponent, NzTestTreeSelectFormComponent, NzTestTreeSelectCustomizedIconComponent ]
     });
     TestBed.compileComponents();
     inject([ OverlayContainer ], (oc: OverlayContainer) => {
@@ -340,6 +340,28 @@ describe('tree-select component', () => {
     }));
   });
 
+  describe('customized icon', () => {
+    let fixture;
+    let testComponent;
+    let treeSelect;
+    let treeSelectComponent;
+    beforeEach(() => {
+      fixture = TestBed.createComponent(NzTestTreeSelectCustomizedIconComponent);
+      testComponent = fixture.debugElement.componentInstance;
+      treeSelect = fixture.debugElement.query(By.directive(NzTreeSelectComponent));
+      treeSelectComponent = treeSelect.componentInstance;
+    });
+    it('should display', fakeAsync(() => {
+      // ExpressionChangedAfterItHasBeenCheckedError
+      treeSelectComponent.ngAfterViewInit();
+      fixture.detectChanges();
+      treeSelect.nativeElement.click();
+      fixture.detectChanges();
+      flush();
+      expect(overlayContainerElement.querySelector('i.anticon.anticon-frown-o')).toBeTruthy();
+    }));
+  });
+
 });
 
 @Component({
@@ -563,3 +585,38 @@ export class NzTestTreeSelectFormComponent {
     this.formGroup.get('select').reset(null);
   }
 }
+
+@Component({
+  selector: 'nz-test-tree-select-customized-icon',
+  template: `
+    <nz-tree-select
+      [nzNodes]="nodes"
+      [(ngModel)]="value"
+    >
+      <ng-template #nzTreeTemplate let-node>
+        <span>
+          <i class="anticon anticon-frown-o"></i> {{node.title}}
+        </span>
+      </ng-template>
+    </nz-tree-select>
+  `
+})
+export class NzTestTreeSelectCustomizedIconComponent {
+  value: string;
+  nodes = [
+    new NzTreeNode({
+      title   : 'root3',
+      key     : '1003',
+      children: [
+        {
+          title: 'child3.1',
+          key  : '10031'
+        },
+        {
+          title: 'child3.2',
+          key  : '10032'
+        }
+      ]
+    })
+  ];
+}