angular-cn/modules/@angular/common/test/directives/ng_plural_spec.ts

/**
 * @license
 * Copyright Google Inc. All Rights Reserved.
 *
 * Use of this source code is governed by an MIT-style license that can be
 * found in the LICENSE file at https://angular.io/license
 */

import {CommonModule, NgLocalization} from '@angular/common';
import {Component, Injectable} from '@angular/core';
import {ComponentFixture, TestBed, async} from '@angular/core/testing';
import {expect} from '@angular/platform-browser/testing/matchers';

export function main() {
  describe('ngPlural', () => {
    let fixture: ComponentFixture<any>;

    function getComponent(): TestComponent { return fixture.componentInstance; }

    function detectChangesAndExpectText<T>(text: string): void {
      fixture.detectChanges();
      expect(fixture.nativeElement).toHaveText(text);
    }

    afterEach(() => { fixture = null; });

    beforeEach(() => {
      TestBed.configureTestingModule({
        declarations: [TestComponent],
        providers: [{provide: NgLocalization, useClass: TestLocalization}],
        imports: [CommonModule]
      });
    });

    it('should display the template according to the exact value', async(() => {
         const template = '<ul [ngPlural]="switchValue">' +
             '<ng-template ngPluralCase="=0"><li>you have no messages.</li></ng-template>' +
             '<ng-template ngPluralCase="=1"><li>you have one message.</li></ng-template>' +
             '</ul>';

         fixture = createTestComponent(template);

         getComponent().switchValue = 0;
         detectChangesAndExpectText('you have no messages.');

         getComponent().switchValue = 1;
         detectChangesAndExpectText('you have one message.');
       }));

    it('should display the template according to the exact numeric value', async(() => {
         const template = '<div>' +
             '<ul [ngPlural]="switchValue">' +
             '<ng-template ngPluralCase="0"><li>you have no messages.</li></ng-template>' +
             '<ng-template ngPluralCase="1"><li>you have one message.</li></ng-template>' +
             '</ul></div>';

         fixture = createTestComponent(template);

         getComponent().switchValue = 0;
         detectChangesAndExpectText('you have no messages.');

         getComponent().switchValue = 1;
         detectChangesAndExpectText('you have one message.');
       }));

    // https://github.com/angular/angular/issues/9868
    // https://github.com/angular/angular/issues/9882
    it('should not throw when ngPluralCase contains expressions', async(() => {
         const template = '<ul [ngPlural]="switchValue">' +
             '<ng-template ngPluralCase="=0"><li>{{ switchValue }}</li></ng-template>' +
             '</ul>';

         fixture = createTestComponent(template);

         getComponent().switchValue = 0;
         expect(() => fixture.detectChanges()).not.toThrow();
       }));


    it('should be applicable to <ng-container> elements', async(() => {
         const template = '<ng-container [ngPlural]="switchValue">' +
             '<ng-template ngPluralCase="=0">you have no messages.</ng-template>' +
             '<ng-template ngPluralCase="=1">you have one message.</ng-template>' +
             '</ng-container>';

         fixture = createTestComponent(template);

         getComponent().switchValue = 0;
         detectChangesAndExpectText('you have no messages.');

         getComponent().switchValue = 1;
         detectChangesAndExpectText('you have one message.');
       }));

    it('should display the template according to the category', async(() => {
         const template = '<ul [ngPlural]="switchValue">' +
             '<ng-template ngPluralCase="few"><li>you have a few messages.</li></ng-template>' +
             '<ng-template ngPluralCase="many"><li>you have many messages.</li></ng-template>' +
             '</ul>';

         fixture = createTestComponent(template);

         getComponent().switchValue = 2;
         detectChangesAndExpectText('you have a few messages.');

         getComponent().switchValue = 8;
         detectChangesAndExpectText('you have many messages.');
       }));

    it('should default to other when no matches are found', async(() => {
         const template = '<ul [ngPlural]="switchValue">' +
             '<ng-template ngPluralCase="few"><li>you have a few messages.</li></ng-template>' +
             '<ng-template ngPluralCase="other"><li>default message.</li></ng-template>' +
             '</ul>';

         fixture = createTestComponent(template);

         getComponent().switchValue = 100;
         detectChangesAndExpectText('default message.');
       }));

    it('should prioritize value matches over category matches', async(() => {
         const template = '<ul [ngPlural]="switchValue">' +
             '<ng-template ngPluralCase="few"><li>you have a few messages.</li></ng-template>' +
             '<ng-template ngPluralCase="=2">you have two messages.</ng-template>' +
             '</ul>';

         fixture = createTestComponent(template);

         getComponent().switchValue = 2;
         detectChangesAndExpectText('you have two messages.');

         getComponent().switchValue = 3;
         detectChangesAndExpectText('you have a few messages.');
       }));
  });
}

@Injectable()
class TestLocalization extends NgLocalization {
  getPluralCategory(value: number): string {
    if (value > 1 && value < 4) {
      return 'few';
    }

    if (value >= 4 && value < 10) {
      return 'many';
    }

    return 'other';
  }
}

@Component({selector: 'test-cmp', template: ''})
class TestComponent {
  switchValue: number = null;
}

function createTestComponent(template: string): ComponentFixture<TestComponent> {
  return TestBed.overrideComponent(TestComponent, {set: {template: template}})
      .createComponent(TestComponent);
}