feat: fix unit test case for retry producing 5 times in batch_consumer

Trendyol · Dec 22, 2024 · 7d8489f · 7d8489f
1 parent 4e38c53
commit 7d8489f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/batch_consumer_test.go b/batch_consumer_test.go
@@ -532,6 +532,13 @@ func (m *mockCronsumer) GetMetricCollectors() []prometheus.Collector {
 }
 
 func (m *mockCronsumer) ProduceBatch([]kcronsumer.Message) error {
+	if m.retryBehaviorOpen {
+		if m.wantErr && m.times <= m.maxRetry {
+			m.times++
+			return errors.New("error")
+		}
+		return nil
+	}
 	if m.wantErr {
 		return errors.New("error")
 	}

diff --git a/examples/docker-compose.yml b/examples/docker-compose.yml
@@ -10,7 +10,7 @@ services:
       - "4318:4318"
 
   zookeeper:
-    image: debezium/zookeeper:2.4
+    image: debezium/zookeeper
     ports:
       - "2181:2181"