Fix TestKeyBasedBatcherOrdering

Author: BewareMyPower

pulsar/internal/key_based_batch_builder_test.go

@@ -25,6 +25,7 @@ import (
 	"github.com/apache/pulsar-client-go/pulsar/internal/compression"
 	pb "github.com/apache/pulsar-client-go/pulsar/internal/pulsar_proto"
 	"github.com/apache/pulsar-client-go/pulsar/log"
+	"github.com/prometheus/client_golang/prometheus"
 	"github.com/sirupsen/logrus"
 	"github.com/stretchr/testify/assert"
 	"google.golang.org/protobuf/proto"
@@ -38,6 +39,7 @@ func (m *mockEncryptor) Encrypt(_ []byte, _ *pb.MessageMetadata) ([]byte, error)
 }
 
 func TestKeyBasedBatcherOrdering(t *testing.T) {
+	metrics := NewMetricsProvider(3, map[string]string{}, prometheus.DefaultRegisterer)
 	keyBatcher, err := NewKeyBasedBatchBuilder(
 		1000,
 		1000,
@@ -46,7 +48,7 @@ func TestKeyBasedBatcherOrdering(t *testing.T) {
 		1,
 		pb.CompressionType_NONE,
 		compression.Level(0),
-		&bufferPoolImpl{},
+		NewBufferPool(metrics.SendingBuffersCount),
 		log.NewLoggerWithLogrus(logrus.StandardLogger()),
 		&mockEncryptor{},
 	)