Fix build

datastax · Jul 19, 2024 · bca996c · bca996c
1 parent 6e4b20d
commit bca996c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/proxy/pkg/zdmproxy/streamidmapper_test.go b/proxy/pkg/zdmproxy/streamidmapper_test.go
@@ -2,20 +2,21 @@ package zdmproxy
 
 import (
 	"github.com/datastax/go-cassandra-native-protocol/primitive"
+	"github.com/datastax/zdm-proxy/proxy/pkg/config"
 	"github.com/stretchr/testify/require"
 	"sync"
 	"testing"
 )
 
 func TestStreamIdMapper(t *testing.T) {
-	var mapper = NewStreamIdMapper(primitive.ProtocolVersion3, 2048, nil)
+	var mapper = NewStreamIdMapper(primitive.ProtocolVersion3, &config.Config{ProxyMaxStreamIds: 2048}, nil)
 	var syntheticId, _ = mapper.GetNewIdFor(1000)
 	var originalId, _ = mapper.ReleaseId(syntheticId)
 	require.Equal(t, int16(1000), originalId)
 }
 
 func BenchmarkStreamIdMapper(b *testing.B) {
-	var mapper = NewStreamIdMapper(primitive.ProtocolVersion3, 2048, nil)
+	var mapper = NewStreamIdMapper(primitive.ProtocolVersion3, &config.Config{ProxyMaxStreamIds: 2048}, nil)
 	for i := 0; i < b.N; i++ {
 		var originalId = int16(i)
 		var syntheticId, _ = mapper.GetNewIdFor(originalId)
@@ -29,7 +30,7 @@ func TestConcurrentStreamIdMapper(t *testing.T) {
 	var wg = sync.WaitGroup{}
 	wg.Add(concurrency)
 	for i := 0; i < concurrency; i++ {
-		var mapper = NewStreamIdMapper(primitive.ProtocolVersion3, 2048, nil)
+		var mapper = NewStreamIdMapper(primitive.ProtocolVersion3, &config.Config{ProxyMaxStreamIds: 2048}, nil)
 		getAndReleaseIds(t, mapper, int16(i), requestCount, &wg)
 	}
 	wg.Wait()