Skip to content

Instantly share code, notes, and snippets.

@robhinds
Created June 20, 2017 14:09
import org.scalatest.FunSpec
import shapeless.{HList, HNil, _}
import scala.util.Random
trait Generator[A] {
def generate: A
}
object Generator {
def generate[A](implicit gen: Generator[A]) = gen.generate
implicit def intGenerator = new Generator[Int] {
override def generate = Random.nextInt
}
implicit def doubleGenerator = new Generator[Double] {
override def generate = Random.nextDouble
}
implicit def StringGenerator = new Generator[String] {
val loremWords = "Lorem ipsum dolor sit amet, consectetur adipisicing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.".split(" ")
override def generate = Random.shuffle(loremWords.toList).take(5).mkString(" ")
}
implicit def booleanGenerator = new Generator[Boolean] {
override def generate = Random.nextBoolean
}
implicit def hnilGenerator = new Generator[HNil] {
override def generate = HNil
}
implicit def hconsGenerator[H, T <: HList](implicit headGen: Generator[H], tailGen: Generator[T]) =
new Generator[H :: T] {
override def generate = headGen.generate :: tailGen.generate
}
implicit def genericToGenerator[T, L <: HList](implicit generic: Generic.Aux[T, L], lGen: Generator[L]): Generator[T] =
new Generator[T] {
override def generate = generic.from(lGen.generate)
}
}
class TestSpec extends FunSpec {
describe("example") {
it("s") {
import Generator._
case class Sample(a: Int, b: Boolean, c: Double, d: String)
println(generate[Sample])
}
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment