mirror of https://github.com/grafana/grafana
Explore: Support mixed data sources for supplementary query (#63036)
* Consolidate logs volume logic (full range and limited) * Fix showing limited histogram message * Test passing meta data to logs volume provider * Improve readability * Clean up types * Add basic support for multiple log volumes * Move the comment back to the right place * Improve readability * Clean up the logic to support Logs Samples * Update docs * Sort log volumes * Provide title to logs volume panel * Move logs volume cache to the provider factory * Add helper functions * Reuse only if queries are the same * Fix alphabetical sorting * Move caching out of the provider * Support errors and loading state * Remove unused code * Consolidate supplementary query utils * Add tests for supplementaryQueries * Update tests * Simplify logs volume extra info * Update tests * Remove comment * Update tests * Fix hiding the histogram for hidden queries * Simplify loading message * Update tests * Wait for full fallback histogram to load before showing it * Fix a typo * Add feedback comments * Move feedback comments to github * Do not filter out hidden queries as they may be used as references in other queries * Group log volume by refId * Support showing fallback histograms per query to avoid duplicates * Improve type-checking * Fix supplementaryQueries.test.ts * Fix logsModel.test.ts * Fix loading fallback results * Fix unit tests * WIP * Update deprecated styles * Simplify test * Simplify rendering zoom info * Update deprecated styles * Simplify getLogsVolumeDataSourceInfo * Simplify isLogsVolumeLimited() * Simplify rendering zoom infopull/63734/head^2
Piotr Jamróz
2 years ago
committed by
GitHub
parent
b2d7bea78b
commit
a7238ba933
@ -0,0 +1,54 @@ |
||||
import { fireEvent, render, screen } from '@testing-library/react'; |
||||
import React from 'react'; |
||||
|
||||
import { DataQueryResponse, LoadingState, EventBusSrv } from '@grafana/data'; |
||||
|
||||
import { LogsVolumePanelList } from './LogsVolumePanelList'; |
||||
|
||||
jest.mock('./Graph/ExploreGraph', () => { |
||||
const ExploreGraph = () => <span>ExploreGraph</span>; |
||||
return { |
||||
ExploreGraph, |
||||
}; |
||||
}); |
||||
|
||||
function renderPanel(logsVolumeData?: DataQueryResponse) { |
||||
render( |
||||
<LogsVolumePanelList |
||||
absoluteRange={{ from: 0, to: 1 }} |
||||
timeZone="timeZone" |
||||
splitOpen={() => {}} |
||||
width={100} |
||||
onUpdateTimeRange={() => {}} |
||||
logsVolumeData={logsVolumeData} |
||||
onLoadLogsVolume={() => {}} |
||||
onHiddenSeriesChanged={() => null} |
||||
eventBus={new EventBusSrv()} |
||||
/> |
||||
); |
||||
} |
||||
|
||||
describe('LogsVolumePanelList', () => { |
||||
it('shows loading message', () => { |
||||
renderPanel({ state: LoadingState.Loading, error: undefined, data: [] }); |
||||
expect(screen.getByText('Loading...')).toBeInTheDocument(); |
||||
}); |
||||
|
||||
it('shows short warning message', () => { |
||||
renderPanel({ state: LoadingState.Error, error: { data: { message: 'Test error message' } }, data: [] }); |
||||
expect(screen.getByText('Failed to load log volume for this query')).toBeInTheDocument(); |
||||
expect(screen.getByText('Test error message')).toBeInTheDocument(); |
||||
}); |
||||
|
||||
it('shows long warning message', () => { |
||||
// we make a long message
|
||||
const messagePart = 'One two three four five six seven eight nine ten.'; |
||||
const message = messagePart + ' ' + messagePart + ' ' + messagePart; |
||||
|
||||
renderPanel({ state: LoadingState.Error, error: { data: { message } }, data: [] }); |
||||
expect(screen.getByText('Failed to load log volume for this query')).toBeInTheDocument(); |
||||
expect(screen.queryByText(message)).not.toBeInTheDocument(); |
||||
fireEvent.click(screen.getByRole('button', { name: 'Show details' })); |
||||
expect(screen.getByText(message)).toBeInTheDocument(); |
||||
}); |
||||
}); |
||||
@ -0,0 +1,120 @@ |
||||
import { css } from '@emotion/css'; |
||||
import { groupBy } from 'lodash'; |
||||
import React, { useMemo } from 'react'; |
||||
|
||||
import { |
||||
AbsoluteTimeRange, |
||||
DataFrame, |
||||
DataQueryResponse, |
||||
EventBus, |
||||
GrafanaTheme2, |
||||
isLogsVolumeLimited, |
||||
LoadingState, |
||||
SplitOpen, |
||||
TimeZone, |
||||
} from '@grafana/data'; |
||||
import { Button, InlineField, useStyles2 } from '@grafana/ui'; |
||||
|
||||
import { LogsVolumePanel } from './LogsVolumePanel'; |
||||
import { SupplementaryResultError } from './SupplementaryResultError'; |
||||
|
||||
type Props = { |
||||
logsVolumeData: DataQueryResponse | undefined; |
||||
absoluteRange: AbsoluteTimeRange; |
||||
timeZone: TimeZone; |
||||
splitOpen: SplitOpen; |
||||
width: number; |
||||
onUpdateTimeRange: (timeRange: AbsoluteTimeRange) => void; |
||||
onLoadLogsVolume: () => void; |
||||
onHiddenSeriesChanged: (hiddenSeries: string[]) => void; |
||||
eventBus: EventBus; |
||||
}; |
||||
|
||||
export const LogsVolumePanelList = ({ |
||||
logsVolumeData, |
||||
absoluteRange, |
||||
onUpdateTimeRange, |
||||
width, |
||||
onLoadLogsVolume, |
||||
onHiddenSeriesChanged, |
||||
eventBus, |
||||
splitOpen, |
||||
timeZone, |
||||
}: Props) => { |
||||
const logVolumes = useMemo( |
||||
() => groupBy(logsVolumeData?.data || [], 'meta.custom.sourceQuery.refId'), |
||||
[logsVolumeData] |
||||
); |
||||
|
||||
const styles = useStyles2(getStyles); |
||||
|
||||
const numberOfLogVolumes = Object.keys(logVolumes).length; |
||||
|
||||
const containsZoomed = Object.values(logVolumes).some((data: DataFrame[]) => { |
||||
const zoomRatio = logsLevelZoomRatio(data, absoluteRange); |
||||
return !isLogsVolumeLimited(data) && zoomRatio && zoomRatio < 1; |
||||
}); |
||||
|
||||
if (logsVolumeData?.state === LoadingState.Loading) { |
||||
return <span>Loading...</span>; |
||||
} |
||||
if (logsVolumeData?.error !== undefined) { |
||||
return <SupplementaryResultError error={logsVolumeData.error} title="Failed to load log volume for this query" />; |
||||
} |
||||
return ( |
||||
<div className={styles.listContainer}> |
||||
{Object.keys(logVolumes).map((name, index) => { |
||||
const logsVolumeData = { data: logVolumes[name] }; |
||||
return ( |
||||
<LogsVolumePanel |
||||
key={index} |
||||
absoluteRange={absoluteRange} |
||||
width={width} |
||||
logsVolumeData={logsVolumeData} |
||||
onUpdateTimeRange={onUpdateTimeRange} |
||||
timeZone={timeZone} |
||||
splitOpen={splitOpen} |
||||
onLoadLogsVolume={onLoadLogsVolume} |
||||
// TODO: Support filtering level from multiple log levels
|
||||
onHiddenSeriesChanged={numberOfLogVolumes > 1 ? () => {} : onHiddenSeriesChanged} |
||||
eventBus={eventBus} |
||||
/> |
||||
); |
||||
})} |
||||
{containsZoomed && ( |
||||
<div className={styles.extraInfoContainer}> |
||||
<InlineField label="Reload log volume" transparent> |
||||
<Button size="xs" icon="sync" variant="secondary" onClick={onLoadLogsVolume} id="reload-volume" /> |
||||
</InlineField> |
||||
</div> |
||||
)} |
||||
</div> |
||||
); |
||||
}; |
||||
|
||||
const getStyles = (theme: GrafanaTheme2) => { |
||||
return { |
||||
listContainer: css` |
||||
padding-top: 10px; |
||||
`,
|
||||
extraInfoContainer: css` |
||||
display: flex; |
||||
justify-content: end; |
||||
position: absolute; |
||||
right: 5px; |
||||
top: 5px; |
||||
`,
|
||||
oldInfoText: css` |
||||
font-size: ${theme.typography.bodySmall.fontSize}; |
||||
color: ${theme.colors.text.secondary}; |
||||
`,
|
||||
}; |
||||
}; |
||||
|
||||
function logsLevelZoomRatio( |
||||
logsVolumeData: DataFrame[] | undefined, |
||||
selectedTimeRange: AbsoluteTimeRange |
||||
): number | undefined { |
||||
const dataRange = logsVolumeData && logsVolumeData[0] && logsVolumeData[0].meta?.custom?.absoluteRange; |
||||
return dataRange ? (selectedTimeRange.from - selectedTimeRange.to) / (dataRange.from - dataRange.to) : undefined; |
||||
} |
||||
@ -0,0 +1,37 @@ |
||||
import { Observable, of } from 'rxjs'; |
||||
|
||||
import { getDefaultTimeRange, LoadingState, LogsModel } from '@grafana/data'; |
||||
|
||||
import { ExplorePanelData } from '../../../types'; |
||||
|
||||
type MockProps = { |
||||
logsResult?: Partial<LogsModel>; |
||||
}; |
||||
|
||||
export const mockExplorePanelData = (props?: MockProps): Observable<ExplorePanelData> => { |
||||
const data: ExplorePanelData = { |
||||
flameGraphFrames: [], |
||||
graphFrames: [], |
||||
graphResult: [], |
||||
logsFrames: [], |
||||
logsResult: { |
||||
hasUniqueLabels: false, |
||||
rows: [], |
||||
meta: [], |
||||
series: [], |
||||
queries: [], |
||||
...(props?.logsResult || {}), |
||||
}, |
||||
nodeGraphFrames: [], |
||||
rawPrometheusFrames: [], |
||||
rawPrometheusResult: null, |
||||
series: [], |
||||
state: LoadingState.Done, |
||||
tableFrames: [], |
||||
tableResult: [], |
||||
timeRange: getDefaultTimeRange(), |
||||
traceFrames: [], |
||||
}; |
||||
|
||||
return of(data); |
||||
}; |
||||
@ -1,49 +0,0 @@ |
||||
import { Observable } from 'rxjs'; |
||||
|
||||
import { |
||||
DataSourceApi, |
||||
SupplementaryQueryType, |
||||
DataQueryResponse, |
||||
hasSupplementaryQuerySupport, |
||||
DataQueryRequest, |
||||
LoadingState, |
||||
LogsVolumeType, |
||||
} from '@grafana/data'; |
||||
|
||||
import { ExplorePanelData } from '../../../types'; |
||||
|
||||
export const getSupplementaryQueryProvider = ( |
||||
datasourceInstance: DataSourceApi, |
||||
type: SupplementaryQueryType, |
||||
request: DataQueryRequest, |
||||
explorePanelData: Observable<ExplorePanelData> |
||||
): Observable<DataQueryResponse> | undefined => { |
||||
if (hasSupplementaryQuerySupport(datasourceInstance, type)) { |
||||
return datasourceInstance.getDataProvider(type, request); |
||||
} else if (type === SupplementaryQueryType.LogsVolume) { |
||||
// Create a fallback to results based logs volume
|
||||
return new Observable<DataQueryResponse>((observer) => { |
||||
explorePanelData.subscribe((exploreData) => { |
||||
if (exploreData.logsResult?.series && exploreData.logsResult?.visibleRange) { |
||||
observer.next({ |
||||
data: exploreData.logsResult.series.map((d) => { |
||||
const custom = d.meta?.custom || {}; |
||||
return { |
||||
...d, |
||||
meta: { |
||||
custom: { |
||||
...custom, |
||||
logsVolumeType: LogsVolumeType.Limited, |
||||
absoluteRange: exploreData.logsResult?.visibleRange, |
||||
}, |
||||
}, |
||||
}; |
||||
}), |
||||
state: LoadingState.Done, |
||||
}); |
||||
} |
||||
}); |
||||
}); |
||||
} |
||||
return undefined; |
||||
}; |
||||
@ -0,0 +1,357 @@ |
||||
import { flatten } from 'lodash'; |
||||
import { from, Observable } from 'rxjs'; |
||||
|
||||
import { |
||||
DataFrame, |
||||
DataQueryRequest, |
||||
DataQueryResponse, |
||||
DataSourceApi, |
||||
DataSourceWithSupplementaryQueriesSupport, |
||||
FieldType, |
||||
LoadingState, |
||||
LogLevel, |
||||
LogsVolumeType, |
||||
MutableDataFrame, |
||||
SupplementaryQueryType, |
||||
toDataFrame, |
||||
} from '@grafana/data'; |
||||
import { getDataSourceSrv } from '@grafana/runtime'; |
||||
import { DataQuery } from '@grafana/schema'; |
||||
|
||||
import { MockDataSourceApi } from '../../../../test/mocks/datasource_srv'; |
||||
import { MockDataQueryRequest, MockQuery } from '../../../../test/mocks/query'; |
||||
import { ExplorePanelData } from '../../../types'; |
||||
import { mockExplorePanelData } from '../__mocks__/data'; |
||||
|
||||
import { getSupplementaryQueryProvider } from './supplementaryQueries'; |
||||
|
||||
class MockDataSourceWithSupplementaryQuerySupport |
||||
extends MockDataSourceApi |
||||
implements DataSourceWithSupplementaryQueriesSupport<DataQuery> |
||||
{ |
||||
private supplementaryQueriesResults: Record<SupplementaryQueryType, DataFrame[] | undefined> = { |
||||
[SupplementaryQueryType.LogsVolume]: undefined, |
||||
[SupplementaryQueryType.LogsSample]: undefined, |
||||
}; |
||||
|
||||
withSupplementaryQuerySupport(type: SupplementaryQueryType, data: DataFrame[]) { |
||||
this.supplementaryQueriesResults[type] = data; |
||||
return this; |
||||
} |
||||
|
||||
getDataProvider( |
||||
type: SupplementaryQueryType, |
||||
request: DataQueryRequest<DataQuery> |
||||
): Observable<DataQueryResponse> | undefined { |
||||
const data = this.supplementaryQueriesResults[type]; |
||||
if (data) { |
||||
return from([ |
||||
{ state: LoadingState.Loading, data: [] }, |
||||
{ state: LoadingState.Done, data }, |
||||
]); |
||||
} |
||||
return undefined; |
||||
} |
||||
|
||||
getSupplementaryQuery(type: SupplementaryQueryType, query: DataQuery): DataQuery | undefined { |
||||
return query; |
||||
} |
||||
|
||||
getSupportedSupplementaryQueryTypes(): SupplementaryQueryType[] { |
||||
return Object.values(SupplementaryQueryType).filter((type) => this.supplementaryQueriesResults[type]); |
||||
} |
||||
} |
||||
|
||||
const createSupplementaryQueryResponse = (type: SupplementaryQueryType, id: string) => { |
||||
return [ |
||||
toDataFrame({ |
||||
refId: `1-${type}-${id}`, |
||||
fields: [{ name: 'value', type: FieldType.string, values: [1] }], |
||||
meta: { |
||||
custom: { |
||||
logsVolumeType: LogsVolumeType.FullRange, |
||||
}, |
||||
}, |
||||
}), |
||||
toDataFrame({ |
||||
refId: `2-${type}-${id}`, |
||||
fields: [{ name: 'value', type: FieldType.string, values: [2] }], |
||||
meta: { |
||||
custom: { |
||||
logsVolumeType: LogsVolumeType.FullRange, |
||||
}, |
||||
}, |
||||
}), |
||||
]; |
||||
}; |
||||
|
||||
const mockRow = (refId: string) => { |
||||
return { |
||||
rowIndex: 0, |
||||
entryFieldIndex: 0, |
||||
dataFrame: new MutableDataFrame({ refId, fields: [{ name: 'A', values: [] }] }), |
||||
entry: '', |
||||
hasAnsi: false, |
||||
hasUnescapedContent: false, |
||||
labels: {}, |
||||
logLevel: LogLevel.info, |
||||
raw: '', |
||||
timeEpochMs: 0, |
||||
timeEpochNs: '0', |
||||
timeFromNow: '', |
||||
timeLocal: '', |
||||
timeUtc: '', |
||||
uid: '1', |
||||
}; |
||||
}; |
||||
|
||||
const mockExploreDataWithLogs = () => |
||||
mockExplorePanelData({ |
||||
logsResult: { |
||||
rows: [mockRow('0'), mockRow('1')], |
||||
visibleRange: { from: 0, to: 1 }, |
||||
bucketSize: 1000, |
||||
}, |
||||
}); |
||||
|
||||
const datasources: DataSourceApi[] = [ |
||||
new MockDataSourceWithSupplementaryQuerySupport('logs-volume-a').withSupplementaryQuerySupport( |
||||
SupplementaryQueryType.LogsVolume, |
||||
createSupplementaryQueryResponse(SupplementaryQueryType.LogsVolume, 'logs-volume-a') |
||||
), |
||||
new MockDataSourceWithSupplementaryQuerySupport('logs-volume-b').withSupplementaryQuerySupport( |
||||
SupplementaryQueryType.LogsVolume, |
||||
createSupplementaryQueryResponse(SupplementaryQueryType.LogsVolume, 'logs-volume-b') |
||||
), |
||||
new MockDataSourceWithSupplementaryQuerySupport('logs-sample-a').withSupplementaryQuerySupport( |
||||
SupplementaryQueryType.LogsSample, |
||||
createSupplementaryQueryResponse(SupplementaryQueryType.LogsSample, 'logs-sample-a') |
||||
), |
||||
new MockDataSourceWithSupplementaryQuerySupport('logs-sample-b').withSupplementaryQuerySupport( |
||||
SupplementaryQueryType.LogsSample, |
||||
createSupplementaryQueryResponse(SupplementaryQueryType.LogsSample, 'logs-sample-b') |
||||
), |
||||
new MockDataSourceApi('no-data-providers'), |
||||
new MockDataSourceApi('no-data-providers-2'), |
||||
new MockDataSourceApi('mixed').setupMixed(true), |
||||
]; |
||||
|
||||
jest.mock('@grafana/runtime', () => ({ |
||||
...jest.requireActual('@grafana/runtime'), |
||||
getDataSourceSrv: () => { |
||||
return { |
||||
get: async ({ uid }: { uid: string }) => datasources.find((ds) => ds.name === uid) || undefined, |
||||
}; |
||||
}, |
||||
})); |
||||
|
||||
const setup = async (rootDataSource: string, type: SupplementaryQueryType, targetSources?: string[]) => { |
||||
const rootDataSourceApiMock = await getDataSourceSrv().get({ uid: rootDataSource }); |
||||
|
||||
targetSources = targetSources || [rootDataSource]; |
||||
|
||||
const requestMock = new MockDataQueryRequest({ |
||||
targets: targetSources.map((source, i) => new MockQuery(`${i}`, 'a', { uid: source })), |
||||
}); |
||||
const explorePanelDataMock: Observable<ExplorePanelData> = mockExploreDataWithLogs(); |
||||
|
||||
return getSupplementaryQueryProvider(rootDataSourceApiMock, type, requestMock, explorePanelDataMock); |
||||
}; |
||||
|
||||
const assertDataFrom = (type: SupplementaryQueryType, ...datasources: string[]) => { |
||||
return flatten( |
||||
datasources.map((name: string) => { |
||||
return [{ refId: `1-${type}-${name}` }, { refId: `2-${type}-${name}` }]; |
||||
}) |
||||
); |
||||
}; |
||||
|
||||
const assertDataFromLogsResults = () => { |
||||
return [{ meta: { custom: { logsVolumeType: LogsVolumeType.Limited } } }]; |
||||
}; |
||||
|
||||
describe('SupplementaryQueries utils', function () { |
||||
describe('Non-mixed data source', function () { |
||||
it('Returns result from the provider', async () => { |
||||
const testProvider = await setup('logs-volume-a', SupplementaryQueryType.LogsVolume); |
||||
|
||||
await expect(testProvider).toEmitValuesWith((received) => { |
||||
expect(received).toMatchObject([ |
||||
{ data: [], state: LoadingState.Loading }, |
||||
{ |
||||
data: assertDataFrom(SupplementaryQueryType.LogsVolume, 'logs-volume-a'), |
||||
state: LoadingState.Done, |
||||
}, |
||||
]); |
||||
}); |
||||
}); |
||||
it('Uses fallback for logs volume', async () => { |
||||
const testProvider = await setup('no-data-providers', SupplementaryQueryType.LogsVolume); |
||||
|
||||
await expect(testProvider).toEmitValuesWith((received) => { |
||||
expect(received).toMatchObject([ |
||||
{ |
||||
data: assertDataFromLogsResults(), |
||||
state: LoadingState.Done, |
||||
}, |
||||
]); |
||||
}); |
||||
}); |
||||
it('Does not use a fallback for logs sample', async () => { |
||||
const testProvider = await setup('no-data-providers', SupplementaryQueryType.LogsSample); |
||||
await expect(testProvider).toEmitValuesWith((received) => { |
||||
expect(received).toMatchObject([ |
||||
{ |
||||
state: LoadingState.NotStarted, |
||||
}, |
||||
]); |
||||
}); |
||||
}); |
||||
}); |
||||
|
||||
describe('Mixed data source', function () { |
||||
describe('Logs volume', function () { |
||||
describe('All data sources support full range logs volume', function () { |
||||
it('Merges all data frames into a single response', async () => { |
||||
const testProvider = await setup('mixed', SupplementaryQueryType.LogsVolume, [ |
||||
'logs-volume-a', |
||||
'logs-volume-b', |
||||
]); |
||||
await expect(testProvider).toEmitValuesWith((received) => { |
||||
expect(received).toMatchObject([ |
||||
{ data: [], state: LoadingState.Loading }, |
||||
{ |
||||
data: assertDataFrom(SupplementaryQueryType.LogsVolume, 'logs-volume-a'), |
||||
state: LoadingState.Done, |
||||
}, |
||||
{ |
||||
data: assertDataFrom(SupplementaryQueryType.LogsVolume, 'logs-volume-a', 'logs-volume-b'), |
||||
state: LoadingState.Done, |
||||
}, |
||||
]); |
||||
}); |
||||
}); |
||||
}); |
||||
|
||||
describe('All data sources do not support full range logs volume', function () { |
||||
it('Creates single fallback result', async () => { |
||||
const testProvider = await setup('mixed', SupplementaryQueryType.LogsVolume, [ |
||||
'no-data-providers', |
||||
'no-data-providers-2', |
||||
]); |
||||
|
||||
await expect(testProvider).toEmitValuesWith((received) => { |
||||
expect(received).toMatchObject([ |
||||
{ |
||||
data: assertDataFromLogsResults(), |
||||
state: LoadingState.Done, |
||||
}, |
||||
{ |
||||
data: [...assertDataFromLogsResults(), ...assertDataFromLogsResults()], |
||||
state: LoadingState.Done, |
||||
}, |
||||
]); |
||||
}); |
||||
}); |
||||
}); |
||||
|
||||
describe('Some data sources support full range logs volume, while others do not', function () { |
||||
it('Creates merged result containing full range and limited logs volume', async () => { |
||||
const testProvider = await setup('mixed', SupplementaryQueryType.LogsVolume, [ |
||||
'logs-volume-a', |
||||
'no-data-providers', |
||||
'logs-volume-b', |
||||
'no-data-providers-2', |
||||
]); |
||||
await expect(testProvider).toEmitValuesWith((received) => { |
||||
expect(received).toMatchObject([ |
||||
{ |
||||
data: [], |
||||
state: LoadingState.Loading, |
||||
}, |
||||
{ |
||||
data: assertDataFrom(SupplementaryQueryType.LogsVolume, 'logs-volume-a'), |
||||
state: LoadingState.Done, |
||||
}, |
||||
{ |
||||
data: [ |
||||
...assertDataFrom(SupplementaryQueryType.LogsVolume, 'logs-volume-a'), |
||||
...assertDataFromLogsResults(), |
||||
], |
||||
state: LoadingState.Done, |
||||
}, |
||||
{ |
||||
data: [ |
||||
...assertDataFrom(SupplementaryQueryType.LogsVolume, 'logs-volume-a'), |
||||
...assertDataFromLogsResults(), |
||||
...assertDataFrom(SupplementaryQueryType.LogsVolume, 'logs-volume-b'), |
||||
], |
||||
state: LoadingState.Done, |
||||
}, |
||||
]); |
||||
}); |
||||
}); |
||||
}); |
||||
}); |
||||
|
||||
describe('Logs sample', function () { |
||||
describe('All data sources support logs sample', function () { |
||||
it('Merges all responses into single result', async () => { |
||||
const testProvider = await setup('mixed', SupplementaryQueryType.LogsSample, [ |
||||
'logs-sample-a', |
||||
'logs-sample-b', |
||||
]); |
||||
await expect(testProvider).toEmitValuesWith((received) => { |
||||
expect(received).toMatchObject([ |
||||
{ data: [], state: LoadingState.Loading }, |
||||
{ |
||||
data: assertDataFrom(SupplementaryQueryType.LogsSample, 'logs-sample-a'), |
||||
state: LoadingState.Done, |
||||
}, |
||||
{ |
||||
data: assertDataFrom(SupplementaryQueryType.LogsSample, 'logs-sample-a', 'logs-sample-b'), |
||||
state: LoadingState.Done, |
||||
}, |
||||
]); |
||||
}); |
||||
}); |
||||
}); |
||||
|
||||
describe('All data sources do not support full range logs volume', function () { |
||||
it('Does not provide fallback result', async () => { |
||||
const testProvider = await setup('mixed', SupplementaryQueryType.LogsSample, [ |
||||
'no-data-providers', |
||||
'no-data-providers-2', |
||||
]); |
||||
await expect(testProvider).toEmitValuesWith((received) => { |
||||
expect(received).toMatchObject([{ state: LoadingState.NotStarted, data: [] }]); |
||||
}); |
||||
}); |
||||
}); |
||||
|
||||
describe('Some data sources support full range logs volume, while others do not', function () { |
||||
it('Returns results only for data sources supporting logs sample', async () => { |
||||
const testProvider = await setup('mixed', SupplementaryQueryType.LogsSample, [ |
||||
'logs-sample-a', |
||||
'no-data-providers', |
||||
'logs-sample-b', |
||||
'no-data-providers-2', |
||||
]); |
||||
await expect(testProvider).toEmitValuesWith((received) => { |
||||
expect(received).toMatchObject([ |
||||
{ data: [], state: LoadingState.Loading }, |
||||
{ |
||||
data: assertDataFrom(SupplementaryQueryType.LogsSample, 'logs-sample-a'), |
||||
state: LoadingState.Done, |
||||
}, |
||||
{ |
||||
data: assertDataFrom(SupplementaryQueryType.LogsSample, 'logs-sample-a', 'logs-sample-b'), |
||||
state: LoadingState.Done, |
||||
}, |
||||
]); |
||||
}); |
||||
}); |
||||
}); |
||||
}); |
||||
}); |
||||
}); |
||||
@ -0,0 +1,30 @@ |
||||
import { CoreApp, DataQueryRequest, getDefaultTimeRange } from '@grafana/data'; |
||||
import { DataQuery, DataSourceRef } from '@grafana/schema'; |
||||
|
||||
export class MockQuery implements DataQuery { |
||||
refId: string; |
||||
testQuery: string; |
||||
datasource?: DataSourceRef; |
||||
|
||||
constructor(refId = 'A', testQuery = '', datasourceRef?: DataSourceRef) { |
||||
this.refId = refId; |
||||
this.testQuery = testQuery; |
||||
this.datasource = datasourceRef; |
||||
} |
||||
} |
||||
|
||||
export class MockDataQueryRequest implements DataQueryRequest<MockQuery> { |
||||
app = CoreApp.Unknown; |
||||
interval = ''; |
||||
intervalMs = 0; |
||||
range = getDefaultTimeRange(); |
||||
requestId = '1'; |
||||
scopedVars = {}; |
||||
startTime = 0; |
||||
targets: MockQuery[]; |
||||
timezone = 'utc'; |
||||
|
||||
constructor({ targets }: { targets: MockQuery[] }) { |
||||
this.targets = targets || []; |
||||
} |
||||
} |
||||
Loading…
Reference in new issue