kibana-with-account/src/ui/public/agg_response/tabify/__tests__/_get_columns.js

kibana-with-account is develop based kibana project

describe('get columns', function () {
  var getColumns;
  var Vis;
  var indexPattern;
  var expect = require('expect.js');
  var ngMock = require('ngMock');

  beforeEach(ngMock.module('kibana'));
  beforeEach(ngMock.inject(function (Private, $injector) {
    getColumns = Private(require('ui/agg_response/tabify/_get_columns'));
    Vis = Private(require('ui/Vis'));
    indexPattern = Private(require('fixtures/stubbed_logstash_index_pattern'));
  }));

  it('should inject a count metric if no aggs exist', function () {
    var vis = new Vis(indexPattern, {
      type: 'pie'
    });
    while (vis.aggs.length) vis.aggs.pop();
    var columns = getColumns(vis);

    expect(columns).to.have.length(1);
    expect(columns[0]).to.have.property('aggConfig');
    expect(columns[0].aggConfig.type).to.have.property('name', 'count');
  });

  it('should inject a count metric if only buckets exist', function () {
    var vis = new Vis(indexPattern, {
      type: 'pie',
      aggs: [
        { type: 'date_histogram', schema: 'segment',  params: { field: '@timestamp' } }
      ]
    });

    var columns = getColumns(vis);

    expect(columns).to.have.length(2);
    expect(columns[1]).to.have.property('aggConfig');
    expect(columns[1].aggConfig.type).to.have.property('name', 'count');
  });

  it('should inject the metric after each bucket if the vis is hierarchical', function () {
    var vis = new Vis(indexPattern, {
      type: 'pie',
      aggs: [
        { type: 'date_histogram', schema: 'segment',  params: { field: '@timestamp' } },
        { type: 'date_histogram', schema: 'segment',  params: { field: '@timestamp' } },
        { type: 'date_histogram', schema: 'segment',  params: { field: '@timestamp' } },
        { type: 'date_histogram', schema: 'segment',  params: { field: '@timestamp' } }
      ]
    });

    var columns = getColumns(vis);

    expect(columns).to.have.length(8);
    columns.forEach(function (column, i) {
      expect(column).to.have.property('aggConfig');
      expect(column.aggConfig.type).to.have.property('name', i % 2 ? 'count' : 'date_histogram');
    });
  });

  it('should inject the multiple metrics after each bucket if the vis is hierarchical', function () {
    var vis = new Vis(indexPattern, {
      type: 'pie',
      aggs: [
        { type: 'date_histogram', schema: 'segment',  params: { field: '@timestamp' } },
        { type: 'avg',            schema: 'metric',   params: { field: 'bytes' } },
        { type: 'date_histogram', schema: 'segment',  params: { field: '@timestamp' } },
        { type: 'date_histogram', schema: 'segment',  params: { field: '@timestamp' } },
        { type: 'sum',            schema: 'metric',   params: { field: 'bytes' } },
        { type: 'date_histogram', schema: 'segment',  params: { field: '@timestamp' } }
      ]
    });

    var columns = getColumns(vis);

    function checkColumns(column, i) {
      expect(column).to.have.property('aggConfig');
      switch (i) {
        case 0:
          expect(column.aggConfig.type).to.have.property('name', 'date_histogram');
          break;
        case 1:
          expect(column.aggConfig.type).to.have.property('name', 'avg');
          break;
        case 2:
          expect(column.aggConfig.type).to.have.property('name', 'sum');
          break;
      }
    }

    expect(columns).to.have.length(12);
    for (var i = 0; i < columns.length; i += 3) {
      var counts = { buckets: 0, metrics: 0 };
      columns.slice(i, i + 3).forEach(checkColumns);
    }
  });

  it('should put all metrics at the end of the columns if the vis is not hierarchical', function () {
    var vis = new Vis(indexPattern, {
      type: 'histogram',
      aggs: [
        { type: 'date_histogram', schema: 'segment',  params: { field: '@timestamp' } },
        { type: 'avg',            schema: 'metric',   params: { field: 'bytes' } },
        { type: 'date_histogram', schema: 'segment',  params: { field: '@timestamp' } },
        { type: 'date_histogram', schema: 'segment',  params: { field: '@timestamp' } },
        { type: 'sum',            schema: 'metric',   params: { field: 'bytes' } },
        { type: 'date_histogram', schema: 'segment',  params: { field: '@timestamp' } }
      ]
    });

    var columns = getColumns(vis);
    expect(columns).to.have.length(6);

    // sum should be last
    expect(columns.pop().aggConfig.type).to.have.property('name', 'sum');
    // avg should be before that
    expect(columns.pop().aggConfig.type).to.have.property('name', 'avg');
    // the rest are date_histograms
    while (columns.length) {
      expect(columns.pop().aggConfig.type).to.have.property('name', 'date_histogram');
    }
  });
});