Skip to content

Commit f0fcc26

Browse files
LiangquanLi930LiangquanLi930
committed
fix PR: 5711 e2e
1 parent 04a4f62 commit f0fcc26

File tree

2 files changed

+106
-2
lines changed

2 files changed

+106
-2
lines changed

‎controllers/awsmachinetemplate_controller.go‎

Lines changed: 62 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@ import (
3333
ctrl "sigs.k8s.io/controller-runtime"
3434
"sigs.k8s.io/controller-runtime/pkg/client"
3535
"sigs.k8s.io/controller-runtime/pkg/controller"
36+
"sigs.k8s.io/controller-runtime/pkg/handler"
3637

3738
infrav1 "sigs.k8s.io/cluster-api-provider-aws/v2/api/v1beta2"
3839
ekscontrolplanev1 "sigs.k8s.io/cluster-api-provider-aws/v2/controlplane/eks/api/v1beta2"
@@ -64,6 +65,17 @@ func (r *AWSMachineTemplateReconciler) SetupWithManager(ctx context.Context, mgr
6465

6566
return ctrl.NewControllerManagedBy(mgr).
6667
For(&infrav1.AWSMachineTemplate{}).
68+
// Watch KubeadmControlPlane creation/update to reconcile referenced AWSMachineTemplate
69+
// This ensures we reconcile when the owner is created, even if cache hasn't synced yet
70+
Watches(
71+
&controlplanev1.KubeadmControlPlane{},
72+
handler.EnqueueRequestsFromMapFunc(r.kubeadmControlPlaneToAWSMachineTemplate),
73+
).
74+
// Watch MachineDeployment creation/update to reconcile referenced AWSMachineTemplate
75+
Watches(
76+
&clusterv1.MachineDeployment{},
77+
handler.EnqueueRequestsFromMapFunc(r.machineDeploymentToAWSMachineTemplate),
78+
).
6779
WithOptions(options).
6880
WithEventFilter(predicates.ResourceNotPausedAndHasFilterLabel(mgr.GetScheme(), log.GetLogger(), r.WatchFilterValue)).
6981
Complete(r)
@@ -420,3 +432,53 @@ func getParentListOptions(obj metav1.ObjectMeta) ([]client.ListOption, error) {
420432
}
421433
return listOpts, nil
422434
}
435+
436+
// kubeadmControlPlaneToAWSMachineTemplate maps KubeadmControlPlane to AWSMachineTemplate reconcile requests.
437+
// This enables the controller to reconcile AWSMachineTemplate when its owner KubeadmControlPlane is created or updated,
438+
// ensuring that nodeInfo can be populated even if the cache hasn't synced yet.
439+
func (r *AWSMachineTemplateReconciler) kubeadmControlPlaneToAWSMachineTemplate(ctx context.Context, o client.Object) []ctrl.Request {
440+
kcp, ok := o.(*controlplanev1.KubeadmControlPlane)
441+
if !ok {
442+
return nil
443+
}
444+
445+
// Check if it references an AWSMachineTemplate
446+
if kcp.Spec.MachineTemplate.Spec.InfrastructureRef.Kind != "AWSMachineTemplate" {
447+
return nil
448+
}
449+
450+
// Return reconcile request for the referenced AWSMachineTemplate
451+
return []ctrl.Request{
452+
{
453+
NamespacedName: client.ObjectKey{
454+
Namespace: kcp.Namespace,
455+
Name: kcp.Spec.MachineTemplate.Spec.InfrastructureRef.Name,
456+
},
457+
},
458+
}
459+
}
460+
461+
// machineDeploymentToAWSMachineTemplate maps MachineDeployment to AWSMachineTemplate reconcile requests.
462+
// This enables the controller to reconcile AWSMachineTemplate when its owner MachineDeployment is created or updated,
463+
// ensuring that nodeInfo can be populated even if the cache hasn't synced yet.
464+
func (r *AWSMachineTemplateReconciler) machineDeploymentToAWSMachineTemplate(ctx context.Context, o client.Object) []ctrl.Request {
465+
md, ok := o.(*clusterv1.MachineDeployment)
466+
if !ok {
467+
return nil
468+
}
469+
470+
// Check if it references an AWSMachineTemplate
471+
if md.Spec.Template.Spec.InfrastructureRef.Kind != "AWSMachineTemplate" {
472+
return nil
473+
}
474+
475+
// Return reconcile request for the referenced AWSMachineTemplate
476+
return []ctrl.Request{
477+
{
478+
NamespacedName: client.ObjectKey{
479+
Namespace: md.Namespace,
480+
Name: md.Spec.Template.Spec.InfrastructureRef.Name,
481+
},
482+
},
483+
}
484+
}

‎test/e2e/suites/unmanaged/unmanaged_functional_test.go‎

Lines changed: 44 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -338,7 +338,7 @@ var _ = ginkgo.Context("[unmanaged] [functional]", func() {
338338
configCluster.ControlPlaneMachineCount = ptr.To[int64](1)
339339
configCluster.WorkerMachineCount = ptr.To[int64](1)
340340
configCluster.Flavor = shared.SSMFlavor
341-
_, md, _ := createCluster(ctx, configCluster, result)
341+
cluster, md, _ := createCluster(ctx, configCluster, result)
342342

343343
workerMachines := framework.GetMachinesByMachineDeployments(ctx, framework.GetMachinesByMachineDeploymentsInput{
344344
Lister: e2eCtx.Environment.BootstrapClusterProxy.GetClient(),
@@ -360,19 +360,61 @@ var _ = ginkgo.Context("[unmanaged] [functional]", func() {
360360
Expect(err).To(BeNil())
361361
Expect(len(awsMachineTemplateList.Items)).To(BeNumerically(">", 0), "Expected at least one AWSMachineTemplate")
362362

363+
ginkgo.By(fmt.Sprintf("Found %d AWSMachineTemplates", len(awsMachineTemplateList.Items)))
364+
ginkgo.By(fmt.Sprintf("Cluster: name=%s, namespace=%s, infrastructureRef=%v",
365+
cluster.Name, cluster.Namespace, cluster.Spec.InfrastructureRef))
366+
367+
// Print each AWSMachineTemplate for debugging
368+
for i, template := range awsMachineTemplateList.Items {
369+
ginkgo.By(fmt.Sprintf("AWSMachineTemplate[%d]: %+v", i, template))
370+
}
371+
363372
foundTemplateWithCapacity := false
364373
foundTemplateWithNodeInfo := false
365-
for _, template := range awsMachineTemplateList.Items {
374+
ginkgo.By(fmt.Sprintf("DEBUG: Found %d AWSMachineTemplates in namespace %s", len(awsMachineTemplateList.Items), namespace.Name))
375+
for i, template := range awsMachineTemplateList.Items {
376+
ginkgo.By(fmt.Sprintf("DEBUG: Template[%d] - Name: %s", i, template.Name))
377+
ginkgo.By(fmt.Sprintf("DEBUG: Template[%d] - Spec.Template.Spec.InstanceType: %s", i, template.Spec.Template.Spec.InstanceType))
378+
ginkgo.By(fmt.Sprintf("DEBUG: Template[%d] - Status.Capacity: %v", i, template.Status.Capacity))
379+
ginkgo.By(fmt.Sprintf("DEBUG: Template[%d] - Status.NodeInfo: %+v", i, template.Status.NodeInfo))
380+
ginkgo.By(fmt.Sprintf("DEBUG: Template[%d] - Status.Conditions: %+v", i, template.Status.Conditions))
381+
366382
if len(template.Status.Capacity) > 0 {
367383
foundTemplateWithCapacity = true
368384
ginkgo.By(fmt.Sprintf("AWSMachineTemplate %s has capacity populated: %v", template.Name, template.Status.Capacity))
369385
Expect(template.Status.Capacity).To(HaveKey(corev1.ResourceCPU), "Expected CPU to be set in capacity")
370386
Expect(template.Status.Capacity).To(HaveKey(corev1.ResourceMemory), "Expected Memory to be set in capacity")
387+
} else {
388+
ginkgo.By(fmt.Sprintf("WARNING: Template %s has NO capacity populated", template.Name))
371389
}
372390
if template.Status.NodeInfo != nil {
373391
foundTemplateWithNodeInfo = true
374392
ginkgo.By(fmt.Sprintf("AWSMachineTemplate %s has nodeInfo populated: %v", template.Name, template.Status.NodeInfo))
375393
// Verify architecture is set (should be either amd64 or arm64 for AWS)
394+
ginkgo.By(fmt.Sprintf("DEBUG: Template %s - NodeInfo.Architecture: '%s' (len: %d)", template.Name, template.Status.NodeInfo.Architecture, len(template.Status.NodeInfo.Architecture)))
395+
ginkgo.By(fmt.Sprintf("DEBUG: Template %s - NodeInfo.OperatingSystem: '%s' (len: %d)", template.Name, template.Status.NodeInfo.OperatingSystem, len(template.Status.NodeInfo.OperatingSystem)))
396+
397+
// Verify architecture is set (should be either amd64 or arm64 for AWS)
398+
if len(template.Status.NodeInfo.Architecture) == 0 {
399+
ginkgo.By(fmt.Sprintf("ERROR: Template %s has empty Architecture field in NodeInfo", template.Name))
400+
}
401+
Expect(template.Status.NodeInfo.Architecture).ToNot(BeEmpty(), fmt.Sprintf("Expected architecture to be set in nodeInfo for template %s", template.Name))
402+
Expect(string(template.Status.NodeInfo.Architecture)).To(MatchRegexp("^(amd64|arm64)$"), fmt.Sprintf("Expected architecture to be amd64 or arm64 for template %s", template.Name))
403+
// Verify operating system is set (if available)
404+
// Note: OperatingSystem may be empty when:
405+
// 1. No explicit AMI ID is specified
406+
// 2. Kubernetes version cannot be determined (MachineDeployment not yet created)
407+
// 3. AMI lookup fails
408+
// In these cases, the controller falls back to instance type architecture only.
409+
// See: controllers/awsmachinetemplate_controller.go:319-323 (Strategy 3)
410+
if len(template.Status.NodeInfo.OperatingSystem) == 0 {
411+
ginkgo.By(fmt.Sprintf("INFO: Template %s has empty OperatingSystem (this is acceptable in fallback scenarios)", template.Name))
412+
ginkgo.By(fmt.Sprintf("DEBUG: Template %s full status: %+v", template.Name, template.Status))
413+
} else {
414+
ginkgo.By(fmt.Sprintf("INFO: Template %s has OperatingSystem set to: %s", template.Name, template.Status.NodeInfo.OperatingSystem))
415+
// If OperatingSystem is set, verify it's a valid value
416+
Expect(string(template.Status.NodeInfo.OperatingSystem)).To(MatchRegexp("^(linux|windows)$"), fmt.Sprintf("Expected operatingSystem to be linux or windows for template %s", template.Name))
417+
}
376418
Expect(template.Status.NodeInfo.Architecture).ToNot(BeEmpty(), "Expected architecture to be set in nodeInfo")
377419
Expect(string(template.Status.NodeInfo.Architecture)).To(MatchRegexp("^(amd64|arm64)$"), "Expected architecture to be amd64 or arm64")
378420
// Verify operating system is set

0 commit comments

Comments
 (0)