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

import { Badge } from '.'
import { SmallSquire, Squire, Circle } from './demos/shapes'

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

  test('different size.', () => {
    const tree = renderer.create(
      <>
        <Badge size="default"><Squire /></Badge>
        <Badge count={100} size="default"><Squire /></Badge>
        <Badge count={1000} overflowCount={999} size="small"><SmallSquire /></Badge>
        <Badge size="small"><SmallSquire /></Badge>
      </>
    ).toJSON()
    expect(tree).toMatchSnapshot()
  })

  test('different type.', () => {
    const tree = renderer.create(
      <>
        <Badge type="warning"><Squire /></Badge>
        <Badge type="primary"><Squire /></Badge>
      </>
    ).toJSON()
    expect(tree).toMatchSnapshot()
  })
})