records test

eo-cqrs · May 8, 2023 · 19674b2 · 19674b2
1 parent 7322e82
commit 19674b2
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/test/java/io/github/eocqrs/kafka/consumer/KfConsumerTest.java b/src/test/java/io/github/eocqrs/kafka/consumer/KfConsumerTest.java
@@ -33,6 +33,7 @@
 import io.github.eocqrs.kafka.parameters.ValueDeserializer;
 import io.github.eocqrs.kafka.xml.KfXmlFlexible;
 
+import java.time.Duration;
 import java.util.Collection;
 
 import org.apache.kafka.clients.consumer.ConsumerRebalanceListener;
@@ -83,6 +84,21 @@ public void onPartitionsAssigned(final Collection<TopicPartition> partitions) {
     );
   }
 
+  @Test
+  void recordsPollingDoesntThrowException(
+    @Mock final ConsumerSettings<String, String> settings,
+    @Mock final KafkaConsumer<String, String> origin
+  ) {
+    Mockito.when(settings.consumer()).thenReturn(origin);
+    final Consumer<String, String> consumer =
+      new KfConsumer<>(settings);
+    assertDoesNotThrow(() ->
+      consumer.records(
+        "TEST", Duration.ofSeconds(5L)
+      )
+    );
+  }
+
   @Test
   void unsubscribesWithoutException(
     @Mock final ConsumerSettings<String, String> settings,