From b68a2d1ce58d8628e9714b0d4976683cc77826d2 Mon Sep 17 00:00:00 2001
From: Maciej Modzelewski <maciej@modzelewski.dev>
Date: Mon, 17 Feb 2025 21:34:22 +0100
Subject: [PATCH] Fix consumer offset tests, add setting consumer offset

---
 .../blocking/ConsumerOffsetsClientBaseTest.java        | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/java-sdk/src/test/java/rs/iggy/clients/blocking/ConsumerOffsetsClientBaseTest.java b/java-sdk/src/test/java/rs/iggy/clients/blocking/ConsumerOffsetsClientBaseTest.java
index f828fc7..c8fd505 100644
--- a/java-sdk/src/test/java/rs/iggy/clients/blocking/ConsumerOffsetsClientBaseTest.java
+++ b/java-sdk/src/test/java/rs/iggy/clients/blocking/ConsumerOffsetsClientBaseTest.java
@@ -6,6 +6,7 @@
 import rs.iggy.identifier.ConsumerId;
 import rs.iggy.identifier.StreamId;
 import rs.iggy.identifier.TopicId;
+import java.math.BigInteger;
 import java.util.Optional;
 import static org.assertj.core.api.Assertions.assertThat;
 
@@ -24,11 +25,16 @@ void beforeEachBase() {
     @Test
     void shouldGetConsumerOffset() {
         // when
+        var consumer = new Consumer(Consumer.Kind.Consumer, ConsumerId.of(1223L));
+        consumerOffsetsClient.storeConsumerOffset(StreamId.of(42L),
+                TopicId.of(42L),
+                Optional.empty(),
+                consumer,
+                BigInteger.ZERO);
         var consumerOffset = consumerOffsetsClient.getConsumerOffset(StreamId.of(42L),
                 TopicId.of(42L),
                 Optional.of(1L),
-                new Consumer(
-                        Consumer.Kind.Consumer, ConsumerId.of(1223L)));
+                consumer);
 
         var nonExistingConsumerOffset = consumerOffsetsClient.getConsumerOffset(StreamId.of(42L),
                 TopicId.of(42L),