From 74e079e0c4d22abe4cb09d54378134facd3536ca Mon Sep 17 00:00:00 2001
From: Steven Vroonland <svroonland@gmail.com>
Date: Sun, 26 Feb 2023 22:38:31 +0100
Subject: [PATCH] Fix test case and reduce test duration

---
 .../test/scala/zio/kafka/consumer/SubscriptionsSpec.scala | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/zio-kafka-test/src/test/scala/zio/kafka/consumer/SubscriptionsSpec.scala b/zio-kafka-test/src/test/scala/zio/kafka/consumer/SubscriptionsSpec.scala
index 16bfe544d..e99b3c979 100644
--- a/zio-kafka-test/src/test/scala/zio/kafka/consumer/SubscriptionsSpec.scala
+++ b/zio-kafka-test/src/test/scala/zio/kafka/consumer/SubscriptionsSpec.scala
@@ -172,7 +172,7 @@ object SubscriptionsSpec extends ZIOKafkaSpec {
       } yield assertCompletes
     },
     test("can resume a stream for the same subscription") {
-      val kvs = (1 to 10000).toList.map(i => (s"key$i", s"msg$i"))
+      val kvs = (1 to 1000).toList.map(i => (s"key$i", s"msg$i"))
       for {
         topic1 <- randomTopic
         client <- randomClient
@@ -186,7 +186,7 @@ object SubscriptionsSpec extends ZIOKafkaSpec {
         _ <-
           Consumer
             .plainStream(Subscription.topics(topic1), Serde.string, Serde.string)
-            .take(200)
+            .take(40)
             .transduce(
               Consumer.offsetBatches.contramap[CommittableRecord[String, String]](_.offset) <&> ZSink
                 .collectAll[CommittableRecord[String, String]]
@@ -195,11 +195,11 @@ object SubscriptionsSpec extends ZIOKafkaSpec {
             .flattenChunks
             .runCollect
             .tap(records => recordsConsumed.update(_ ++ records))
-            .repeatN(50)
+            .repeatN(24)
             .provideSomeLayer[Kafka with Scope](consumer(client, Some(group)))
         consumed <- recordsConsumed.get
       } yield assert(consumed.map(r => r.value))(hasSameElements(Chunk.fromIterable(kvs.map(_._2))))
-    } @@ TestAspect.nonFlaky(5)
+    } @@ TestAspect.nonFlaky(3)
   ).provideSomeLayerShared[TestEnvironment & Kafka](
     producer ++ Scope.default ++ Runtime.removeDefaultLoggers ++ Runtime.addLogger(logger)
   ) @@ withLiveClock @@ TestAspect.sequential @@ timeout(600.seconds)