grafana/public/test/specs/graphiteTargetCtrl-specs.js

define([
  'helpers',
  'plugins/datasource/graphite/gfunc',
  'plugins/datasource/graphite/queryCtrl'
], function(helpers, gfunc) {
  'use strict';

  describe('GraphiteQueryCtrl', function() {
    var ctx = new helpers.ControllerTestContext();

    beforeEach(module('grafana.controllers'));
    beforeEach(ctx.providePhase());
    beforeEach(ctx.createControllerPhase('GraphiteQueryCtrl'));

    beforeEach(function() {
      ctx.scope.target = {target: 'aliasByNode(scaleToSeconds(test.prod.*,1),2)'};

      ctx.scope.datasource = ctx.datasource;
      ctx.scope.datasource.metricFindQuery = sinon.stub().returns(ctx.$q.when([]));
    });

    describe('init', function() {
      beforeEach(function() {
        ctx.scope.init();
        ctx.scope.$digest();
      });

      it('should validate metric key exists', function() {
        expect(ctx.scope.datasource.metricFindQuery.getCall(0).args[0]).to.be('test.prod.*');
      });

      it('should delete last segment if no metrics are found', function() {
        expect(ctx.scope.segments[2].value).to.be('select metric');
      });

      it('should parse expression and build function model', function() {
        expect(ctx.scope.functions.length).to.be(2);
      });
    });

    describe('when adding function', function() {
      beforeEach(function() {
        ctx.scope.target.target = 'test.prod.*.count';
        ctx.scope.datasource.metricFindQuery.returns(ctx.$q.when([{expandable: false}]));
        ctx.scope.init();
        ctx.scope.$digest();

        ctx.scope.$parent = { get_data: sinon.spy() };
        ctx.scope.addFunction(gfunc.getFuncDef('aliasByNode'));
      });

      it('should add function with correct node number', function() {
        expect(ctx.scope.functions[0].params[0]).to.be(2);
      });

      it('should update target', function() {
        expect(ctx.scope.target.target).to.be('aliasByNode(test.prod.*.count, 2)');
      });

      it('should call get_data', function() {
        expect(ctx.scope.$parent.get_data.called).to.be(true);
      });
    });

    describe('when adding function before any metric segment', function() {
      beforeEach(function() {
        ctx.scope.target.target = '';
        ctx.scope.datasource.metricFindQuery.returns(ctx.$q.when([{expandable: true}]));
        ctx.scope.init();
        ctx.scope.$digest();

        ctx.scope.$parent = { get_data: sinon.spy() };
        ctx.scope.addFunction(gfunc.getFuncDef('asPercent'));
      });

      it('should add function and remove select metric link', function() {
        expect(ctx.scope.segments.length).to.be(0);
      });
    });

    describe('when initalizing target without metric expression and only function', function() {
      beforeEach(function() {
        ctx.scope.target.target = 'asPercent(#A, #B)';
        ctx.scope.datasource.metricFindQuery.returns(ctx.$q.when([]));
        ctx.scope.init();
        ctx.scope.$digest();
        ctx.scope.$parent = { get_data: sinon.spy() };
      });

      it('should not add select metric segment', function() {
        expect(ctx.scope.segments.length).to.be(0);
      });

      it('should add both series refs as params', function() {
        expect(ctx.scope.functions[0].params.length).to.be(2);
      });

    });

    describe('when initializing a target with single param func using variable', function() {
      beforeEach(function() {
        ctx.scope.target.target = 'movingAverage(prod.count, $var)';
        ctx.scope.datasource.metricFindQuery.returns(ctx.$q.when([]));
        ctx.scope.init();
        ctx.scope.$digest();
        ctx.scope.$parent = { get_data: sinon.spy() };
      });

      it('should add 2 segments', function() {
        expect(ctx.scope.segments.length).to.be(2);
      });

      it('should add function param', function() {
        expect(ctx.scope.functions[0].params.length).to.be(1);
      });

    });

    describe('when initalizing target without metric expression and function with series-ref', function() {
      beforeEach(function() {
        ctx.scope.target.target = 'asPercent(metric.node.count, #A)';
        ctx.scope.datasource.metricFindQuery.returns(ctx.$q.when([]));
        ctx.scope.init();
        ctx.scope.$digest();
        ctx.scope.$parent = { get_data: sinon.spy() };
      });

      it('should add segments', function() {
        expect(ctx.scope.segments.length).to.be(3);
      });

      it('should have correct func params', function() {
        expect(ctx.scope.functions[0].params.length).to.be(1);
      });
    });

    describe('when getting altSegments and metricFindQuery retuns empty array', function() {
      beforeEach(function() {
        ctx.scope.target.target = 'test.count';
        ctx.scope.datasource.metricFindQuery.returns(ctx.$q.when([]));
        ctx.scope.init();
        ctx.scope.getAltSegments(1).then(function(results) {
          ctx.altSegments = results;
        });
        ctx.scope.$digest();
        ctx.scope.$parent = { get_data: sinon.spy() };
      });

      it('should have no segments', function() {
        expect(ctx.altSegments.length).to.be(0);
      });

    });

    describe('targetChanged', function() {
      beforeEach(function() {
        ctx.scope.datasource.metricFindQuery.returns(ctx.$q.when([{expandable: false}]));
        ctx.scope.init();
        ctx.scope.$digest();

        ctx.scope.$parent = { get_data: sinon.spy() };
        ctx.scope.target.target = '';
        ctx.scope.targetChanged();
      });

      it('should rebuld target after expression model', function() {
        expect(ctx.scope.target.target).to.be('aliasByNode(scaleToSeconds(test.prod.*, 1), 2)');
      });

      it('should call get_data', function() {
        expect(ctx.scope.$parent.get_data.called).to.be(true);
      });
    });


  });
});