Fix tests

gatewayd-io · Nov 19, 2023 · 58514fd · 58514fd
1 parent 96c2e7d
commit 58514fd
Show file tree

Hide file tree

Showing 5 changed files with 20 additions and 17 deletions.
diff --git a/api/api_test.go b/api/api_test.go
@@ -180,7 +180,7 @@ func TestGetProxies(t *testing.T) {
 		Network: config.DefaultNetwork,
 		Address: config.DefaultAddress,
 	}
-	client := network.NewClient(context.TODO(), clientConfig, zerolog.Logger{})
+	client := network.NewClient(context.TODO(), clientConfig, zerolog.Logger{}, nil)
 	newPool := pool.NewPool(context.TODO(), 1)
 	assert.Nil(t, newPool.Put(client.ID, client))
 
@@ -225,7 +225,7 @@ func TestGetServers(t *testing.T) {
 		Network: config.DefaultNetwork,
 		Address: config.DefaultAddress,
 	}
-	client := network.NewClient(context.TODO(), clientConfig, zerolog.Logger{})
+	client := network.NewClient(context.TODO(), clientConfig, zerolog.Logger{}, nil)
 	newPool := pool.NewPool(context.TODO(), 1)
 	assert.Nil(t, newPool.Put(client.ID, client))
 

diff --git a/network/client_test.go b/network/client_test.go
@@ -37,7 +37,8 @@ func CreateNewClient(t *testing.T) *Client {
 			TCPKeepAlive:       false,
 			TCPKeepAlivePeriod: config.DefaultTCPKeepAlivePeriod,
 		},
-		logger)
+		logger,
+		nil)
 
 	return client
 }
@@ -148,7 +149,7 @@ func BenchmarkNewClient(b *testing.B) {
 			DialTimeout:        config.DefaultDialTimeout,
 			TCPKeepAlive:       false,
 			TCPKeepAlivePeriod: config.DefaultTCPKeepAlivePeriod,
-		}, logger)
+		}, logger, nil)
 		client.Close()
 	}
 }
@@ -174,7 +175,7 @@ func BenchmarkSend(b *testing.B) {
 			TCPKeepAlive:       false,
 			TCPKeepAlivePeriod: config.DefaultTCPKeepAlivePeriod,
 		},
-		logger)
+		logger, nil)
 	defer client.Close()
 
 	packet := CreatePgStartupPacket()
@@ -205,7 +206,7 @@ func BenchmarkReceive(b *testing.B) {
 			TCPKeepAlive:       false,
 			TCPKeepAlivePeriod: config.DefaultTCPKeepAlivePeriod,
 		},
-		logger)
+		logger, nil)
 	defer client.Close()
 
 	packet := CreatePgStartupPacket()
@@ -236,7 +237,7 @@ func BenchmarkIsConnected(b *testing.B) {
 			TCPKeepAlive:       false,
 			TCPKeepAlivePeriod: config.DefaultTCPKeepAlivePeriod,
 		},
-		logger)
+		logger, nil)
 	defer client.Close()
 
 	for i := 0; i < b.N; i++ {

diff --git a/network/proxy_test.go b/network/proxy_test.go
@@ -38,7 +38,8 @@ func TestNewProxy(t *testing.T) {
 			TCPKeepAlive:       false,
 			TCPKeepAlivePeriod: config.DefaultTCPKeepAlivePeriod,
 		},
-		logger)
+		logger,
+		nil)
 	err := newPool.Put(client.ID, client)
 	assert.Nil(t, err)
 
@@ -232,7 +233,7 @@ func BenchmarkProxyConnectDisconnect(b *testing.B) {
 		TCPKeepAlive:       false,
 		TCPKeepAlivePeriod: config.DefaultTCPKeepAlivePeriod,
 	}
-	newPool.Put("client", NewClient(context.Background(), &clientConfig, logger)) //nolint:errcheck
+	newPool.Put("client", NewClient(context.Background(), &clientConfig, logger, nil)) //nolint:errcheck
 
 	// Create a proxy with a fixed buffer newPool
 	proxy := NewProxy(
@@ -286,7 +287,7 @@ func BenchmarkProxyPassThrough(b *testing.B) {
 		TCPKeepAlive:       false,
 		TCPKeepAlivePeriod: config.DefaultTCPKeepAlivePeriod,
 	}
-	newPool.Put("client", NewClient(context.Background(), &clientConfig, logger)) //nolint:errcheck
+	newPool.Put("client", NewClient(context.Background(), &clientConfig, logger, nil)) //nolint:errcheck
 
 	// Create a proxy with a fixed buffer newPool
 	proxy := NewProxy(
@@ -344,7 +345,7 @@ func BenchmarkProxyIsHealthyAndIsExhausted(b *testing.B) {
 		TCPKeepAlive:       false,
 		TCPKeepAlivePeriod: config.DefaultTCPKeepAlivePeriod,
 	}
-	client := NewClient(context.Background(), &clientConfig, logger)
+	client := NewClient(context.Background(), &clientConfig, logger, nil)
 	newPool.Put("client", client) //nolint:errcheck
 
 	// Create a proxy with a fixed buffer newPool
@@ -401,7 +402,7 @@ func BenchmarkProxyAvailableAndBusyConnections(b *testing.B) {
 		TCPKeepAlive:       false,
 		TCPKeepAlivePeriod: config.DefaultTCPKeepAlivePeriod,
 	}
-	client := NewClient(context.Background(), &clientConfig, logger)
+	client := NewClient(context.Background(), &clientConfig, logger, nil)
 	newPool.Put("client", client) //nolint:errcheck
 
 	// Create a proxy with a fixed buffer newPool

diff --git a/network/server_test.go b/network/server_test.go
@@ -65,13 +65,13 @@ func TestRunServer(t *testing.T) {
 
 	// Create a connection newPool.
 	newPool := pool.NewPool(context.Background(), 3)
-	client1 := NewClient(context.Background(), &clientConfig, logger)
+	client1 := NewClient(context.Background(), &clientConfig, logger, nil)
 	err := newPool.Put(client1.ID, client1)
 	assert.Nil(t, err)
-	client2 := NewClient(context.Background(), &clientConfig, logger)
+	client2 := NewClient(context.Background(), &clientConfig, logger, nil)
 	err = newPool.Put(client2.ID, client2)
 	assert.Nil(t, err)
-	client3 := NewClient(context.Background(), &clientConfig, logger)
+	client3 := NewClient(context.Background(), &clientConfig, logger, nil)
 	err = newPool.Put(client3.ID, client3)
 	assert.Nil(t, err)
 
@@ -138,7 +138,8 @@ func TestRunServer(t *testing.T) {
 				TCPKeepAlive:       false,
 				TCPKeepAlivePeriod: config.DefaultTCPKeepAlivePeriod,
 			},
-			logger)
+			logger,
+			nil)
 
 		assert.NotNil(t, client)
 		sent, err := client.Send(CreatePgStartupPacket())

diff --git a/network/utils_test.go b/network/utils_test.go
@@ -165,7 +165,7 @@ func BenchmarkTrafficData(b *testing.B) {
 		TCPKeepAlive:       false,
 		TCPKeepAlivePeriod: time.Second * 10,
 		ReceiveChunkSize:   1024,
-	}, logger)
+	}, logger, nil)
 	fields := []Field{
 		{
 			Name:  "test",