Fix test

zio · Nov 9, 2024 · 5ef97ef · 5ef97ef
1 parent d6f485c
commit 5ef97ef
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/zio-kafka/src/main/scala/zio/kafka/consumer/internal/Runloop.scala b/zio-kafka/src/main/scala/zio/kafka/consumer/internal/Runloop.scala
@@ -484,7 +484,8 @@ private[consumer] final class Runloop private (
       pollResult <-
         consumer.runloopAccess { c =>
           for {
-            _ <- verifyAssignedStreamsMatchesAssignment(state.assignedStreams, c.assignment().asScala.toSet)
+            assignment <- ZIO.attempt(c.assignment())
+            _          <- verifyAssignedStreamsMatchesAssignment(state.assignedStreams, assignment.asScala.toSet)
             resumeAndPauseCounts <- resumeAndPausePartitions(c, partitionsToFetch)
             (toResumeCount, toPauseCount) = resumeAndPauseCounts