superset2/superset/assets/spec/javascripts/explore/components/RunQueryActionButton_spec.jsx

import React from 'react';
import { expect } from 'chai';
import { describe, it, beforeEach } from 'mocha';
import { shallow } from 'enzyme';

import RunQueryActionButton
  from '../../../../src/SqlLab/components/RunQueryActionButton';
import Button from '../../../../src/components/Button';

describe('RunQueryActionButton', () => {
  let wrapper;
  const defaultProps = {
    allowAsync: false,
    dbId: 1,
    queryState: 'pending',
    runQuery: () => {}, // eslint-disable-line
    selectedText: null,
    stopQuery: () => {}, // eslint-disable-line
    sql: '',
  };

  beforeEach(() => {
    wrapper = shallow(<RunQueryActionButton {...defaultProps} />);
  });

  it('is a valid react element', () => {
    expect(
      React.isValidElement(<RunQueryActionButton {...defaultProps} />),
    ).to.equal(true);
  });

  it('renders a single Button', () => {
    expect(wrapper.find(Button)).to.have.lengthOf(1);
  });
});