IntenseWebs/grafana
3
0
Fork 0
mirror of https://github.com/grafana/grafana.git synced 2025-02-25 18:55:37 -06:00
Code Issues Packages Projects Releases Wiki Activity

Chore: update language provider test

Browse Source
This commit is contained in:
Matias Chomicki 2022-09-29 15:19:59 +02:00
parent f6b6c69194
commit 22e547f625
1 changed files with 49 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
public/app/plugins/datasource/loki/LanguageProvider.test.ts
View File

@ -1,14 +1,17 @@
import Plain from 'slate-plain-serializer';
import { AbstractLabelOperator } from '@grafana/data';
import { AbstractLabelOperator, DataFrame } from '@grafana/data';
import { TypeaheadInput } from '@grafana/ui';
import { TemplateSrv } from 'app/features/templating/template_srv';
import { executeAnnotationQuery } from 'app/features/annotations/executeAnnotationQuery';
import LanguageProvider, { LokiHistoryItem } from './LanguageProvider';
import { LokiDatasource } from './datasource';
import { createLokiDatasource, createMetadataRequest } from './mocks';
import { extractLogParserFromDataFrame } from './responseUtils';
import { LokiQueryType } from './types';
jest.mock('./responseUtils');
jest.mock('app/store/store', () => ({
store: {
getState: jest.fn().mockReturnValue({
@ -297,6 +300,49 @@ describe('Query imports', () => {
});
});
});
describe('getParserAndLabelKeys()', () => {
let datasource: LokiDatasource, languageProvider: LanguageProvider;
const extractLogParserFromDataFrameMock = extractLogParserFromDataFrame as jest.Mock;
beforeEach(() => {
datasource = createLokiDatasource();
languageProvider = new LanguageProvider(datasource);
});
it('identifies selectors with JSON parser data', async () => {
jest.spyOn(datasource, 'getDataSamples').mockResolvedValue([{}] as DataFrame[]);
extractLogParserFromDataFrameMock.mockReturnValueOnce({ hasLogfmt: false, hasJSON: true });
expect(await languageProvider.getParserAndLabelKeys('{place="luna"}')).toEqual({
extractedLabelKeys: [],
hasJSON: true,
hasLogfmt: false,
});
});
it('identifies selectors with Logfmt parser data', async () => {
jest.spyOn(datasource, 'getDataSamples').mockResolvedValue([{}] as DataFrame[]);
extractLogParserFromDataFrameMock.mockReturnValueOnce({ hasLogfmt: true, hasJSON: false });
expect(await languageProvider.getParserAndLabelKeys('{place="luna"}')).toEqual({
extractedLabelKeys: [],
hasJSON: false,
hasLogfmt: true,
});
});
it('correctly processes empty data', async () => {
jest.spyOn(datasource, 'getDataSamples').mockResolvedValue([]);
extractLogParserFromDataFrameMock.mockClear();
expect(await languageProvider.getParserAndLabelKeys('{place="luna"}')).toEqual({
extractedLabelKeys: [],
hasJSON: false,
hasLogfmt: false,
});
expect(extractLogParserFromDataFrameMock).not.toHaveBeenCalled();
});
});
});
async function getLanguageProvider(datasource: LokiDatasource) {
@ -334,7 +380,7 @@ function setup(
labelsAndValues: Record<string, string[]>,
series?: Record<string, Array<Record<string, string>>>
): LokiDatasource {
const datasource = createLokiDatasource({} as unknown as TemplateSrv);
const datasource = createLokiDatasource();
const rangeMock = {
start: 1560153109000,