pill.test.tsx

import { shallow } from 'enzyme'
import * as React from 'react'
import { Badge } from 'react-bootstrap'

import { Pill } from '../src'

describe('Pill', () => {
  it('Pill renders itself without crashing', () => {
    const pillWrapper = shallow(<Pill />)
    const badge = pillWrapper.find(Badge)
    expect(badge.props().pill).toBeTruthy()
    expect(pillWrapper.find(Badge)).toHaveLength(1)
  })

  it('Pill defaults color to primary', () => {
    const pillWrapper = shallow(<Pill />)
    const badge = pillWrapper.find(Badge)
    expect(badge.props().variant).toEqual('primary')
  })

  it('Pill can change colors', () => {
    const pillWrapper = shallow(<Pill color="secondary" />)
    const badge = pillWrapper.find(Badge)
    expect(badge.props().variant).toEqual('secondary')
  })

  it('Pill can redner children elements', () => {
    const pillWrapper = shallow(<Pill>Hello World</Pill>)
    const badge = pillWrapper.find(Badge)
    expect(badge.text()).toEqual('Hello World')
  })

  it('Pill can use custom class', () => {
    const pillWrapper = shallow(<Pill className="customClass" />)
    const badge = pillWrapper.find(Badge)
    expect(badge.props().className).toEqual('customClass')
  })

  it('Pill can use custom style', () => {
    const pillWrapper = shallow(<Pill style={{ background: 'red' }} />)
    const badge = pillWrapper.find(Badge)
    expect(badge.props().style).toMatchObject({ background: 'red' })
  })
})