Fix test

softwaremill · Aug 23, 2023 · fb19069 · fb19069
1 parent d0ddc99
commit fb19069
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/kafka/src/main/scala/ox/kafka/KafkaConsumerActor.scala b/kafka/src/main/scala/ox/kafka/KafkaConsumerActor.scala
@@ -56,7 +56,7 @@ object KafkaConsumerActor:
       finally
         if closeWhenComplete then
           logger.debug("Closing the Kafka consumer")
-          consumer.close()
+          uninterruptible(consumer.close())
     }
 
     c

diff --git a/kafka/src/main/scala/ox/kafka/KafkaSink.scala b/kafka/src/main/scala/ox/kafka/KafkaSink.scala
@@ -96,7 +96,7 @@ object KafkaSink:
       finally
         if closeWhenComplete then
           logger.debug("Closing the Kafka producer")
-          producer.close()
+          uninterruptible(producer.close())
     }
 
     c

diff --git a/kafka/src/test/scala/ox/kafka/KafkaTest.scala b/kafka/src/test/scala/ox/kafka/KafkaTest.scala
@@ -64,10 +64,7 @@ class KafkaTest extends AnyFlatSpec with Matchers with EmbeddedKafka with Before
 
     // then
     given Deserializer[String] = new StringDeserializer()
-    consumeFirstStringMessageFrom(topic) shouldBe "a"
-    consumeFirstStringMessageFrom(topic) shouldBe "b"
-    consumeFirstStringMessageFrom(topic) shouldBe "c"
-// TODO   consumeNumberMessagesFrom[String](topic, 3, timeout = 30.seconds) shouldBe List("a", "b", "c")
+    consumeNumberMessagesFrom[String](topic, 3, timeout = 30.seconds) shouldBe List("a", "b", "c")
   }
 
   it should "commit offsets of processed messages" in {