Created
June 20, 2017 14:09
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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