From 442c970c63bcac4ed3336c0f7a47305821819a70 Mon Sep 17 00:00:00 2001 From: Erik van Oosten Date: Sun, 12 Jan 2025 11:41:02 +0100 Subject: [PATCH] Fix logging in test --- .../src/test/scala/zio/kafka/consumer/ConsumerSpec.scala | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/zio-kafka-test/src/test/scala/zio/kafka/consumer/ConsumerSpec.scala b/zio-kafka-test/src/test/scala/zio/kafka/consumer/ConsumerSpec.scala index 6a04e81b1..8dd44fbee 100644 --- a/zio-kafka-test/src/test/scala/zio/kafka/consumer/ConsumerSpec.scala +++ b/zio-kafka-test/src/test/scala/zio/kafka/consumer/ConsumerSpec.scala @@ -946,7 +946,10 @@ object ConsumerSpec extends ZIOSpecDefaultSlf4j with KafkaRandom { c1Diagnostics = new Diagnostics { override def emit(event: => DiagnosticEvent): UIO[Unit] = event match { case r: DiagnosticEvent.Rebalance if r.assigned.size == 1 => - Clock.instant.flatMap(rebalanceEndTimePromise.succeed).unit + ZIO.logDebug(s"Rebalance finished: $r") *> + Clock.instant.flatMap(rebalanceEndTimePromise.succeed).unit + case r: DiagnosticEvent.Rebalance => + ZIO.logDebug(s"Rebalance finished: $r") case _ => ZIO.unit } } @@ -997,7 +1000,7 @@ object ConsumerSpec extends ZIOSpecDefaultSlf4j with KafkaRandom { .truncatedTo(ChronoUnit.SECONDS) .getSeconds .toInt - _ <- ZIO.logError(s"Rebalance took $rebalanceDuration seconds") + _ <- ZIO.logDebug(s"Rebalance took $rebalanceDuration seconds") } yield assertTrue(rebalanceDuration <= 2) }, test("partitions for topic doesn't fail if doesn't exist") {