| /** |
| * 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. |
| */ |
| /* eslint-disable no-unused-expressions */ |
| import React from 'react'; |
| import sinon from 'sinon'; |
| import { shallow } from 'enzyme'; |
| import { FormGroup } from 'react-bootstrap'; |
| import Button from 'src/components/Button'; |
| |
| import { AGGREGATES } from 'src/explore/constants'; |
| import AdhocMetricEditPopover from 'src/explore/components/controls/MetricControl/AdhocMetricEditPopover'; |
| import AdhocMetric, { |
| EXPRESSION_TYPES, |
| } from 'src/explore/components/controls/MetricControl/AdhocMetric'; |
| |
| const columns = [ |
| { type: 'VARCHAR(255)', column_name: 'source', id: 1 }, |
| { type: 'VARCHAR(255)', column_name: 'target', id: 2 }, |
| { type: 'DOUBLE', column_name: 'value', id: 3 }, |
| ]; |
| |
| const sumValueAdhocMetric = new AdhocMetric({ |
| expressionType: EXPRESSION_TYPES.SIMPLE, |
| column: columns[2], |
| aggregate: AGGREGATES.SUM, |
| }); |
| |
| const sqlExpressionAdhocMetric = new AdhocMetric({ |
| expressionType: EXPRESSION_TYPES.SQL, |
| sqlExpression: 'COUNT(*)', |
| }); |
| |
| function setup(overrides) { |
| const onChange = sinon.spy(); |
| const onClose = sinon.spy(); |
| const props = { |
| adhocMetric: sumValueAdhocMetric, |
| savedMetric: { metric_name: 'foo', expression: 'COUNT(*)' }, |
| savedMetrics: [], |
| onChange, |
| onClose, |
| onResize: () => {}, |
| getCurrentLabel: () => {}, |
| columns, |
| ...overrides, |
| }; |
| const wrapper = shallow(<AdhocMetricEditPopover {...props} />); |
| return { wrapper, onChange, onClose }; |
| } |
| |
| describe('AdhocMetricEditPopover', () => { |
| it('renders a popover with edit metric form contents', () => { |
| const { wrapper } = setup(); |
| expect(wrapper.find(FormGroup)).toHaveLength(4); |
| expect(wrapper.find(Button)).toHaveLength(2); |
| }); |
| |
| it('overwrites the adhocMetric in state with onColumnChange', () => { |
| const { wrapper } = setup(); |
| wrapper.instance().onColumnChange(columns[0].id); |
| expect(wrapper.state('adhocMetric')).toEqual( |
| sumValueAdhocMetric.duplicateWith({ column: columns[0] }), |
| ); |
| }); |
| |
| it('overwrites the adhocMetric in state with onAggregateChange', () => { |
| const { wrapper } = setup(); |
| wrapper.instance().onAggregateChange(AGGREGATES.AVG); |
| expect(wrapper.state('adhocMetric')).toEqual( |
| sumValueAdhocMetric.duplicateWith({ aggregate: AGGREGATES.AVG }), |
| ); |
| }); |
| |
| it('overwrites the adhocMetric in state with onSqlExpressionChange', () => { |
| const { wrapper } = setup({ adhocMetric: sqlExpressionAdhocMetric }); |
| wrapper.instance().onSqlExpressionChange('COUNT(1)'); |
| expect(wrapper.state('adhocMetric')).toEqual( |
| sqlExpressionAdhocMetric.duplicateWith({ sqlExpression: 'COUNT(1)' }), |
| ); |
| }); |
| |
| it('prevents saving if no column or aggregate is chosen', () => { |
| const { wrapper } = setup(); |
| expect(wrapper.find(Button).find({ disabled: true })).not.toExist(); |
| wrapper.instance().onColumnChange(null); |
| expect(wrapper.find(Button).find({ disabled: true })).toExist(); |
| wrapper.instance().onColumnChange(columns[0].id); |
| expect(wrapper.find(Button).find({ disabled: true })).not.toExist(); |
| wrapper.instance().onAggregateChange(null); |
| expect(wrapper.find(Button).find({ disabled: true })).toExist(); |
| }); |
| |
| it('highlights save if changes are present', () => { |
| const { wrapper } = setup(); |
| expect(wrapper.find(Button).find({ buttonStyle: 'primary' })).not.toExist(); |
| wrapper.instance().onColumnChange(columns[1].id); |
| expect(wrapper.find(Button).find({ buttonStyle: 'primary' })).toExist(); |
| }); |
| |
| it('will initiate a drag when clicked', () => { |
| const { wrapper } = setup(); |
| wrapper.instance().onDragDown = sinon.spy(); |
| wrapper.instance().forceUpdate(); |
| |
| expect(wrapper.find('.fa-expand')).toExist(); |
| expect(wrapper.instance().onDragDown.calledOnce).toBe(false); |
| wrapper.find('.fa-expand').simulate('mouseDown'); |
| expect(wrapper.instance().onDragDown.calledOnce).toBe(true); |
| }); |
| }); |