chore: mitigate ubuntu issue causing flakey tests on new instance typ…

…es (#4435)
aws · Aug 15, 2023 · a0231ac · a0231ac
1 parent f04fd9f
commit a0231ac
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/test/suites/integration/ami_test.go b/test/suites/integration/ami_test.go
@@ -189,6 +189,20 @@ var _ = Describe("AMI", func() {
 			}})
 			provisioner := test.Provisioner(test.ProvisionerOptions{
 				ProviderRef: &v1alpha5.MachineTemplateRef{Name: provider.Name},
+				// TODO: remove requirements after Ubuntu fixes bootstrap script issue w/
+				// new instance types not included in the max-pods.txt file.
+				Requirements: []v1.NodeSelectorRequirement{
+					{
+						Key:      v1alpha1.LabelInstanceGeneration,
+						Operator: v1.NodeSelectorOpLt,
+						Values:   []string{"7"},
+					},
+					{
+						Key:      v1.LabelOSStable,
+						Operator: v1.NodeSelectorOpIn,
+						Values:   []string{string(v1.Linux)},
+					},
+				},
 			})
 			pod := test.Pod()
 			env.ExpectCreated(provider, provisioner, pod)