import React from 'react'
import renderer from 'react-test-renderer'

import Button from '.'
import { AddIcon } from '../icons'

describe('Button', () => {
  test('basic.', () => {
    const tree = renderer.create(
      <Button>button</Button>
    ).toJSON()
    expect(tree).toMatchSnapshot()
  })

  test('different size.', () => {
    const tree = renderer.create(
      <>
        <Button size="default">button</Button>
        <Button size="large">button</Button>
        <Button size="small">button</Button>
      </>
    ).toJSON()
    expect(tree).toMatchSnapshot()
  })

  test('with icon and endIcon.', () => {
    const tree = renderer.create(
      <Button icon={<AddIcon />} endIcon={<AddIcon />}>button</Button>
    ).toJSON()
    expect(tree).toMatchSnapshot()
  })
})