EVOLUTION-MANAGER

Edit File: PromExploreQueryEditor.test.tsx

import React from 'react';
import { shallow, mount } from 'enzyme';
import { act } from 'react-dom/test-utils';
import PromExploreQueryEditor from './PromExploreQueryEditor';
import { PromExploreExtraField } from './PromExploreExtraField';
import { PrometheusDatasource } from '../datasource';
import { PromQuery } from '../types';
import { PanelData, LoadingState, dateTime } from '@grafana/data';

const props: any = {
    query,
    data,
    datasource,
    exploreMode,
    history,
    onChange,
    onRunQuery,
  };

Object.assign(props, propOverrides);

return renderMethod(<PromExploreQueryEditor {...props} />);
};

describe('PromExploreQueryEditor', () => {
  let originalGetSelection: typeof window.getSelection;
  beforeAll(() => {
    originalGetSelection = window.getSelection;
    window.getSelection = () => null;
  });

afterAll(() => {
    window.getSelection = originalGetSelection;
  });

it('should render component', () => {
    const wrapper = setup(shallow);
    expect(wrapper).toMatchSnapshot();
  });

it('should render PromQueryField with ExtraFieldElement', async () => {
    await act(async () => {
      const wrapper = setup(mount);
      expect(wrapper.find(PromExploreExtraField).length).toBe(1);
    });
  });
});