superset-frontend/packages/superset-ui-core/test/chart/components/FallbackComponent.test.tsx

import React from 'react';
import { shallow } from 'enzyme';
import FallbackComponent from '@superset-ui/core/src/chart/components/FallbackComponent';

describe('FallbackComponent', () => {
  const ERROR = new Error('CaffeineOverLoadException');
  const STACK_TRACE = 'Error at line 1: x.drink(coffee)';

  it('renders error and stack trace', () => {
    const wrapper = shallow(<FallbackComponent componentStack={STACK_TRACE} error={ERROR} />);
    const messages = wrapper.find('code');
    expect(messages).toHaveLength(2);
    expect(messages.at(0).text()).toEqual('Error: CaffeineOverLoadException');
    expect(messages.at(1).text()).toEqual('Error at line 1: x.drink(coffee)');
  });

  it('renders error only', () => {
    const wrapper = shallow(<FallbackComponent error={ERROR} />);
    const messages = wrapper.find('code');
    expect(messages).toHaveLength(1);
    expect(messages.at(0).text()).toEqual('Error: CaffeineOverLoadException');
  });

  it('renders stacktrace only', () => {
    const wrapper = shallow(<FallbackComponent componentStack={STACK_TRACE} />);
    const messages = wrapper.find('code');
    expect(messages).toHaveLength(2);
    expect(messages.at(0).text()).toEqual('Unknown Error');
    expect(messages.at(1).text()).toEqual('Error at line 1: x.drink(coffee)');
  });

  it('renders when nothing is given', () => {
    const wrapper = shallow(<FallbackComponent />);
    const messages = wrapper.find('code');
    expect(messages).toHaveLength(1);
    expect(messages.at(0).text()).toEqual('Unknown Error');
  });
});