fix: pivot v2 charts created before GENERIC_CHART_AXES is enabled (#23731)

This commit is contained in:
Beto Dealmeida
2023-04-25 19:44:50 -07:00
committed by GitHub
parent e8121b1cd2
commit 314987f32d
3 changed files with 34 additions and 1 deletions

View File

@@ -17,6 +17,8 @@
* under the License.
*/
import { TimeGranularity } from '@superset-ui/core';
import * as supersetCoreModule from '@superset-ui/core';
import buildQuery from '../../src/plugin/buildQuery';
import { PivotTableQueryFormData } from '../../src/types';
@@ -55,4 +57,29 @@ describe('PivotTableChart buildQuery', () => {
const [query] = queryContext.queries;
expect(query.columns).toEqual(['col1', 'col2', 'row1', 'row2']);
});
it('should work with old charts after GENERIC_CHART_AXES is enabled', () => {
Object.defineProperty(supersetCoreModule, 'hasGenericChartAxes', {
value: true,
});
const modifiedFormData = {
...formData,
time_grain_sqla: TimeGranularity.MONTH,
granularity_sqla: 'col1',
};
const queryContext = buildQuery(modifiedFormData);
const [query] = queryContext.queries;
expect(query.columns).toEqual([
{
timeGrain: 'P1M',
columnType: 'BASE_AXIS',
sqlExpression: 'col1',
label: 'col1',
expressionType: 'SQL',
},
'col2',
'row1',
'row2',
]);
});
});