energylink/public/app/core/components/DeleteButton/DeleteButton.test.tsx

import React from 'react';
import DeleteButton from './DeleteButton';
import { shallow } from 'enzyme';

describe('DeleteButton', () => {
  let wrapper;
  let deleted;

  beforeAll(() => {
    deleted = false;

    function deleteItem() {
      deleted = true;
    }
    wrapper = shallow(<DeleteButton onConfirmDelete={() => deleteItem()} />);
  });

  it('should show confirm delete when clicked', () => {
    expect(wrapper.state().showConfirm).toBe(false);
    wrapper.find('.delete-button').simulate('click');
    expect(wrapper.state().showConfirm).toBe(true);
  });

  it('should hide confirm delete when clicked', () => {
    wrapper.find('.delete-button').simulate('click');
    expect(wrapper.state().showConfirm).toBe(true);
    wrapper
      .find('.confirm-delete')
      .find('.btn')
      .at(0)
      .simulate('click');
    expect(wrapper.state().showConfirm).toBe(false);
  });

  it('should show confirm delete when clicked', () => {
    expect(deleted).toBe(false);
    wrapper
      .find('.confirm-delete')
      .find('.btn')
      .at(1)
      .simulate('click');
    expect(deleted).toBe(true);
  });
});