superset/superset-frontend/cypress-base/cypress/e2e/explore/visualizations/treemap.test.js
2023-04-26 16:17:59 +02:00

97 lines
3.1 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.
*/
describe('Visualization > Treemap', () => {
beforeEach(() => {
cy.intercept('POST', '/superset/explore_json/**').as('getJson');
});
const TREEMAP_FORM_DATA = {
datasource: '2__table',
viz_type: 'treemap',
slice_id: 10,
granularity_sqla: 'year',
time_grain_sqla: 'P1D',
time_range: 'No filter',
metrics: ['sum__SP_POP_TOTL'],
adhoc_filters: [],
groupby: ['country_code'],
row_limit: 50000,
color_scheme: 'bnbColors',
treemap_ratio: 1.618033988749895,
number_format: '.3s',
};
const level0 = '.chart-container rect[style="fill: rgb(255, 90, 95);"]';
const level1 = '.chart-container rect[style="fill: rgb(123, 0, 81);"]';
const level2 = '.chart-container rect[style="fill: rgb(0, 122, 135);"]';
function verify(formData) {
cy.visitChartByParams(formData);
cy.verifySliceSuccess({ waitAlias: '@getJson', chartSelector: 'svg' });
}
it('should work', () => {
verify(TREEMAP_FORM_DATA);
cy.get(level0).should('have.length', 1);
cy.get(level1).should('have.length', 214);
});
it('should work with multiple groupby', () => {
verify({
...TREEMAP_FORM_DATA,
groupby: ['region', 'country_code'],
});
cy.get(level0).should('have.length', 1);
cy.get(level1).should('have.length', 7);
cy.get(level2).should('have.length', 214);
});
it('should work with filter', () => {
verify({
...TREEMAP_FORM_DATA,
adhoc_filters: [
{
expressionType: 'SIMPLE',
subject: 'region',
operator: '==',
comparator: 'South Asia',
clause: 'WHERE',
sqlExpression: null,
filterOptionName: 'filter_8aqxcf5co1a_x7lm2d1fq0l',
},
],
});
cy.get(level1).should('have.length', 8);
});
it('should allow type to search color schemes and apply the scheme', () => {
verify(TREEMAP_FORM_DATA);
cy.get('.Control[data-test="color_scheme"]').scrollIntoView();
cy.get('.Control[data-test="color_scheme"] input[type="search"]')
.focus()
.type('supersetColors{enter}');
cy.get(
'.Control[data-test="color_scheme"] .ant-select-selection-item [data-test="supersetColors"]',
).should('exist');
cy.get('[data-test=run-query-button]').click();
cy.get('#rect-IND').should('have.css', 'fill', 'rgb(69, 78, 124)');
});
});