Merge pull request #451 from moonpolysoft/fix-deadlock-in-consumer-mock

fix a deadlock in the consumer mock
lovoo · Apr 28, 2024 · d8004e7 · d8004e7
2 parents 3c728ec + fdd1ae5
commit d8004e7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mockautoconsumers.go b/mockautoconsumers.go
@@ -249,7 +249,9 @@ func (pc *MockAutoPartitionConsumer) Close() error {
 			pc.t.Errorf("Expected the messages channel for %s/%d to be drained on close, but found %d messages.", pc.topic, pc.partition, len(pc.messages))
 		}
 
-		pc.AsyncClose()
+		close(pc.messages)
+		close(pc.errors)
+		pc.consumed = false
 
 		var (
 			closeErr error