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';
|
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
|
|
|
});
|
2020-05-06 12:32:29 -04:00
|
|
|
it('should build datasource for druid sources and set force to true', () => {
|
2018-12-12 16:18:27 -05:00
|
|
|
const queryContext = buildQueryContext({
|
|
|
|
datasource: '5__druid',
|
|
|
|
granularity: 'ds',
|
|
|
|
viz_type: 'table',
|
2020-05-06 12:32:29 -04:00
|
|
|
force: true,
|
2018-12-12 16:18:27 -05:00
|
|
|
});
|
2018-12-04 16:21:49 -05:00
|
|
|
expect(queryContext.datasource.id).toBe(5);
|
|
|
|
expect(queryContext.datasource.type).toBe('druid');
|
2020-05-06 12:32:29 -04:00
|
|
|
expect(queryContext.force).toBe(true);
|
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'],
|
|
|
|
}),
|
|
|
|
]),
|
|
|
|
);
|
|
|
|
});
|
2018-12-04 16:21:49 -05:00
|
|
|
});
|