mirror of
https://github.com/apache/superset.git
synced 2026-04-08 19:05:46 +00:00
fca8a49561
Co-authored-by: Richard <richard@ip-192-168-1-32.sa-east-1.compute.internal> Co-authored-by: richard <richard@richards-MacBook-Pro-2.local> Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com> Co-authored-by: Evan Rusackas <evan@preset.io>
98 lines
3.1 KiB
TypeScript
98 lines
3.1 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 buildQuery from '../../src/Graph/buildQuery';
|
|
import { DEFAULT_FORM_DATA } from '../../src/Graph/types';
|
|
|
|
describe('Graph buildQuery', () => {
|
|
const formData = {
|
|
...DEFAULT_FORM_DATA,
|
|
datasource: '5__table',
|
|
granularity_sqla: 'ds',
|
|
source: 'dummy_source',
|
|
target: 'dummy_target',
|
|
metrics: ['foo', 'bar'],
|
|
viz_type: 'my_chart',
|
|
};
|
|
|
|
test('should build groupby with source and target categories', () => {
|
|
const formDataWithCategories = {
|
|
...formData,
|
|
source: 'dummy_source',
|
|
target: 'dummy_target',
|
|
source_category: 'dummy_source_category',
|
|
target_category: 'dummy_target_category',
|
|
};
|
|
const queryContext = buildQuery(formDataWithCategories);
|
|
const [query] = queryContext.queries;
|
|
expect(query.columns).toEqual([
|
|
'dummy_source',
|
|
'dummy_target',
|
|
'dummy_source_category',
|
|
'dummy_target_category',
|
|
]);
|
|
expect(query.metrics).toEqual(['foo', 'bar']);
|
|
});
|
|
|
|
test('should build groupby with source category', () => {
|
|
const formDataWithCategories = {
|
|
...formData,
|
|
source: 'dummy_source',
|
|
target: 'dummy_target',
|
|
source_category: 'dummy_source_category',
|
|
};
|
|
const queryContext = buildQuery(formDataWithCategories);
|
|
const [query] = queryContext.queries;
|
|
expect(query.columns).toEqual([
|
|
'dummy_source',
|
|
'dummy_target',
|
|
'dummy_source_category',
|
|
]);
|
|
expect(query.metrics).toEqual(['foo', 'bar']);
|
|
});
|
|
|
|
test('should build groupby with target category', () => {
|
|
const formDataWithCategories = {
|
|
...formData,
|
|
source: 'dummy_source',
|
|
target: 'dummy_target',
|
|
target_category: 'dummy_target_category',
|
|
};
|
|
const queryContext = buildQuery(formDataWithCategories);
|
|
const [query] = queryContext.queries;
|
|
expect(query.columns).toEqual([
|
|
'dummy_source',
|
|
'dummy_target',
|
|
'dummy_target_category',
|
|
]);
|
|
expect(query.metrics).toEqual(['foo', 'bar']);
|
|
});
|
|
|
|
test('should build groupby without any category', () => {
|
|
const formDataWithCategories = {
|
|
...formData,
|
|
source: 'dummy_source',
|
|
target: 'dummy_target',
|
|
};
|
|
const queryContext = buildQuery(formDataWithCategories);
|
|
const [query] = queryContext.queries;
|
|
expect(query.columns).toEqual(['dummy_source', 'dummy_target']);
|
|
expect(query.metrics).toEqual(['foo', 'bar']);
|
|
});
|
|
});
|