Remove scalacheck fork by implementing buildableOfCollCond

Author: mdedetrich

build.sbt

@@ -133,7 +133,7 @@ lazy val core = project
       "com.typesafe.akka"          %% "akka-stream"                    % akkaVersion                % Test,
       "org.scalatest"              %% "scalatest"                      % scalaTestVersion           % Test,
       "org.scalatestplus"          %% scalaTestScalaCheckArtifact      % scalaTestScalaCheckVersion % Test,
-      "org.mdedetrich"             %% "scalacheck"                     % scalaCheckVersion          % Test,
+      "org.scalacheck"             %% "scalacheck"                     % scalaCheckVersion          % Test,
       "com.rallyhealth"            %% "scalacheck-ops_1-16"            % scalaCheckOpsVersion       % Test,
       "com.softwaremill.diffx"     %% "diffx-scalatest-must"           % diffxVersion               % Test,
       "com.typesafe.akka"          %% "akka-stream-testkit"            % akkaVersion                % Test,

core/src/test/scala/io/aiven/guardian/kafka/Generators.scala

@@ -3,6 +3,7 @@ package io.aiven.guardian.kafka
 import io.aiven.guardian.kafka.models.ReducedConsumerRecord
 import org.apache.kafka.common.record.TimestampType
 import org.scalacheck.Gen
+import org.scalacheck.util.Buildable
 
 import scala.collection.mutable.ListBuffer
 import scala.concurrent.duration.FiniteDuration
@@ -113,6 +114,32 @@ object Generators {
     Gen.choose[Long](head.timestamp, last.timestamp - 1).map(millis => FiniteDuration(millis, MILLISECONDS))
   }
 
+  @SuppressWarnings(
+    Array(
+      "scalafix:DisableSyntax.while",
+      "scalafix:DisableSyntax.return"
+    )
+  )
+  @scala.annotation.nowarn("msg=return statement uses an exception")
+  private def buildableOfCollCond[C <: Iterable[T], T](cond: C => Boolean, g: Gen[C])(implicit
+      evb: Buildable[T, C]
+  ): Gen[C] =
+    Gen.infiniteLazyList(g).map { ll =>
+      val it   = ll.iterator
+      val bldr = evb.builder
+      while (true) {
+        val result = bldr.result()
+        if (cond(result)) {
+          return result
+        }
+        bldr ++= it.next()
+      }
+      return bldr.result()
+    }
+
+  private def listOfFillCond[T](finishCondition: List[T] => Boolean, g: => Gen[List[T]]) =
+    buildableOfCollCond[List[T], T](finishCondition, g)
+
   def kafkaDateGen(min: Int = 2,
                    max: Int = 100,
                    padTimestampsMillis: Range = Range.inclusive(1, 10),
@@ -121,7 +148,7 @@ object Generators {
     topic <- kafkaTopic
     records <- {
       val base = Generators.kafkaReducedConsumerRecordsGen(topic, min, max, padTimestampsMillis)
-      condition.fold(base)(c => Gen.listOfFillCond(c, base))
+      condition.fold(base)(c => listOfFillCond(c, base))
     }
   } yield records