define([ 'helpers', 'plugins/datasource/graphite/gfunc', 'plugins/datasource/graphite/queryCtrl', 'services/uiSegmentSrv' ], function(helpers, gfunc) { 'use strict'; describe('GraphiteQueryCtrl', function() { var ctx = new helpers.ControllerTestContext(); beforeEach(module('grafana.controllers')); beforeEach(module('grafana.services')); 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); }); }); }); });