energylink/public/app/plugins/datasource/grafana/specs/datasource.test.ts

import { GrafanaDatasource } from '../datasource';
import q from 'q';
import { dateTime } from '@grafana/ui/src/utils/moment_wrapper';

describe('grafana data source', () => {
  describe('when executing an annotations query', () => {
    let calledBackendSrvParams;
    const backendSrvStub = {
      get: (url: string, options) => {
        calledBackendSrvParams = options;
        return q.resolve([]);
      },
    };

    const templateSrvStub = {
      replace: (val: string) => {
        return val.replace('$var2', 'replaced__delimiter__replaced2').replace('$var', 'replaced');
      },
    };

    const ds = new GrafanaDatasource(backendSrvStub, q, templateSrvStub);

    describe('with tags that have template variables', () => {
      const options = setupAnnotationQueryOptions({ tags: ['tag1:$var'] });

      beforeEach(() => {
        return ds.annotationQuery(options);
      });

      it('should interpolate template variables in tags in query options', () => {
        expect(calledBackendSrvParams.tags[0]).toBe('tag1:replaced');
      });
    });

    describe('with tags that have multi value template variables', () => {
      const options = setupAnnotationQueryOptions({ tags: ['$var2'] });

      beforeEach(() => {
        return ds.annotationQuery(options);
      });

      it('should interpolate template variables in tags in query options', () => {
        expect(calledBackendSrvParams.tags[0]).toBe('replaced');
        expect(calledBackendSrvParams.tags[1]).toBe('replaced2');
      });
    });

    describe('with type dashboard', () => {
      const options = setupAnnotationQueryOptions(
        {
          type: 'dashboard',
          tags: ['tag1'],
        },
        { id: 1 }
      );

      beforeEach(() => {
        return ds.annotationQuery(options);
      });

      it('should remove tags from query options', () => {
        expect(calledBackendSrvParams.tags).toBe(undefined);
      });
    });
  });
});

function setupAnnotationQueryOptions(annotation, dashboard?) {
  return {
    annotation: annotation,
    dashboard: dashboard,
    range: {
      from: dateTime(1432288354),
      to: dateTime(1432288401),
    },
    rangeRaw: { from: 'now-24h', to: 'now' },
  };
}