airbnb/caravel

/**
 * 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 { render, screen, fireEvent } from 'spec/helpers/testing-library';
import { AGGREGATES } from 'src/explore/constants';
import AdhocMetricOption from 'src/explore/components/controls/MetricControl/AdhocMetricOption';
import AdhocMetric from 'src/explore/components/controls/MetricControl/AdhocMetric';
import userEvent from '@testing-library/user-event';

const columns = [
  { type: 'VARCHAR(255)', column_name: 'source' },
  { type: 'VARCHAR(255)', column_name: 'target' },
  { type: 'DOUBLE', column_name: 'value' },
];

const sumValueAdhocMetric = new AdhocMetric({
  column: columns[2],
  aggregate: AGGREGATES.SUM,
});

const datasource = {
  type: 'table',
  id: 1,
  uid: '1__table',
  columnFormats: {},
  verboseMap: {},
};

const defaultProps = {
  adhocMetric: sumValueAdhocMetric,
  savedMetric: {},
  savedMetricsOptions: [],
  onMetricEdit: jest.fn(),
  columns,
  datasource,
  onMoveLabel: jest.fn(),
  onDropLabel: jest.fn(),
  index: 0,
};

function setup(overrides) {
  const props = {
    ...defaultProps,
    ...overrides,
  };
  return render(<AdhocMetricOption {...props} />, { useDnd: true });
}

test('renders an overlay trigger wrapper for the label', () => {
  setup();
  expect(screen.getByText('SUM(value)')).toBeInTheDocument();
});

test('overwrites the adhocMetric in state with onLabelChange', async () => {
  setup();
  userEvent.click(screen.getByText('SUM(value)'));
  userEvent.click(screen.getByTestId(/AdhocMetricEditTitle#trigger/i));
  const labelInput = await screen.findByTestId(/AdhocMetricEditTitle#input/i);
  userEvent.clear(labelInput);
  userEvent.type(labelInput, 'new label');
  expect(labelInput).toHaveValue('new label');
  fireEvent.keyPress(labelInput, {
    key: 'Enter',
    charCode: 13,
  });
  expect(screen.getByText(/new label/i)).toBeInTheDocument();
});

test('returns to default labels when the custom label is cleared', async () => {
  setup();
  userEvent.click(screen.getByText('SUM(value)'));
  userEvent.click(screen.getByTestId(/AdhocMetricEditTitle#trigger/i));
  const labelInput = await screen.findByTestId(/AdhocMetricEditTitle#input/i);
  userEvent.clear(labelInput);
  userEvent.type(labelInput, 'new label');
  fireEvent.keyPress(labelInput, {
    key: 'Enter',
    charCode: 13,
  });
  expect(labelInput).not.toBeInTheDocument();
  expect(screen.getByText(/new label/i)).toBeInTheDocument();
  userEvent.click(screen.getByTestId(/AdhocMetricEditTitle#trigger/i));
  expect(screen.getByPlaceholderText(/new label/i)).toBeInTheDocument();
  userEvent.clear(labelInput);
  fireEvent.keyPress(labelInput, {
    key: 'Enter',
    charCode: 13,
  });
  expect(screen.getByPlaceholderText('SUM(value)')).toBeInTheDocument();
});