ck-ops/code/grafana/public/app/plugins/datasource/loki/querybuilder/state.test.ts

import { QueryEditorMode } from '../../prometheus/querybuilder/shared/types';

import { changeEditorMode, getQueryWithDefaults } from './state';

describe('getQueryWithDefaults(', () => {
  it('should set defaults', () => {
    expect(getQueryWithDefaults({ refId: 'A' } as any)).toEqual({
      editorMode: 'builder',
      expr: '',
      queryType: 'range',
      refId: 'A',
    });
  });

  it('changing editor mode with blank query should change default', () => {
    changeEditorMode({ refId: 'A', expr: '' }, QueryEditorMode.Code, (query) => {
      expect(query.editorMode).toBe(QueryEditorMode.Code);
    });

    expect(getQueryWithDefaults({ refId: 'A' } as any).editorMode).toEqual(QueryEditorMode.Code);
  });
});