energylink/public/app/features/plugins/specs/datasource_srv.jest.ts

import config from 'app/core/config';
import 'app/features/plugins/datasource_srv';
import { DatasourceSrv } from 'app/features/plugins/datasource_srv';

// Datasource variable $datasource with current value 'BBB'
const templateSrv = {
  variables: [
    {
      type: 'datasource',
      name: 'datasource',
      current: {
        value: 'BBB',
      },
    },
  ],
};

describe('datasource_srv', function() {
  let _datasourceSrv = new DatasourceSrv({}, {}, {}, templateSrv);

  describe('when loading explore sources', () => {
    beforeEach(() => {
      config.datasources = {
        explore1: {
          name: 'explore1',
          meta: { explore: true, metrics: true },
        },
        explore2: {
          name: 'explore2',
          meta: { explore: true, metrics: false },
        },
        nonExplore: {
          name: 'nonExplore',
          meta: { explore: false, metrics: true },
        },
      };
    });

    it('should return list of explore sources', () => {
      const exploreSources = _datasourceSrv.getExploreSources();
      expect(exploreSources.length).toBe(2);
      expect(exploreSources[0].name).toBe('explore1');
      expect(exploreSources[1].name).toBe('explore2');
    });
  });

  describe('when loading metric sources', () => {
    let metricSources;
    let unsortedDatasources = {
      mmm: {
        type: 'test-db',
        meta: { metrics: { m: 1 } },
      },
      '--Grafana--': {
        type: 'grafana',
        meta: { builtIn: true, metrics: { m: 1 }, id: 'grafana' },
      },
      '--Mixed--': {
        type: 'test-db',
        meta: { builtIn: true, metrics: { m: 1 }, id: 'mixed' },
      },
      ZZZ: {
        type: 'test-db',
        meta: { metrics: { m: 1 } },
      },
      aaa: {
        type: 'test-db',
        meta: { metrics: { m: 1 } },
      },
      BBB: {
        type: 'test-db',
        meta: { metrics: { m: 1 } },
      },
    };
    beforeEach(() => {
      config.datasources = unsortedDatasources;
      metricSources = _datasourceSrv.getMetricSources({});
      config.defaultDatasource = 'BBB';
    });

    it('should return a list of sources sorted case insensitively with builtin sources last', () => {
      expect(metricSources[1].name).toBe('aaa');
      expect(metricSources[2].name).toBe('BBB');
      expect(metricSources[3].name).toBe('mmm');
      expect(metricSources[4].name).toBe('ZZZ');
      expect(metricSources[5].name).toBe('--Grafana--');
      expect(metricSources[6].name).toBe('--Mixed--');
    });

    it('should set default data source', () => {
      expect(metricSources[3].name).toBe('default');
      expect(metricSources[3].sort).toBe('BBB');
    });

    it('should set default inject the variable datasources', () => {
      expect(metricSources[0].name).toBe('$datasource');
      expect(metricSources[0].sort).toBe('$datasource');
    });
  });
});