update other related test

aws · Jan 15, 2025 · 2d34f19 · 2d34f19
1 parent 8953cf9
commit 2d34f19
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/packages/aws-cdk-lib/aws-batch/test/managed-compute-environment.test.ts b/packages/aws-cdk-lib/aws-batch/test/managed-compute-environment.test.ts
@@ -1,3 +1,4 @@
+import { KubectlV31Layer } from '@aws-cdk/lambda-layer-kubectl-v31';
 import { capitalizePropertyNames } from './utils';
 import { Template } from '../../assertions';
 import * as ec2 from '../../aws-ec2';
@@ -79,6 +80,7 @@ describe.each([ManagedEc2EcsComputeEnvironment, ManagedEc2EksComputeEnvironment]
       kubernetesNamespace: 'cdk-test-namespace',
       eksCluster: new eks.Cluster(stack, 'eksTestCluster', {
         version: eks.KubernetesVersion.V1_24,
+        kubectlLayer: new KubectlV31Layer(stack, 'KubectlLayer'),
       }),
     };
     expectedProps = ComputeEnvironment === ManagedEc2EcsComputeEnvironment
@@ -740,6 +742,7 @@ describe('ManagedEc2EcsComputeEnvironment', () => {
       kubernetesNamespace: 'cdk-test-namespace',
       eksCluster: new eks.Cluster(stack, 'eksTestCluster', {
         version: eks.KubernetesVersion.V1_24,
+        kubectlLayer: new KubectlV31Layer(stack, 'KubectlLayer'),
       }),
     };
   });
@@ -909,6 +912,7 @@ describe('ManagedEc2EksComputeEnvironment', () => {
       kubernetesNamespace: 'cdk-test-namespace',
       eksCluster: new eks.Cluster(stack, 'eksTestCluster', {
         version: eks.KubernetesVersion.V1_24,
+        kubectlLayer: new KubectlV31Layer(stack, 'KubectlLayer'),
       }),
     };
   });

diff --git a/packages/aws-cdk-lib/aws-stepfunctions-tasks/test/eks/call.test.ts b/packages/aws-cdk-lib/aws-stepfunctions-tasks/test/eks/call.test.ts
@@ -1,3 +1,4 @@
+import { KubectlV31Layer } from '@aws-cdk/lambda-layer-kubectl-v31';
 import * as eks from '../../../aws-eks';
 import * as sfn from '../../../aws-stepfunctions';
 import { Stack } from '../../../core';
@@ -12,6 +13,7 @@ beforeEach(() => {
   cluster = new eks.Cluster(stack, 'Cluster', {
     version: eks.KubernetesVersion.V1_21,
     clusterName: 'eksCluster',
+    kubectlLayer: new KubectlV31Layer(stack, 'KubectlLayer'),
   });
 });
 
@@ -199,6 +201,7 @@ test('Task throws if cluster supplied does not have clusterEndpoint configured',
   const importedCluster = eks.Cluster.fromClusterAttributes(stack, 'InvalidCluster', {
     clusterName: 'importedCluster',
     clusterCertificateAuthorityData: 'clusterCertificateAuthorityData',
+    kubectlLayer: new KubectlV31Layer(stack, 'KubectlLayer'),
   });
   expect(() => {
     new EksCall(stack, 'Call', {
@@ -218,6 +221,7 @@ test('Task throws if cluster supplied does not have clusterCertificateAuthorityD
   const importedCluster = eks.Cluster.fromClusterAttributes(stack, 'InvalidCluster', {
     clusterName: 'importedCluster',
     clusterEndpoint: 'clusterEndpoint',
+    kubectlLayer: new KubectlV31Layer(stack, 'KubectlLayer'),
   });
   expect(() => {
     new EksCall(stack, 'Call', {

diff --git a/...ges/aws-cdk-lib/aws-stepfunctions-tasks/test/emrcontainers/create-virtual-cluster.test.ts b/...ges/aws-cdk-lib/aws-stepfunctions-tasks/test/emrcontainers/create-virtual-cluster.test.ts
@@ -1,3 +1,4 @@
+import { KubectlV31Layer } from '@aws-cdk/lambda-layer-kubectl-v31';
 import { Template } from '../../../assertions';
 import * as eks from '../../../aws-eks';
 import * as sfn from '../../../aws-stepfunctions';
@@ -104,6 +105,7 @@ describe('Invoke emr-containers CreateVirtualCluster with ', () => {
     // WHEN
     const eksCluster = new eks.Cluster(stack, 'EKS Cluster', {
       version: eks.KubernetesVersion.V1_20,
+      kubectlLayer: new KubectlV31Layer(stack, 'KubectlLayer'),
     });
 
     const task = new EmrContainersCreateVirtualCluster(stack, 'Task', {