Fix

Author: RobertIndie

pulsar/internal/key_based_batch_builder_test.go

@@ -30,13 +30,6 @@ import (
 	"google.golang.org/protobuf/proto"
 )
 
-type mockBufferPool struct {
-}
-
-func (m *mockBufferPool) GetBuffer() Buffer {
-	return nil
-}
-
 type mockEncryptor struct {
 }
 
@@ -53,7 +46,7 @@ func TestKeyBasedBatcherOrdering(t *testing.T) {
 		1,
 		pb.CompressionType_NONE,
 		compression.Level(0),
-		&mockBufferPool{},
+		&bufferPoolImpl{},
 		log.NewLoggerWithLogrus(logrus.StandardLogger()),
 		&mockEncryptor{},
 	)

pulsar/producer_test.go

@@ -2734,7 +2734,7 @@ func TestSendingBuffersCleanupAfterMultipleReconnections(t *testing.T) {
 		// Send many messages asynchronously without waiting for completion
 		// This generates a lot of sending buffers that need to be cleaned up
 		for j := 0; j < 1000; j++ {
-			p.SendAsync(t.Context(), &ProducerMessage{
+			p.SendAsync(context.Background(), &ProducerMessage{
 				Payload: []byte("test"),
 			}, nil)
 		}