energylink/public/app/plugins/datasource/opentsdb/specs/query-ctrl-specs.ts

import {
  describe,
  beforeEach,
  it,
  sinon,
  expect,
  angularMocks
} from "test/lib/common";
import helpers from "test/specs/helpers";
import { OpenTsQueryCtrl } from "../query_ctrl";

describe("OpenTsQueryCtrl", function() {
  var ctx = new helpers.ControllerTestContext();

  beforeEach(angularMocks.module("grafana.core"));
  beforeEach(angularMocks.module("grafana.services"));
  beforeEach(
    angularMocks.module(function($compileProvider) {
      $compileProvider.preAssignBindingsEnabled(true);
    })
  );

  beforeEach(ctx.providePhase(["backendSrv", "templateSrv"]));

  beforeEach(ctx.providePhase());
  beforeEach(
    angularMocks.inject(($rootScope, $controller, $q) => {
      ctx.$q = $q;
      ctx.scope = $rootScope.$new();
      ctx.target = { target: "" };
      ctx.panelCtrl = {
        panel: {
          targets: [ctx.target]
        }
      };
      ctx.panelCtrl.refresh = sinon.spy();
      ctx.datasource.getAggregators = sinon.stub().returns(ctx.$q.when([]));
      ctx.datasource.getFilterTypes = sinon.stub().returns(ctx.$q.when([]));

      ctx.ctrl = $controller(
        OpenTsQueryCtrl,
        { $scope: ctx.scope },
        {
          panelCtrl: ctx.panelCtrl,
          datasource: ctx.datasource,
          target: ctx.target
        }
      );
      ctx.scope.$digest();
    })
  );

  describe("init query_ctrl variables", function() {
    it("filter types should be initialized", function() {
      expect(ctx.ctrl.filterTypes.length).to.be(7);
    });

    it("aggregators should be initialized", function() {
      expect(ctx.ctrl.aggregators.length).to.be(8);
    });

    it("fill policy options should be initialized", function() {
      expect(ctx.ctrl.fillPolicies.length).to.be(4);
    });
  });

  describe("when adding filters and tags", function() {
    it("addTagMode should be false when closed", function() {
      ctx.ctrl.addTagMode = true;
      ctx.ctrl.closeAddTagMode();
      expect(ctx.ctrl.addTagMode).to.be(false);
    });

    it("addFilterMode should be false when closed", function() {
      ctx.ctrl.addFilterMode = true;
      ctx.ctrl.closeAddFilterMode();
      expect(ctx.ctrl.addFilterMode).to.be(false);
    });

    it("removing a tag from the tags list", function() {
      ctx.ctrl.target.tags = { tagk: "tag_key", tagk2: "tag_value2" };
      ctx.ctrl.removeTag("tagk");
      expect(Object.keys(ctx.ctrl.target.tags).length).to.be(1);
    });

    it("removing a filter from the filters list", function() {
      ctx.ctrl.target.filters = [
        {
          tagk: "tag_key",
          filter: "tag_value2",
          type: "wildcard",
          groupBy: true
        }
      ];
      ctx.ctrl.removeFilter(0);
      expect(ctx.ctrl.target.filters.length).to.be(0);
    });

    it("adding a filter when tags exist should generate error", function() {
      ctx.ctrl.target.tags = { tagk: "tag_key", tagk2: "tag_value2" };
      ctx.ctrl.addFilter();
      expect(ctx.ctrl.errors.filters).to.be(
        "Please remove tags to use filters, tags and filters are mutually exclusive."
      );
    });

    it("adding a tag when filters exist should generate error", function() {
      ctx.ctrl.target.filters = [
        {
          tagk: "tag_key",
          filter: "tag_value2",
          type: "wildcard",
          groupBy: true
        }
      ];
      ctx.ctrl.addTag();
      expect(ctx.ctrl.errors.tags).to.be(
        "Please remove filters to use tags, tags and filters are mutually exclusive."
      );
    });
  });
});