import { factory } from 'factory-bot'
import { datatype, random } from 'faker'
import { sample } from 'lodash'

import { SmartfitUnit } from '../../src/models'

factory.define(
  'smartfit_unit',
  SmartfitUnit,
  {
    id: () => datatype.uuid(),
    name: () => random.word(),
    acronym: () => random.word(),
    type: () => sample(['own', 'franchise']),
    country_code: () => 'BR',
    allow_teacher_indication: () => datatype.boolean(),
  },
  SmartfitUnit,
)

export { factory }