import { plainToClass } from 'class-transformer'
import { factory } from 'factory-bot'
import { datatype, date, internet, random } from 'faker'
import { sample } from 'lodash'

import { Competition, CompetitionCategories } from '../../src/models'

factory.define(
  'competition',
  Competition,
  {
    id: datatype.uuid,
    title: random.words,
    category: () => sample(Object.values(CompetitionCategories)),
    patient_id: datatype.uuid,
    winner_id: datatype.uuid,
    link: internet.url,
    created_at: date.past,
    will_end_at: date.recent(-30),
    cancelled_at: null,
    finished_at: null,
  },
  {
    afterBuild: model => {
      return plainToClass(Competition, { ...model })
    },
  },
)

export { factory }