Files
superset2/superset-frontend/plugins/plugin-chart-partition/test/transformProps.test.ts
T

79 lines
2.7 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 { supersetTheme } from '@apache-superset/core/theme';
import { ChartProps } from '@superset-ui/core';
import transformProps from '../src/transformProps';
const baseFormData = {
colorScheme: 'supersetColors',
groupby: ['gender', 'state'],
metrics: ['sum__num'],
timeSeriesOption: 'not_time',
numberFormat: 'SMART_NUMBER',
sliceId: 1,
};
function buildChartProps(overrides: Record<string, unknown> = {}) {
return new ChartProps({
width: 400,
height: 300,
theme: supersetTheme,
formData: { ...baseFormData, ...overrides },
datasource: { verboseMap: { gender: 'Gender', state: 'State' } },
queriesData: [
{
data: [
{ gender: 'boy', state: 'CA', sum__num: 10 },
{ gender: 'girl', state: 'NY', sum__num: 20 },
],
},
],
}) as ChartProps;
}
test('builds the hierarchy from v1 flat records and maps levels through verboseMap', () => {
const result = transformProps(buildChartProps());
expect(result.width).toEqual(400);
expect(result.height).toEqual(300);
expect(result.levels).toEqual(['Gender', 'State']);
expect(result.data).toHaveLength(1);
expect(result.data[0].val).toEqual(30);
});
test('falls back to the raw column name when verboseMap has no entry', () => {
const result = transformProps(
buildChartProps({ groupby: ['gender', 'unmapped_col'] }),
);
expect(result.levels).toEqual(['Gender', 'unmapped_col']);
});
test('parses partitionLimit and partitionThreshold as integers', () => {
const result = transformProps(
buildChartProps({ partitionLimit: '5', partitionThreshold: '0.1' }),
);
expect(result.partitionLimit).toEqual(5);
expect(result.partitionThreshold).toEqual(0);
});
test('leaves partitionLimit and partitionThreshold undefined when unset', () => {
const result = transformProps(buildChartProps());
expect(result.partitionLimit).toBeFalsy();
expect(result.partitionThreshold).toBeFalsy();
});