angular-cn/modules/benchpress/test/metric/multi_metric_spec.js

import {ddescribe, describe, it, iit, xit, expect, beforeEach, afterEach} from 'angular2/test_lib';

import { List, ListWrapper, StringMap } from 'angular2/src/facade/collection';
import { PromiseWrapper, Promise } from 'angular2/src/facade/async';

import { Metric, MultiMetric, bind, Injector } from 'benchpress/common';

export function main() {
  function createMetric(ids) {
    return new Injector([
      ListWrapper.map(ids, (id) => bind(id).toValue(new MockMetric(id)) ),
      MultiMetric.createBindings(ids)
    ]).asyncGet(MultiMetric);
  }

  describe('multi metric', () => {

    it('should merge descriptions', (done) => {
      createMetric(['m1', 'm2']).then( (m) => {
        expect(m.describe()).toEqual({
          'm1': 'describe', 'm2': 'describe'
        });
        done();
      });
    });

    it('should merge all beginMeasure calls', (done) => {
      createMetric(['m1', 'm2'])
        .then( (m) => m.beginMeasure() )
        .then( (values) => {
          expect(values).toEqual([
            'm1_beginMeasure', 'm2_beginMeasure'
          ]);
          done();
        });
    });

    [false, true].forEach( (restartFlag) => {
      it(`should merge all endMeasure calls for restart=${restartFlag}`, (done) => {
        createMetric(['m1', 'm2'])
          .then( (m) => m.endMeasure(restartFlag) )
          .then( (values) => {
            expect(values).toEqual({
              'm1': { 'restart': restartFlag },
              'm2': { 'restart': restartFlag }
            });
            done();
          });
      });
    });

  });
}

class MockMetric extends Metric {
  _id:string;

  constructor(id) {
    super();
    this._id = id;
  }

  beginMeasure():Promise {
    return PromiseWrapper.resolve(`${this._id}_beginMeasure`);
  }

  endMeasure(restart:boolean):Promise<StringMap> {
    var result = {};
    result[this._id] = {
      'restart': restart
    };
    return PromiseWrapper.resolve(result);
  }

  describe():StringMap {
    var result = {};
    result[this._id] = 'describe';
    return result;
  }
}