energylink/public/app/plugins/datasource/prometheus/specs/completer_specs.ts

import {describe, it, sinon, expect} from 'test/lib/common';

import {PromCompleter} from '../completer';
import {PrometheusDatasource} from '../datasource';

describe('Prometheus editor completer', function() {
  function getSessionStub(data) {
    return {
      getTokenAt: sinon.stub().returns(data.currentToken),
      getTokens: sinon.stub().returns(data.tokens),
      getLine: sinon.stub().returns(data.line),
    };
  }

  let editor = {};
  let datasourceStub = <PrometheusDatasource>{
    performInstantQuery: sinon
      .stub()
      .withArgs({expr: '{__name__="node_cpu"'})
      .returns(
        Promise.resolve({
          data: {
            data: {
              result: [
                {
                  metric: {
                    job: 'node',
                    instance: 'localhost:9100',
                  },
                },
              ],
            },
          },
        }),
      ),
    performSuggestQuery: sinon
      .stub()
      .withArgs('node', true)
      .returns(Promise.resolve(['node_cpu'])),
  };

  let completer = new PromCompleter(datasourceStub);

  describe('When inside brackets', () => {
    it('Should return range vectors', () => {
      const session = getSessionStub({
        currentToken: {type: 'paren.lparen', value: '[', index: 2, start: 9},
        tokens: [
          {type: 'identifier', value: 'node_cpu'},
          {type: 'paren.lparen', value: '['}
        ],
        line: 'node_cpu[',
      });

      return completer.getCompletions(editor, session, {row: 0, column: 10}, '[', (s, res) => {
        expect(res[0].caption).to.eql('1s');
        expect(res[0].value).to.eql('[1s');
        expect(res[0].meta).to.eql('range vector');
      });
    });
  });

  describe('When inside label matcher, and located at label name', () => {
    it('Should return label name list', () => {
      const session = getSessionStub({
        currentToken: {type: 'entity.name.tag', value: 'j', index: 2, start: 9},
        tokens: [
          {type: 'identifier', value: 'node_cpu'},
          {type: 'paren.lparen', value: '{'},
          {type: 'entity.name.tag', value: 'j', index: 2, start: 9},
          {type: 'paren.rparen', value: '}'},
        ],
        line: 'node_cpu{j}',
      });

      return completer.getCompletions(editor, session, {row: 0, column: 10}, 'j', (s, res) => {
        expect(res[0].meta).to.eql('label name');
      });
    });
  });

  describe('When inside label matcher, and located at label name with __name__ match', () => {
    it('Should return label name list', () => {
      const session = getSessionStub({
        currentToken: {type: 'entity.name.tag', value: 'j', index: 5, start: 22},
        tokens: [
          {type: 'paren.lparen', value: '{'},
          {type: 'entity.name.tag', value: '__name__'},
          {type: 'keyword.operator', value: '=~'},
          {type: 'string.quoted', value: '"node_cpu"'},
          {type: 'punctuation.operator', value: ','},
          {type: 'entity.name.tag', value: 'j', index: 5, start: 22},
          {type: 'paren.rparen', value: '}'},
        ],
        line: '{__name__=~"node_cpu",j}',
      });

      return completer.getCompletions(editor, session, {row: 0, column: 23}, 'j', (s, res) => {
        expect(res[0].meta).to.eql('label name');
      });
    });
  });

  describe('When inside label matcher, and located at label value', () => {
    it('Should return label value list', () => {
      const session = getSessionStub({
        currentToken: {type: 'string.quoted', value: '"n"', index: 4, start: 13},
        tokens: [
          {type: 'identifier', value: 'node_cpu'},
          {type: 'paren.lparen', value: '{'},
          {type: 'entity.name.tag', value: 'job'},
          {type: 'keyword.operator', value: '='},
          {type: 'string.quoted', value: '"n"', index: 4, start: 13},
          {type: 'paren.rparen', value: '}'},
        ],
        line: 'node_cpu{job="n"}',
      });

      return completer.getCompletions(editor, session, {row: 0, column: 15}, 'n', (s, res) => {
        expect(res[0].meta).to.eql('label value');
      });
    });
  });
});