Chore: add completion data provider test

public/app/plugins/datasource/loki/components/monaco-query-field/monaco-completion-provider/CompletionDataProvider.test.ts

@@ -0,0 +1,90 @@
+import { HistoryItem } from '@grafana/data';
+
+import LokiLanguageProvider from '../../../LanguageProvider';
+import { LokiDatasource } from '../../../datasource';
+import { createLokiDatasource } from '../../../mocks';
+import { LokiQuery } from '../../../types';
+
+import { CompletionDataProvider } from './CompletionDataProvider';
+import { Label } from './situation';
+
+const history = [
+  {
+    ts: 12345678,
+    query: {
+      refId: 'test-1',
+      expr: '{test: unit}',
+    },
+  },
+  {
+    ts: 87654321,
+    query: {
+      refId: 'test-1',
+      expr: '{unit: test}',
+    },
+  },
+  {
+    ts: 0,
+    query: {
+      refId: 'test-0',
+    },
+  },
+];
+const labelKeys = ['place', 'source'];
+const labelValues = ['moon', 'luna'];
+const otherLabels: Label[] = [
+  {
+    name: 'place',
+    value: 'luna',
+    op: '=',
+  },
+];
+const seriesLabels = { place: ['series', 'labels'], source: [], other: [] };
+const parserAndLabelKeys = {
+  extractedLabelKeys: ['extracted', 'label', 'keys'],
+  hasJSON: true,
+  hasLogfmt: false,
+};
+
+describe('CompletionDataProvider', () => {
+  let completionProvider: CompletionDataProvider, languageProvider: LokiLanguageProvider, datasource: LokiDatasource;
+  beforeEach(() => {
+    datasource = createLokiDatasource();
+    languageProvider = new LokiLanguageProvider(datasource);
+    completionProvider = new CompletionDataProvider(languageProvider, history as Array<HistoryItem<LokiQuery>>);
+
+    jest.spyOn(languageProvider, 'getLabelKeys').mockReturnValue(labelKeys);
+    jest.spyOn(languageProvider, 'getLabelValues').mockResolvedValue(labelValues);
+    jest.spyOn(languageProvider, 'getSeriesLabels').mockResolvedValue(seriesLabels);
+    jest.spyOn(languageProvider, 'getParserAndLabelKeys').mockResolvedValue(parserAndLabelKeys);
+  });
+
+  test('Returns the expected history entries', () => {
+    expect(completionProvider.getHistory()).toEqual(['{test: unit}', '{unit: test}']);
+  });
+
+  test('Returns the expected label names with no other labels', async () => {
+    expect(await completionProvider.getLabelNames([])).toEqual(labelKeys);
+  });
+
+  test('Returns the expected label names with other labels', async () => {
+    expect(await completionProvider.getLabelNames(otherLabels)).toEqual(['source', 'other']);
+  });
+
+  test('Returns the expected label values with no other labels', async () => {
+    expect(await completionProvider.getLabelValues('label', [])).toEqual(labelValues);
+  });
+
+  test('Returns the expected label values with other labels', async () => {
+    expect(await completionProvider.getLabelValues('place', otherLabels)).toEqual(['series', 'labels']);
+    expect(await completionProvider.getLabelValues('other label', otherLabels)).toEqual([]);
+  });
+
+  test('Returns the expected parser and label keys', async () => {
+    expect(await completionProvider.getParserAndLabelKeys([])).toEqual(parserAndLabelKeys);
+  });
+
+  test('Returns the expected series labels', async () => {
+    expect(await completionProvider.getSeriesLabels([])).toEqual(seriesLabels);
+  });
+});