Add unit tests for blueprint, current state, & desired state.

Author: willie-yao

internal/controllers/topology/cluster/blueprint_test.go

@@ -56,6 +56,8 @@ func TestGetBlueprint(t *testing.T) {
 
 	workerInfrastructureMachineTemplate := builder.InfrastructureMachineTemplate(metav1.NamespaceDefault, "workerinframachinetemplate1").
 		Build()
+	workerInfrastructureMachinePoolTemplate := builder.InfrastructureMachinePoolTemplate(metav1.NamespaceDefault, "workerinframachinepooltemplate1").
+		Build()
 	workerBootstrapTemplate := builder.BootstrapTemplate(metav1.NamespaceDefault, "workerbootstraptemplate1").
 		Build()
 	machineHealthCheck := &clusterv1.MachineHealthCheckClass{
@@ -73,6 +75,14 @@ func TestGetBlueprint(t *testing.T) {
 
 	mds := []clusterv1.MachineDeploymentClass{*machineDeployment}
 
+	machinePools := builder.MachinePoolClass("workerclass2").
+		WithLabels(map[string]string{"foo": "bar"}).
+		WithAnnotations(map[string]string{"a": "b"}).
+		WithInfrastructureTemplate(workerInfrastructureMachinePoolTemplate).
+		WithBootstrapTemplate(workerBootstrapTemplate).
+		Build()
+	mps := []clusterv1.MachinePoolClass{*machinePools}
+
 	// Define test cases.
 	tests := []struct {
 		name         string
@@ -141,6 +151,7 @@ func TestGetBlueprint(t *testing.T) {
 					Template: controlPlaneTemplate,
 				},
 				MachineDeployments: map[string]*scope.MachineDeploymentBlueprint{},
+				MachinePools:       map[string]*scope.MachinePoolBlueprint{},
 			},
 		},
 		{
@@ -167,6 +178,7 @@ func TestGetBlueprint(t *testing.T) {
 					InfrastructureMachineTemplate: controlPlaneInfrastructureMachineTemplate,
 				},
 				MachineDeployments: map[string]*scope.MachineDeploymentBlueprint{},
+				MachinePools:       map[string]*scope.MachinePoolBlueprint{},
 			},
 		},
 		{
@@ -217,6 +229,7 @@ func TestGetBlueprint(t *testing.T) {
 						MachineHealthCheck:            machineHealthCheck,
 					},
 				},
+				MachinePools: map[string]*scope.MachinePoolBlueprint{},
 			},
 		},
 		{
@@ -276,8 +289,60 @@ func TestGetBlueprint(t *testing.T) {
 					MachineHealthCheck:            machineHealthCheck,
 				},
 				MachineDeployments: map[string]*scope.MachineDeploymentBlueprint{},
+				MachinePools:       map[string]*scope.MachinePoolBlueprint{},
 			},
 		},
+		{
+			name: "Should read a ClusterClass with a MachinePoolClass",
+			clusterClass: builder.ClusterClass(metav1.NamespaceDefault, "class1").
+				WithInfrastructureClusterTemplate(infraClusterTemplate).
+				WithControlPlaneTemplate(controlPlaneTemplate).
+				WithWorkerMachinePoolClasses(mps...).
+				Build(),
+			objects: []client.Object{
+				infraClusterTemplate,
+				controlPlaneTemplate,
+				workerInfrastructureMachinePoolTemplate,
+				workerBootstrapTemplate,
+			},
+			want: &scope.ClusterBlueprint{
+				ClusterClass: builder.ClusterClass(metav1.NamespaceDefault, "class1").
+					WithInfrastructureClusterTemplate(infraClusterTemplate).
+					WithControlPlaneTemplate(controlPlaneTemplate).
+					WithWorkerMachinePoolClasses(mps...).
+					Build(),
+				InfrastructureClusterTemplate: infraClusterTemplate,
+				ControlPlane: &scope.ControlPlaneBlueprint{
+					Template: controlPlaneTemplate,
+				},
+				MachineDeployments: map[string]*scope.MachineDeploymentBlueprint{},
+				MachinePools: map[string]*scope.MachinePoolBlueprint{
+					"workerclass2": {
+						Metadata: clusterv1.ObjectMeta{
+							Labels:      map[string]string{"foo": "bar"},
+							Annotations: map[string]string{"a": "b"},
+						},
+						InfrastructureMachinePoolTemplate: workerInfrastructureMachinePoolTemplate,
+						BootstrapTemplate:                 workerBootstrapTemplate,
+					},
+				},
+			},
+		},
+		{
+			name: "Fails if ClusterClass has a MachinePoolClass referencing a BootstrapConfig that does not exist",
+			clusterClass: builder.ClusterClass(metav1.NamespaceDefault, "class1").
+				WithInfrastructureClusterTemplate(infraClusterTemplate).
+				WithControlPlaneTemplate(controlPlaneTemplate).
+				WithWorkerMachinePoolClasses(mps...).
+				Build(),
+			objects: []client.Object{
+				infraClusterTemplate,
+				controlPlaneTemplate,
+				workerInfrastructureMachinePoolTemplate,
+				// workerInfrastructureMachinePoolTemplate is missing!
+			},
+			wantErr: true,
+		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
@@ -334,6 +399,7 @@ func TestGetBlueprint(t *testing.T) {
 			g.Expect(tt.want.InfrastructureClusterTemplate).To(EqualObject(got.InfrastructureClusterTemplate), cmp.Diff(got.InfrastructureClusterTemplate, tt.want.InfrastructureClusterTemplate))
 			g.Expect(got.ControlPlane).To(BeComparableTo(tt.want.ControlPlane), cmp.Diff(got.ControlPlane, tt.want.ControlPlane))
 			g.Expect(tt.want.MachineDeployments).To(BeComparableTo(got.MachineDeployments), got.MachineDeployments, tt.want.MachineDeployments)
+			g.Expect(tt.want.MachinePools).To(BeComparableTo(got.MachinePools), got.MachinePools, tt.want.MachinePools)
 		})
 	}
 }