Add K8S_CONTROLLER attributes unit tests

Author: runakash

pkg/cloudmap/client_test.go

@@ -310,6 +310,7 @@ func TestServiceDiscoveryClient_RegisterEndpoints(t *testing.T) {
 		model.EndpointHostnameAttr:      test.Hostname,
 		model.EndpointNodeNameAttr:      test.Nodename,
 		model.ServiceExportCreationAttr: strconv.FormatInt(test.SvcExportCreationTimestamp, 10),
+		model.K8sVersionAttr:            test.PackageVersion,
 	}
 	attrs2 := map[string]string{
 		model.ClusterIdAttr:             test.ClusterId1,
@@ -327,6 +328,7 @@ func TestServiceDiscoveryClient_RegisterEndpoints(t *testing.T) {
 		model.EndpointHostnameAttr:      test.Hostname,
 		model.EndpointNodeNameAttr:      test.Nodename,
 		model.ServiceExportCreationAttr: strconv.FormatInt(test.SvcExportCreationTimestamp, 10),
+		model.K8sVersionAttr:            test.PackageVersion,
 	}
 
 	tc.mockApi.EXPECT().RegisterInstance(context.TODO(), test.SvcId, test.EndptId1, attrs1).
@@ -370,6 +372,7 @@ func TestServiceDiscoveryClient_DeleteEndpoints(t *testing.T) {
 }
 
 func getTestSdClient(t *testing.T) *testSdClient {
+	test.SetTestVersion()
 	mockController := gomock.NewController(t)
 	mockCache := cloudmapMock.NewMockServiceDiscoveryClientCache(mockController)
 	mockApi := cloudmapMock.NewMockServiceDiscoveryApi(mockController)
@@ -409,6 +412,7 @@ func getHttpInstanceSummaryForTest() []types.HttpInstanceSummary {
 				model.EndpointHostnameAttr:      test.Hostname,
 				model.EndpointNodeNameAttr:      test.Nodename,
 				model.ServiceExportCreationAttr: strconv.FormatInt(test.SvcExportCreationTimestamp, 10),
+				model.K8sVersionAttr:            test.PackageVersion,
 			},
 		},
 		{
@@ -429,6 +433,7 @@ func getHttpInstanceSummaryForTest() []types.HttpInstanceSummary {
 				model.EndpointHostnameAttr:      test.Hostname,
 				model.EndpointNodeNameAttr:      test.Nodename,
 				model.ServiceExportCreationAttr: strconv.FormatInt(test.SvcExportCreationTimestamp, 10),
+				model.K8sVersionAttr:            test.PackageVersion,
 			},
 		},
 	}

pkg/controllers/multicluster/serviceexport_controller_test.go

@@ -205,6 +205,7 @@ func getServiceExportScheme() *runtime.Scheme {
 }
 
 func getServiceExportReconciler(t *testing.T, mockClient *cloudmapMock.MockServiceDiscoveryClient, client client.Client) *ServiceExportReconciler {
+	test.SetTestVersion()
 	return &ServiceExportReconciler{
 		Client:       client,
 		Log:          common.NewLoggerWithLogr(testr.New(t)),

test/test-constants.go

@@ -2,7 +2,7 @@ package test
 
 import (
 	"fmt"
-
+	"github.com/aws/aws-cloud-map-mcs-controller-for-k8s/pkg/version"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 
 	aboutv1alpha1 "github.com/aws/aws-cloud-map-mcs-controller-for-k8s/pkg/apis/about/v1alpha1"
@@ -46,8 +46,14 @@ const (
 	SvcExportCreationTimestamp int64 = 1640995200000
 	Hostname                         = "host"
 	Nodename                         = "node"
+	PackageVersion                   = "aws-cloud-map-mcs-controller-for-k8s 0.0.1 (abcd)"
 )
 
+func SetTestVersion() {
+	version.GitVersion = "v0.0.1"
+	version.GitCommit = "abcd"
+}
+
 func GetTestHttpNamespace() *model.Namespace {
 	return &model.Namespace{
 		Id:   HttpNsId,
@@ -111,7 +117,7 @@ func GetTestEndpoint1() *model.Endpoint {
 		ClusterSetId:                   ClusterSet,
 		ServiceType:                    model.ClusterSetIPType,
 		ServiceExportCreationTimestamp: SvcExportCreationTimestamp,
-		Attributes:                     make(map[string]string),
+		Attributes:                     map[string]string{model.K8sVersionAttr: PackageVersion},
 	}
 }
 
@@ -137,7 +143,7 @@ func GetTestEndpoint2() *model.Endpoint {
 		ClusterSetId:                   ClusterSet,
 		ServiceType:                    model.ClusterSetIPType,
 		ServiceExportCreationTimestamp: SvcExportCreationTimestamp,
-		Attributes:                     make(map[string]string),
+		Attributes:                     map[string]string{model.K8sVersionAttr: PackageVersion},
 	}
 }