diff --git a/mocks/sync_producer.go b/mocks/sync_producer.go index 3f4986e2f8..8a089e0dc2 100644 --- a/mocks/sync_producer.go +++ b/mocks/sync_producer.go @@ -97,6 +97,8 @@ func (sp *SyncProducer) SendMessages(msgs []*sarama.ProducerMessage) error { if expectation.Result != errProduceSuccess { return expectation.Result } + sp.lastOffset++ + msgs[i].Offset = sp.lastOffset } return nil } diff --git a/mocks/sync_producer_test.go b/mocks/sync_producer_test.go index bf2c71a195..55799839a7 100644 --- a/mocks/sync_producer_test.go +++ b/mocks/sync_producer_test.go @@ -161,6 +161,12 @@ func TestSyncProducerWithCheckerFunctionForSendMessagesWithoutError(t *testing.T t.Error("No error expected on SendMessages call, found: ", err) } + for i, msg := range msgs { + if int64(i+1) != msg.Offset { + t.Errorf("The %dth message should have been assigned offset 1, but got %d", i, msg.Offset) + } + } + if err := sp.Close(); err != nil { t.Error(err) }