weaver.scalacheck.Checkers.scala Maven / Gradle / Ivy
package weaver
package scalacheck
import cats.syntax.all._
import cats.{ Applicative, Defer, Show }
import org.scalacheck.rng.Seed
import org.scalacheck.{ Arbitrary, Gen }
import CECompat.Ref
trait Checkers {
self: EffectSuiteAux =>
import Checkers._
type F[A] = this.EffectType[A]
type PropF[A] = Prop[F, A]
private def liftProp[A, B: PropF](f: A => B): A => F[Expectations] = {
f andThen (b => Prop[F, B].lift(b))
}
// Configuration for property-based tests
def checkConfig: CheckConfig = CheckConfig.default
class PartiallyAppliedForall(config: CheckConfig) {
def apply[A1: Arbitrary: Show, B: PropF](f: A1 => B)(
implicit loc: SourceLocation): F[Expectations] =
forall(implicitly[Arbitrary[A1]].arbitrary)(liftProp[A1, B](f))
def apply[A1: Arbitrary: Show, A2: Arbitrary: Show, B: PropF](f: (
A1,
A2) => B)(
implicit loc: SourceLocation): F[Expectations] =
forall(implicitly[Arbitrary[(A1, A2)]].arbitrary)(liftProp(
f.tupled))
def apply[
A1: Arbitrary: Show,
A2: Arbitrary: Show,
A3: Arbitrary: Show,
B: PropF](
f: (A1, A2, A3) => B)(
implicit loc: SourceLocation): F[Expectations] = {
implicit val tuple3Show: Show[(A1, A2, A3)] = {
case (a1, a2, a3) => s"(${a1.show},${a2.show},${a3.show})"
}
forall(implicitly[Arbitrary[(A1, A2, A3)]].arbitrary)(liftProp(
f.tupled))
}
def apply[
A1: Arbitrary: Show,
A2: Arbitrary: Show,
A3: Arbitrary: Show,
A4: Arbitrary: Show,
B: PropF
](f: (A1, A2, A3, A4) => B)(
implicit loc: SourceLocation): F[Expectations] = {
implicit val tuple3Show: Show[(A1, A2, A3, A4)] = {
case (a1, a2, a3, a4) =>
s"(${a1.show},${a2.show},${a3.show},${a4.show})"
}
forall(implicitly[Arbitrary[(A1, A2, A3, A4)]].arbitrary)(
liftProp(f.tupled))
}
def apply[
A1: Arbitrary: Show,
A2: Arbitrary: Show,
A3: Arbitrary: Show,
A4: Arbitrary: Show,
A5: Arbitrary: Show,
B: PropF
](f: (A1, A2, A3, A4, A5) => B)(
implicit loc: SourceLocation): F[Expectations] = {
implicit val tuple3Show: Show[(A1, A2, A3, A4, A5)] = {
case (a1, a2, a3, a4, a5) =>
s"(${a1.show},${a2.show},${a3.show},${a4.show},${a5.show})"
}
forall(implicitly[Arbitrary[(A1, A2, A3, A4, A5)]].arbitrary)(
liftProp(f.tupled))
}
def apply[
A1: Arbitrary: Show,
A2: Arbitrary: Show,
A3: Arbitrary: Show,
A4: Arbitrary: Show,
A5: Arbitrary: Show,
A6: Arbitrary: Show,
B: PropF
](f: (A1, A2, A3, A4, A5, A6) => B)(
implicit loc: SourceLocation): F[Expectations] = {
implicit val tuple3Show: Show[(A1, A2, A3, A4, A5, A6)] = {
case (a1, a2, a3, a4, a5, a6) =>
s"(${a1.show},${a2.show},${a3.show},${a4.show},${a5.show},${a6.show})"
}
forall(implicitly[Arbitrary[(A1, A2, A3, A4, A5, A6)]].arbitrary)(
liftProp(f.tupled))
}
def apply[A: Show, B: PropF](gen: Gen[A])(f: A => B)(
implicit loc: SourceLocation): F[Expectations] =
Ref[F].of(Status.start[A]).flatMap(forall_(gen, liftProp(f)))
private def forall_[A: Show](gen: Gen[A], f: A => F[Expectations])(
state: Ref[F, Status[A]])(
implicit loc: SourceLocation): F[Expectations] = {
paramStream
.parEvalMapUnordered(config.perPropertyParallelism) {
testOneTupled(gen, state, f)
}
.takeWhile(_.shouldContinue, takeFailure = true)
.takeRight(1) // getting the first error (which finishes the stream)
.compile
.last
.map { (x: Option[Status[A]]) =>
x match {
case Some(status) => status.endResult
case None => Expectations.Helpers.success
}
}
}
private def paramStream: fs2.Stream[F, (Gen.Parameters, Seed)] = {
val initial = startSeed(
Gen.Parameters.default
.withSize(config.maximumGeneratorSize)
.withInitialSeed(config.initialSeed.map(Seed(_))))
fs2.Stream.iterate(initial) {
case (p, s) => (p, s.slide)
}
}
}
object forall extends PartiallyAppliedForall(checkConfig) {
def withConfig(config: CheckConfig) = new PartiallyAppliedForall(config)
}
private def testOneTupled[T: Show](
gen: Gen[T],
state: Ref[F, Status[T]],
f: T => F[Expectations])(ps: (Gen.Parameters, Seed)) =
testOne(gen, state, f)(ps._1, ps._2)
private def testOne[T: Show](
gen: Gen[T],
state: Ref[F, Status[T]],
f: T => F[Expectations])(
params: Gen.Parameters,
seed: Seed): F[Status[T]] = {
Defer[F](self.effect).defer {
gen(params, seed)
.traverse(x => f(x).map(x -> _))
.flatTap { (x: Option[(T, Expectations)]) =>
x match {
case Some((_, ex)) if ex.run.isValid => state.update(_.addSuccess)
case Some((t, ex)) => state.update(_.addFailure(t.show, seed, ex))
case None => state.update(_.addDiscard)
}
}
.productR(state.get)
}
}
def startSeed(params: Gen.Parameters): (Gen.Parameters, Seed) =
params.initialSeed match {
case Some(seed) => (params.withNoInitialSeed, seed)
case None => (params, Seed.random())
}
private[scalacheck] case class Status[T](
succeeded: Int,
discarded: Int,
failure: Option[Expectations]
) {
def addSuccess: Status[T] =
if (failure.isEmpty) copy(succeeded = succeeded + 1) else this
def addDiscard: Status[T] =
if (failure.isEmpty) copy(discarded = discarded + 1) else this
def addFailure(input: String, seed: Seed, exp: Expectations): Status[T] =
if (failure.isEmpty) {
val ith = succeeded + discarded + 1
val failure = Expectations.Helpers
.failure(
s"Property test failed on try $ith with seed ${seed} and input $input")
.and(exp)
copy(failure = Some(failure))
} else this
def shouldStop =
failure.isDefined ||
succeeded >= checkConfig.minimumSuccessful ||
discarded >= checkConfig.maximumDiscarded
def shouldContinue = !shouldStop
def endResult(implicit loc: SourceLocation) = failure.getOrElse {
if (succeeded < checkConfig.minimumSuccessful)
Expectations.Helpers.failure(
s"Discarded more inputs ($discarded) than allowed")
else Expectations.Helpers.success
}
}
private object Status {
def start[T] = Status[T](0, 0, None)
}
}
object Checkers {
trait Prop[F[_], A] {
def lift(a: A): F[Expectations]
}
object Prop {
def apply[F[_], B](implicit ev: Prop[F, B]): Prop[F, B] = ev
implicit def wrap[F[_]: Applicative]: Prop[F, Expectations] =
new Prop[F, Expectations] {
def lift(a: Expectations): F[Expectations] = Applicative[F].pure(a)
}
implicit def unwrapped[F[_], FE](
implicit ev: FE <:< F[Expectations]): Prop[F, FE] =
new Prop[F, FE] {
def lift(a: FE): F[Expectations] = ev(a)
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy