2020-12-02 01:20:00 -05:00
|
|
|
/**
|
|
|
|
* 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.
|
|
|
|
*/
|
2022-01-07 20:16:24 -05:00
|
|
|
import { buildQueryContext } from '@superset-ui/core';
|
2022-09-07 04:24:15 -04:00
|
|
|
import * as queryModule from '../../src/query/normalizeTimeColumn';
|
2022-11-01 20:21:17 -04:00
|
|
|
import * as getXAxisModule from '../../src/query/getXAxis';
|
2018-12-04 16:21:49 -05:00
|
|
|
|
2020-04-27 15:47:32 -04:00
|
|
|
describe('buildQueryContext', () => {
|
2020-06-05 14:57:47 -04:00
|
|
|
it('should build datasource for table sources and apply defaults', () => {
|
2018-12-12 16:18:27 -05:00
|
|
|
const queryContext = buildQueryContext({
|
|
|
|
datasource: '5__table',
|
|
|
|
granularity_sqla: 'ds',
|
|
|
|
viz_type: 'table',
|
|
|
|
});
|
2018-12-04 16:21:49 -05:00
|
|
|
expect(queryContext.datasource.id).toBe(5);
|
|
|
|
expect(queryContext.datasource.type).toBe('table');
|
2020-05-06 12:32:29 -04:00
|
|
|
expect(queryContext.force).toBe(false);
|
2020-06-05 14:57:47 -04:00
|
|
|
expect(queryContext.result_format).toBe('json');
|
|
|
|
expect(queryContext.result_type).toBe('full');
|
2018-12-04 16:21:49 -05:00
|
|
|
});
|
2021-02-19 14:06:36 -05:00
|
|
|
it('should build datasource for table sources with columns', () => {
|
|
|
|
const queryContext = buildQueryContext(
|
|
|
|
{
|
|
|
|
datasource: '5__table',
|
|
|
|
granularity_sqla: 'ds',
|
|
|
|
viz_type: 'table',
|
|
|
|
source: 'source_column',
|
|
|
|
source_category: 'source_category_column',
|
|
|
|
target: 'target_column',
|
|
|
|
target_category: 'target_category_column',
|
|
|
|
},
|
|
|
|
{
|
|
|
|
queryFields: {
|
|
|
|
source: 'columns',
|
|
|
|
source_category: 'columns',
|
|
|
|
target: 'columns',
|
|
|
|
target_category: 'columns',
|
|
|
|
},
|
|
|
|
},
|
|
|
|
);
|
|
|
|
expect(queryContext.datasource.id).toBe(5);
|
|
|
|
expect(queryContext.datasource.type).toBe('table');
|
|
|
|
expect(queryContext.force).toBe(false);
|
|
|
|
expect(queryContext.result_format).toBe('json');
|
|
|
|
expect(queryContext.result_type).toBe('full');
|
|
|
|
expect(queryContext.queries).toEqual(
|
|
|
|
expect.arrayContaining([
|
|
|
|
expect.objectContaining({
|
|
|
|
columns: [
|
|
|
|
'source_column',
|
|
|
|
'source_category_column',
|
|
|
|
'target_column',
|
|
|
|
'target_category_column',
|
|
|
|
],
|
|
|
|
}),
|
|
|
|
]),
|
|
|
|
);
|
|
|
|
});
|
|
|
|
it('should build datasource for table sources and process with custom function', () => {
|
|
|
|
const queryContext = buildQueryContext(
|
|
|
|
{
|
|
|
|
datasource: '5__table',
|
|
|
|
granularity_sqla: 'ds',
|
|
|
|
viz_type: 'table',
|
|
|
|
source: 'source_column',
|
|
|
|
source_category: 'source_category_column',
|
|
|
|
target: 'target_column',
|
|
|
|
target_category: 'target_category_column',
|
|
|
|
},
|
|
|
|
function addExtraColumn(queryObject) {
|
|
|
|
return [{ ...queryObject, columns: ['dummy_column'] }];
|
|
|
|
},
|
|
|
|
);
|
|
|
|
expect(queryContext.datasource.id).toBe(5);
|
|
|
|
expect(queryContext.datasource.type).toBe('table');
|
|
|
|
expect(queryContext.force).toBe(false);
|
|
|
|
expect(queryContext.result_format).toBe('json');
|
|
|
|
expect(queryContext.result_type).toBe('full');
|
|
|
|
expect(queryContext.queries).toEqual(
|
|
|
|
expect.arrayContaining([
|
|
|
|
expect.objectContaining({
|
|
|
|
columns: ['dummy_column'],
|
|
|
|
}),
|
|
|
|
]),
|
|
|
|
);
|
|
|
|
});
|
2022-11-01 20:21:17 -04:00
|
|
|
// todo(Yongjie): move these test case into buildQueryObject.test.ts
|
2022-06-08 04:54:30 -04:00
|
|
|
it('should remove undefined value in post_processing', () => {
|
|
|
|
const queryContext = buildQueryContext(
|
|
|
|
{
|
|
|
|
datasource: '5__table',
|
|
|
|
viz_type: 'table',
|
|
|
|
},
|
|
|
|
() => [
|
|
|
|
{
|
|
|
|
post_processing: [
|
|
|
|
undefined,
|
|
|
|
undefined,
|
|
|
|
{
|
|
|
|
operation: 'flatten',
|
|
|
|
},
|
|
|
|
undefined,
|
|
|
|
],
|
|
|
|
},
|
|
|
|
],
|
|
|
|
);
|
|
|
|
expect(queryContext.queries[0].post_processing).toEqual([
|
|
|
|
{
|
|
|
|
operation: 'flatten',
|
|
|
|
},
|
|
|
|
]);
|
|
|
|
});
|
2022-09-20 21:41:21 -04:00
|
|
|
it('should call normalizeTimeColumn if GENERIC_CHART_AXES is enabled and has x_axis', () => {
|
2022-11-01 20:21:17 -04:00
|
|
|
Object.defineProperty(getXAxisModule, 'hasGenericChartAxes', {
|
|
|
|
value: true,
|
|
|
|
});
|
2022-09-07 04:24:15 -04:00
|
|
|
const spyNormalizeTimeColumn = jest.spyOn(
|
|
|
|
queryModule,
|
|
|
|
'normalizeTimeColumn',
|
|
|
|
);
|
|
|
|
|
|
|
|
buildQueryContext(
|
|
|
|
{
|
|
|
|
datasource: '5__table',
|
|
|
|
viz_type: 'table',
|
2022-09-20 21:41:21 -04:00
|
|
|
x_axis: 'axis',
|
2022-09-07 04:24:15 -04:00
|
|
|
},
|
|
|
|
() => [{}],
|
|
|
|
);
|
|
|
|
expect(spyNormalizeTimeColumn).toBeCalled();
|
|
|
|
spyNormalizeTimeColumn.mockRestore();
|
|
|
|
});
|
|
|
|
it("shouldn't call normalizeTimeColumn if GENERIC_CHART_AXES is disabled", () => {
|
2022-11-01 20:21:17 -04:00
|
|
|
Object.defineProperty(getXAxisModule, 'hasGenericChartAxes', {
|
|
|
|
value: false,
|
|
|
|
});
|
2022-09-07 04:24:15 -04:00
|
|
|
const spyNormalizeTimeColumn = jest.spyOn(
|
|
|
|
queryModule,
|
|
|
|
'normalizeTimeColumn',
|
|
|
|
);
|
|
|
|
|
|
|
|
buildQueryContext(
|
|
|
|
{
|
|
|
|
datasource: '5__table',
|
|
|
|
viz_type: 'table',
|
|
|
|
},
|
|
|
|
() => [{}],
|
|
|
|
);
|
|
|
|
expect(spyNormalizeTimeColumn).not.toBeCalled();
|
|
|
|
spyNormalizeTimeColumn.mockRestore();
|
|
|
|
});
|
2018-12-04 16:21:49 -05:00
|
|
|
});
|