mirror of
https://github.com/apache/superset.git
synced 2024-09-19 20:19:37 -04:00
2913063924
* move assets out, get webpack dev working * update docs to reference superset-frontend * draw the rest of the owl * fix docs * fix webpack script * rats * correct docs * fix tox dox
265 lines
7.8 KiB
JavaScript
265 lines
7.8 KiB
JavaScript
/**
|
|
* 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 React from 'react';
|
|
import configureStore from 'redux-mock-store';
|
|
import thunk from 'redux-thunk';
|
|
|
|
import { shallow } from 'enzyme';
|
|
import sinon from 'sinon';
|
|
import fetchMock from 'fetch-mock';
|
|
|
|
import shortid from 'shortid';
|
|
import { queries, queryWithBadColumns } from './fixtures';
|
|
import sqlLabReducer from '../../../src/SqlLab/reducers/index';
|
|
import * as actions from '../../../src/SqlLab/actions/sqlLab';
|
|
import ExploreResultsButton from '../../../src/SqlLab/components/ExploreResultsButton';
|
|
import * as exploreUtils from '../../../src/explore/exploreUtils';
|
|
import Button from '../../../src/components/Button';
|
|
|
|
describe('ExploreResultsButton', () => {
|
|
const middlewares = [thunk];
|
|
const mockStore = configureStore(middlewares);
|
|
const database = {
|
|
allows_subquery: true,
|
|
};
|
|
const initialState = {
|
|
sqlLab: {
|
|
...sqlLabReducer(undefined, {}),
|
|
},
|
|
common: {
|
|
conf: { SUPERSET_WEBSERVER_TIMEOUT: 45 },
|
|
},
|
|
};
|
|
const store = mockStore(initialState);
|
|
const mockedProps = {
|
|
database,
|
|
show: true,
|
|
query: queries[0],
|
|
};
|
|
const mockColumns = {
|
|
ds: {
|
|
is_date: true,
|
|
name: 'ds',
|
|
type: 'STRING',
|
|
},
|
|
gender: {
|
|
is_date: false,
|
|
name: 'gender',
|
|
type: 'STRING',
|
|
},
|
|
};
|
|
const mockChartTypeBarChart = {
|
|
label: 'Distribution - Bar Chart',
|
|
requiresTime: false,
|
|
value: 'dist_bar',
|
|
};
|
|
const mockChartTypeTB = {
|
|
label: 'Time Series - Bar Chart',
|
|
requiresTime: true,
|
|
value: 'bar',
|
|
};
|
|
const getExploreResultsButtonWrapper = (props = mockedProps) =>
|
|
shallow(<ExploreResultsButton {...props} />, {
|
|
context: { store },
|
|
}).dive();
|
|
|
|
it('renders', () => {
|
|
expect(React.isValidElement(<ExploreResultsButton />)).toBe(true);
|
|
});
|
|
|
|
it('renders with props', () => {
|
|
expect(
|
|
React.isValidElement(<ExploreResultsButton {...mockedProps} />),
|
|
).toBe(true);
|
|
});
|
|
|
|
it('detects bad columns', () => {
|
|
const wrapper = getExploreResultsButtonWrapper({
|
|
database,
|
|
show: true,
|
|
query: queryWithBadColumns,
|
|
});
|
|
|
|
const badCols = wrapper.instance().getInvalidColumns();
|
|
expect(badCols).toEqual([
|
|
'COUNT(*)',
|
|
'1',
|
|
'123',
|
|
'CASE WHEN 1=1 THEN 1 ELSE 0 END',
|
|
]);
|
|
|
|
const msgWrapper = shallow(wrapper.instance().renderInvalidColumnMessage());
|
|
expect(msgWrapper.find('div')).toHaveLength(1);
|
|
});
|
|
|
|
it('renders a Button', () => {
|
|
const wrapper = getExploreResultsButtonWrapper();
|
|
expect(wrapper.find(Button)).toHaveLength(1);
|
|
});
|
|
|
|
describe('datasourceName', () => {
|
|
let wrapper;
|
|
let stub;
|
|
beforeEach(() => {
|
|
wrapper = getExploreResultsButtonWrapper();
|
|
stub = sinon.stub(shortid, 'generate').returns('abcd');
|
|
});
|
|
afterEach(() => {
|
|
stub.restore();
|
|
});
|
|
|
|
it('should generate data source name from query', () => {
|
|
const sampleQuery = queries[0];
|
|
const name = wrapper.instance().datasourceName();
|
|
expect(name).toBe(`${sampleQuery.user}-${sampleQuery.tab}-abcd`);
|
|
});
|
|
it('should generate data source name with empty query', () => {
|
|
wrapper.setProps({ query: {} });
|
|
const name = wrapper.instance().datasourceName();
|
|
expect(name).toBe('undefined-abcd');
|
|
});
|
|
|
|
it('should build viz options', () => {
|
|
wrapper.setState({ chartType: mockChartTypeTB });
|
|
const spy = sinon.spy(wrapper.instance(), 'buildVizOptions');
|
|
wrapper.instance().buildVizOptions();
|
|
expect(spy.returnValues[0]).toEqual({
|
|
schema: 'test_schema',
|
|
sql: wrapper.instance().props.query.sql,
|
|
dbId: wrapper.instance().props.query.dbId,
|
|
columns: Object.values(mockColumns),
|
|
templateParams: undefined,
|
|
datasourceName: 'admin-Demo-abcd',
|
|
});
|
|
});
|
|
});
|
|
|
|
it('should build visualize advise for long query', () => {
|
|
const longQuery = { ...queries[0], endDttm: 1476910666798 };
|
|
const props = {
|
|
show: true,
|
|
query: longQuery,
|
|
database,
|
|
};
|
|
const longQueryWrapper = shallow(<ExploreResultsButton {...props} />, {
|
|
context: { store },
|
|
}).dive();
|
|
const inst = longQueryWrapper.instance();
|
|
expect(inst.getQueryDuration()).toBe(100.7050400390625);
|
|
});
|
|
|
|
describe('visualize', () => {
|
|
const wrapper = getExploreResultsButtonWrapper();
|
|
const mockOptions = { attr: 'mockOptions' };
|
|
wrapper.setState({
|
|
chartType: mockChartTypeBarChart,
|
|
datasourceName: 'mockDatasourceName',
|
|
});
|
|
|
|
const visualizeURL = '/superset/sqllab_viz/';
|
|
const visualizeEndpoint = `glob:*${visualizeURL}`;
|
|
const visualizationPayload = { table_id: 107 };
|
|
fetchMock.post(visualizeEndpoint, visualizationPayload);
|
|
|
|
beforeEach(() => {
|
|
sinon.stub(exploreUtils, 'getExploreUrlAndPayload').callsFake(() => ({
|
|
url: 'mockURL',
|
|
payload: { datasource: '107__table' },
|
|
}));
|
|
sinon.spy(exploreUtils, 'exportChart');
|
|
sinon
|
|
.stub(wrapper.instance(), 'buildVizOptions')
|
|
.callsFake(() => mockOptions);
|
|
});
|
|
afterEach(() => {
|
|
exploreUtils.getExploreUrlAndPayload.restore();
|
|
exploreUtils.exportChart.restore();
|
|
wrapper.instance().buildVizOptions.restore();
|
|
fetchMock.reset();
|
|
});
|
|
|
|
it('should build request with correct args', done => {
|
|
wrapper.instance().visualize();
|
|
|
|
setTimeout(() => {
|
|
const calls = fetchMock.calls(visualizeEndpoint);
|
|
expect(calls).toHaveLength(1);
|
|
const formData = calls[0][1].body;
|
|
|
|
Object.keys(mockOptions).forEach(key => {
|
|
// eslint-disable-next-line no-unused-expressions
|
|
expect(formData.get(key)).toBeDefined();
|
|
});
|
|
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should export chart and add an info toast', done => {
|
|
const infoToastSpy = sinon.spy();
|
|
const datasourceSpy = sinon.stub();
|
|
|
|
datasourceSpy.callsFake(() => Promise.resolve(visualizationPayload));
|
|
|
|
wrapper.setProps({
|
|
actions: {
|
|
addInfoToast: infoToastSpy,
|
|
createDatasource: datasourceSpy,
|
|
},
|
|
});
|
|
|
|
wrapper.instance().visualize();
|
|
|
|
setTimeout(() => {
|
|
expect(datasourceSpy.callCount).toBe(1);
|
|
expect(exploreUtils.exportChart.callCount).toBe(1);
|
|
expect(exploreUtils.exportChart.getCall(0).args[0].datasource).toBe(
|
|
'107__table',
|
|
);
|
|
expect(infoToastSpy.callCount).toBe(1);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should add error toast', done => {
|
|
const dangerToastSpy = sinon.stub(actions, 'addDangerToast');
|
|
const datasourceSpy = sinon.stub();
|
|
|
|
datasourceSpy.callsFake(() => Promise.reject({ error: 'error' }));
|
|
|
|
wrapper.setProps({
|
|
actions: {
|
|
createDatasource: datasourceSpy,
|
|
addDangerToast: dangerToastSpy,
|
|
},
|
|
});
|
|
|
|
wrapper.instance().visualize();
|
|
|
|
setTimeout(() => {
|
|
expect(datasourceSpy.callCount).toBe(1);
|
|
expect(exploreUtils.exportChart.callCount).toBe(0);
|
|
expect(dangerToastSpy.callCount).toBe(1);
|
|
dangerToastSpy.restore();
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
});
|