Aerospike Module: Fix unit test

elastic · Feb 23, 2024 · 8965a52 · 8965a52
1 parent 59df82f
commit 8965a52
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/metricbeat/module/aerospike/aerospike_test.go b/metricbeat/module/aerospike/aerospike_test.go
@@ -18,7 +18,6 @@
 package aerospike
 
 import (
-	"crypto/tls"
 	"errors"
 	"fmt"
 	"testing"
@@ -118,7 +117,8 @@ func TestParseClientPolicy(t *testing.T) {
 	UserPasswordClientPolicy.Password = samplePassword
 
 	TLSPolicy := as.NewClientPolicy()
-	TLSPolicy.TlsConfig = &tls.Config{}
+	tlsconfig, _ := tlscommon.LoadTLSConfig(&tlscommon.Config{Enabled: getBoolPointer(true)})
+	TLSPolicy.TlsConfig = tlsconfig.ToConfig()
 
 	ClusterNamePolicy := as.NewClientPolicy()
 	ClusterNamePolicy.ClusterName = sampleClusterName
@@ -167,7 +167,6 @@ func TestParseClientPolicy(t *testing.T) {
 					Enabled: getBoolPointer(true),
 				},
 			},
-
 			expectedClientPolicy: TLSPolicy,
 			expectedErr:          nil,
 		},
@@ -193,8 +192,9 @@ func TestParseClientPolicy(t *testing.T) {
 		}
 		assert.Equal(t, test.expectedClientPolicy.User, result.User, test.Name)
 		assert.Equal(t, test.expectedClientPolicy.Password, result.Password, test.Name)
+		assert.Equal(t, test.expectedClientPolicy.ClusterName, result.ClusterName, test.Name)
 		if test.Config.TLS.IsEnabled() {
-			assert.NotNil(t, test.expectedClientPolicy.TlsConfig)
+			assert.NotNil(t, result.TlsConfig)
 		}
 	}
 }