import React from 'react'
import { shallow } from 'enzyme'
import CardType from '../index'

const wrap = (props = {}) => {
  const rendered = React.createElement(CardType, props)
  return shallow(rendered)
}

function whenCard(number) {
  return {
    typeIs(expectedName) {
      const props = {
        className: 'card-enzyme-wrap',
        card: number,
      }
      const wrapper = wrap(props)
      expect(wrapper.html()).toContain(expectedName)
    },
  }
}

describe('CardType', () => {
  it('works for visa', () => whenCard('001').typeIs('visa'))
  it('works for mastercard', () => whenCard('002').typeIs('mastercard'))
  it('works for amex', () => whenCard('003').typeIs('amex'))
})