fix(mock consumer): HighWaterMarkOffset

IBM · Mar 3, 2023 · e47c908 · e47c908
1 parent 397cee4
commit e47c908
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/mocks/consumer.go b/mocks/consumer.go
@@ -343,7 +343,7 @@ func (pc *PartitionConsumer) Messages() <-chan *sarama.ConsumerMessage {
 }
 
 func (pc *PartitionConsumer) HighWaterMarkOffset() int64 {
-	return atomic.LoadInt64(&pc.highWaterMarkOffset) + 1
+	return atomic.LoadInt64(&pc.highWaterMarkOffset)
 }
 
 // Pause implements the Pause method from the sarama.PartitionConsumer interface.

diff --git a/mocks/consumer_test.go b/mocks/consumer_test.go
@@ -7,6 +7,7 @@ import (
 	"testing"
 
 	"github.com/Shopify/sarama"
+	assert "github.com/stretchr/testify/require"
 )
 
 func TestMockConsumerImplementsConsumerInterface(t *testing.T) {
@@ -395,6 +396,8 @@ func TestConsumerOffsetsAreManagedCorrectlyWithSpecifiedOffset(t *testing.T) {
 	if len(trm.errors) != 0 {
 		t.Errorf("Expected to not report any errors, found: %v", trm.errors)
 	}
+
+	assert.True(t, pc.HighWaterMarkOffset() == message2.Offset+1)
 }
 
 func TestConsumerInvalidConfiguration(t *testing.T) {