test: Adds tests to the AnnotationLayer component

superset-frontend/src/explore/components/controls/AnnotationLayer.jsx → superset-frontend/src/explore/components/controls/AnnotationLayerControl/AnnotationLayer.jsx

@@ -29,22 +29,19 @@ import {
   validateNonEmpty,
 } from '@superset-ui/core';
 
-import SelectControl from './SelectControl';
-import TextControl from './TextControl';
-import CheckboxControl from './CheckboxControl';
-
+import SelectControl from 'src/explore/components/controls/SelectControl';
+import TextControl from 'src/explore/components/controls/TextControl';
+import CheckboxControl from 'src/explore/components/controls/CheckboxControl';
 import {
   ANNOTATION_SOURCE_TYPES,
   ANNOTATION_TYPES,
   ANNOTATION_TYPES_METADATA,
   DEFAULT_ANNOTATION_TYPE,
   requiresQuery,
   ANNOTATION_SOURCE_TYPES_METADATA,
-} from '../../../modules/AnnotationTypes';
-
-import PopoverSection from '../../../components/PopoverSection';
-import ControlHeader from '../ControlHeader';
-import './AnnotationLayer.less';
+} from 'src/modules/AnnotationTypes';
+import PopoverSection from 'src/components/PopoverSection';
+import ControlHeader from 'src/explore/components/ControlHeader';
 
 const AUTOMATIC_COLOR = '';
 
@@ -359,7 +356,14 @@ export default class AnnotationLayer extends React.PureComponent {
 
   renderOption(option) {
     return (
-      <span className="optionWrapper" title={option.label}>
+      <span
+        css={{
+          overflow: 'hidden',
+          textOverflow: 'ellipsis',
+          whiteSpace: 'nowrap',
+        }}
+        title={option.label}
+      >
         {option.label}
       </span>
     );

superset-frontend/src/explore/components/controls/AnnotationLayerControl/AnnotationLayer.test.tsx

@@ -0,0 +1,190 @@
+/**
+ * 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 { render, screen, waitFor } from 'spec/helpers/testing-library';
+import userEvent from '@testing-library/user-event';
+import { getChartMetadataRegistry, ChartMetadata } from '@superset-ui/core';
+import fetchMock from 'fetch-mock';
+import setupColors from 'src/setup/setupColors';
+import { ANNOTATION_TYPES_METADATA } from 'src/modules/AnnotationTypes';
+import AnnotationLayer from './AnnotationLayer';
+
+const defaultProps = {
+  value: '',
+  vizType: 'table',
+  annotationType: ANNOTATION_TYPES_METADATA.FORMULA.value,
+};
+
+beforeAll(() => {
+  const supportedAnnotationTypes = Object.values(ANNOTATION_TYPES_METADATA).map(
+    value => value.value,
+  );
+
+  fetchMock.get('glob:*/annotationlayermodelview/api/read?*', {
+    result: [{ label: 'Chart A', value: 'a' }],
+  });
+
+  fetchMock.get('glob:*/superset/user_slices*', [
+    { id: 'a', title: 'Chart A', viz_type: 'table', data: {} },
+  ]);
+
+  setupColors();
+
+  getChartMetadataRegistry().registerValue(
+    'table',
+    new ChartMetadata({
+      name: 'Table',
+      thumbnail: '',
+      supportedAnnotationTypes,
+      canBeAnnotationTypes: ['EVENT'],
+    }),
+  );
+});
+
+test('renders with default props', () => {
+  const { container } = render(<AnnotationLayer {...defaultProps} />);
+  expect(container).toBeInTheDocument();
+  expect(screen.getByRole('button', { name: 'Apply' })).toBeDisabled();
+  expect(screen.getByRole('button', { name: 'OK' })).toBeDisabled();
+  expect(screen.getByRole('button', { name: 'Cancel' })).toBeEnabled();
+});
+
+test('renders extra checkboxes when type is time series', () => {
+  render(<AnnotationLayer {...defaultProps} />);
+  expect(
+    screen.queryByRole('button', { name: 'Show Markers' }),
+  ).not.toBeInTheDocument();
+  expect(
+    screen.queryByRole('button', { name: 'Hide Line' }),
+  ).not.toBeInTheDocument();
+  userEvent.click(screen.getAllByText('Formula')[0]);
+  userEvent.click(screen.getByText('Time series'));
+  expect(
+    screen.getByRole('button', { name: 'Show Markers' }),
+  ).toBeInTheDocument();
+  expect(screen.getByRole('button', { name: 'Hide Line' })).toBeInTheDocument();
+});
+
+test('enables apply and ok buttons', async () => {
+  render(<AnnotationLayer {...defaultProps} />);
+  userEvent.type(screen.getByLabelText('Name'), 'Test');
+  userEvent.type(screen.getByLabelText('Formula'), '2x');
+  await waitFor(() => {
+    expect(screen.getByRole('button', { name: 'Apply' })).toBeEnabled();
+    expect(screen.getByRole('button', { name: 'OK' })).toBeEnabled();
+  });
+});
+
+test('triggers addAnnotationLayer when apply button is clicked', () => {
+  const addAnnotationLayer = jest.fn();
+  render(
+    <AnnotationLayer
+      {...defaultProps}
+      name="Test"
+      value="2x"
+      addAnnotationLayer={addAnnotationLayer}
+    />,
+  );
+  userEvent.click(screen.getByRole('button', { name: 'Apply' }));
+  expect(addAnnotationLayer).toHaveBeenCalled();
+});
+
+test('triggers addAnnotationLayer and close when ok button is clicked', () => {
+  const addAnnotationLayer = jest.fn();
+  const close = jest.fn();
+  render(
+    <AnnotationLayer
+      {...defaultProps}
+      name="Test"
+      value="2x"
+      addAnnotationLayer={addAnnotationLayer}
+      close={close}
+    />,
+  );
+  userEvent.click(screen.getByRole('button', { name: 'OK' }));
+  expect(addAnnotationLayer).toHaveBeenCalled();
+  expect(close).toHaveBeenCalled();
+});
+
+test('triggers close when cancel button is clicked', () => {
+  const close = jest.fn();
+  render(<AnnotationLayer {...defaultProps} close={close} />);
+  userEvent.click(screen.getByRole('button', { name: 'Cancel' }));
+  expect(close).toHaveBeenCalled();
+});
+
+test('triggers removeAnnotationLayer and close when remove button is clicked', () => {
+  const removeAnnotationLayer = jest.fn();
+  const close = jest.fn();
+  render(
+    <AnnotationLayer
+      {...defaultProps}
+      name="Test"
+      value="2x"
+      removeAnnotationLayer={removeAnnotationLayer}
+      close={close}
+    />,
+  );
+  userEvent.click(screen.getByRole('button', { name: 'Remove' }));
+  expect(removeAnnotationLayer).toHaveBeenCalled();
+  expect(close).toHaveBeenCalled();
+});
+
+test('renders chart options', async () => {
+  render(
+    <AnnotationLayer
+      {...defaultProps}
+      annotationType={ANNOTATION_TYPES_METADATA.EVENT.value}
+    />,
+  );
+  userEvent.click(screen.getByText('2 option(s)'));
+  userEvent.click(screen.getByText('Superset annotation'));
+  expect(await screen.findByLabelText('Annotation layer')).toBeInTheDocument();
+  userEvent.click(screen.getByText('Superset annotation'));
+  userEvent.click(screen.getByText('Table'));
+  expect(await screen.findByLabelText('Chart')).toBeInTheDocument();
+});
+
+test('keeps apply disabled when missing required fields', async () => {
+  render(
+    <AnnotationLayer
+      {...defaultProps}
+      annotationType={ANNOTATION_TYPES_METADATA.EVENT.value}
+      sourceType="Table"
+    />,
+  );
+  userEvent.click(await screen.findByText('1 option(s)'));
+  userEvent.click(screen.getByText('Chart A'));
+  expect(
+    screen.getByText('Annotation Slice Configuration'),
+  ).toBeInTheDocument();
+
+  userEvent.click(screen.getByRole('button', { name: 'Automatic Color' }));
+  userEvent.click(screen.getByLabelText('Title Column'));
+  userEvent.click(screen.getByText('None'));
+  userEvent.click(screen.getByLabelText('Style'));
+  userEvent.click(screen.getByText('Dashed'));
+  userEvent.click(screen.getByLabelText('Opacity'));
+  userEvent.click(screen.getByText('0.5'));
+
+  const checkboxes = screen.getAllByRole('checkbox');
+  checkboxes.forEach(checkbox => userEvent.click(checkbox));
+
+  expect(screen.getByRole('button', { name: 'Apply' })).toBeDisabled();
+});

superset-frontend/src/explore/components/controls/SelectControl.jsx

@@ -221,6 +221,7 @@ export default class SelectControl extends React.PureComponent {
       disabled,
       filterOption,
       isLoading,
+      label,
       menuPlacement,
       name,
       noResultsText,
@@ -255,6 +256,7 @@ export default class SelectControl extends React.PureComponent {
 
     const selectProps = {
       autoFocus,
+      'aria-label': label,
       clearable,
       disabled,
       filterOption,

superset-frontend/src/explore/components/controls/TextControl/index.tsx

@@ -26,6 +26,7 @@ import ControlHeader from 'src/explore/components/ControlHeader';
 type InputValueType = string | number;
 
 export interface TextControlProps<T extends InputValueType = InputValueType> {
+  label?: string;
   disabled?: boolean;
   isFloat?: boolean;
   isInt?: boolean;
@@ -109,7 +110,7 @@ export default class TextControl<
     return (
       <div>
         <ControlHeader {...this.props} />
-        <FormGroup controlId={this.state.controlId} bsSize="medium">
+        <FormGroup controlId={this.state.controlId}>
           <FormControl
             type="text"
             data-test="inline-name"
@@ -118,6 +119,7 @@ export default class TextControl<
             onFocus={this.props.onFocus}
             value={value}
             disabled={this.props.disabled}
+            aria-label={this.props.label}
           />
         </FormGroup>
       </div>