Add test coverage for VimMachineService

kubernetes-sigs · Dec 18, 2023 · 6cdadaa · 6cdadaa
1 parent 919c03f
commit 6cdadaa
Show file tree

Hide file tree

Showing 4 changed files with 592 additions and 227 deletions.
diff --git a/pkg/context/fake/constants.go b/pkg/context/fake/constants.go
@@ -50,6 +50,9 @@ const (
 	// Clusterv1a2UUID is the UID of the fake CAPI v1alpha3 Cluster resource.
 	Clusterv1a2UUID = "00000000-0000-0000-0000-000000000000"
 
+	// InfrastructureRefName is the name of the InfrastructureRef in fake CAPI Cluster resource.
+	InfrastructureRefName = "fake-cluster-infrastructure-ref"
+
 	VSphereClusterName = "fake-vsphere-cluster"
 
 	// VSphereClusterUUID is the UID of the fake VSphereCluster resource.

diff --git a/pkg/context/fake/fake_cluster_context.go b/pkg/context/fake/fake_cluster_context.go
@@ -19,6 +19,7 @@ package fake
 import (
 	"context"
 
+	corev1 "k8s.io/api/core/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	clusterv1 "sigs.k8s.io/cluster-api/api/v1beta1"
 
@@ -63,6 +64,10 @@ func newClusterV1() clusterv1.Cluster {
 					CIDRBlocks: []string{ServiceCIDR},
 				},
 			},
+			InfrastructureRef: &corev1.ObjectReference{
+				Namespace: Namespace,
+				Name:      InfrastructureRefName,
+			},
 		},
 	}
 }

diff --git a/pkg/services/services_suite_test.go b/pkg/services/services_suite_test.go