fix: mock sync producer does not handle the offset while sending mess…

…ages
IBM · Jul 10, 2020 · e6cb2f9 · e6cb2f9
1 parent 5933302
commit e6cb2f9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git 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
@@ -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)
 	}