fix: ensure mock SyncProducer's SendMessage returns msg.Partition instead of 0 (#3122)

magiusdarrigo · web-flow · commit 54480cc728ef · 2025-03-14T00:57:46.000Z
Signed-off-by: Matteo Agius &lt;magiusdarrigo@gmail.com&gt;
diff --git a/mocks/sync_producer.go b/mocks/sync_producer.go
@@ -87,7 +87,7 @@ func (sp *SyncProducer) SendMessage(msg *sarama.ProducerMessage) (partition int3
 		if errors.Is(expectation.Result, errProduceSuccess) {
 			sp.lastOffset++
 			msg.Offset = sp.lastOffset
-			return 0, msg.Offset, nil
+			return msg.Partition, msg.Offset, nil
 		}
 		return -1, -1, expectation.Result
 	}

Original file line number	Diff line number	Diff line change
`@@ -87,7 +87,7 @@ func (sp SyncProducer) SendMessage(msg sarama.ProducerMessage) (partition int3`
`87`	`87`	`if errors.Is(expectation.Result, errProduceSuccess) {`
`88`	`88`	`sp.lastOffset++`
`89`	`89`	`msg.Offset = sp.lastOffset`
`90`		`- return 0, msg.Offset, nil`
	`90`	`+ return msg.Partition, msg.Offset, nil`
`91`	`91`	`}`
`92`	`92`	`return -1, -1, expectation.Result`
`93`	`93`	`}`