Files
superset2/superset-frontend/plugins/plugin-chart-rose/test/transformData.test.ts
2026-07-22 17:50:35 -07:00

99 lines
3.0 KiB
TypeScript

/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
import transformData, {
seriesKeyOf,
splitFlatLabel,
} from '../src/transformData';
const t1 = 1704067200000;
const t2 = 1704153600000;
test('re-keys flattened pivot records by timestamp with NaN as 0', () => {
const data = transformData(
[
{ __timestamp: t1, 'sum__num, boy': 10, 'sum__num, girl': 20 },
{ __timestamp: t2, 'sum__num, boy': 30, 'sum__num, girl': null },
],
{ metricLabels: ['sum__num'], timeCompare: [] },
);
expect(data[String(t1)]).toEqual([
{ key: ['boy'], value: 10, name: ['boy'], time: t1 },
{ key: ['girl'], value: 20, name: ['girl'], time: t1 },
]);
expect(data[String(t2)][1]).toEqual({
key: ['girl'],
value: 0,
name: ['girl'],
time: t2,
});
});
test('keys ungrouped series by the metric name', () => {
const data = transformData([{ __timestamp: t1, sum__num: 5 }], {
metricLabels: ['sum__num'],
timeCompare: [],
});
expect(data[String(t1)]).toEqual([
{ key: 'sum__num', value: 5, name: 'sum__num', time: t1 },
]);
});
test('splits flattened labels honoring escaped commas', () => {
expect(splitFlatLabel('sum__num, CA\\, USA')).toEqual([
'sum__num',
'CA, USA',
]);
});
test('labels time-shifted columns with the legacy offset suffix', () => {
expect(
seriesKeyOf(['sum__num__1 week ago', 'boy'], {
metricLabels: ['sum__num'],
timeCompare: ['1 week ago'],
}),
).toEqual(['boy', '1 week ago offset']);
expect(
seriesKeyOf(['sum__num__1 week ago'], {
metricLabels: ['sum__num'],
timeCompare: ['1 week ago'],
}),
).toEqual(['sum__num', '1 week ago offset']);
});
test('maps comparison columns and drops originals', () => {
const options = {
metricLabels: ['sum__num'],
timeCompare: ['1 week ago'],
comparisonType: 'difference',
};
expect(
seriesKeyOf(['difference__sum__num__sum__num__1 week ago', 'boy'], options),
).toEqual(['boy', '1 week ago offset']);
expect(seriesKeyOf(['sum__num', 'boy'], options)).toBeNull();
});
test('keeps the metric in multi-metric grouped keys', () => {
expect(
seriesKeyOf(['count', 'boy'], {
metricLabels: ['count', 'sum__num'],
timeCompare: [],
}),
).toEqual(['count', 'boy']);
});