company-generators-1.hs

company-generators-1.hs

import           Data.Registry
import           Data.Registry.Hedgehog
import           Hedgehog.Gen
import           Hedgehog.Range
import           Protolude                  hiding (list)
import           Test.Data.Registry.Company

registry =
    genFun Company
 <: genFun Department
 <: genFun Employee
 -- we can generate Lists or Maybe of elements
 <: fun (listOf @Department)
 <: fun (listOf @Employee)
 <: fun (maybeOf @Int)
 <: genVal genInt
 <: genVal genText

genInt :: Gen Int
genInt = integral (linear 1 3)

genText :: Gen Text
genText = text (linear 1 10) alpha

myCompanyGen :: GenIO Company
myCompanyGen = make registry