From b6853189224308be150ce70dc986ca8ec5d6bf98 Mon Sep 17 00:00:00 2001
From: Nancy Butler <42977925+mantis-toboggan-md@users.noreply.github.com>
Date: Wed, 17 Jul 2024 10:56:17 -0700
Subject: [PATCH] fix EKS project network isolation field and add unit tests

---
 pkg/eks/components/CruEKS.vue               |  3 ++-
 pkg/eks/components/__tests__/CruEKS.test.ts | 26 +++++++++++++++++++++
 2 files changed, 28 insertions(+), 1 deletion(-)

diff --git a/pkg/eks/components/CruEKS.vue b/pkg/eks/components/CruEKS.vue
index d1d3e5e980e..d8c07f8213b 100644
--- a/pkg/eks/components/CruEKS.vue
+++ b/pkg/eks/components/CruEKS.vue
@@ -676,7 +676,7 @@ export default defineComponent({
       >
         <Config
           :kubernetes-version.sync="config.kubernetesVersion"
-          :enable-network-policy.sync="config.enableNetworkPolicy"
+          :enable-network-policy.sync="normanCluster.enableNetworkPolicy"
           :ebs-c-s-i-driver.sync="config.ebsCSIDriver"
           :service-role.sync="config.serviceRole"
           :kms-key.sync="config.kmsKey"
@@ -686,6 +686,7 @@ export default defineComponent({
           :eks-roles="eksRoles"
           :loading-iam="loadingIam"
           :original-version="originalVersion"
+          data-testid="eks-config-section"
           @error="e=>errors.push(e)"
         />
       </Accordion>
diff --git a/pkg/eks/components/__tests__/CruEKS.test.ts b/pkg/eks/components/__tests__/CruEKS.test.ts
index 8b7f31b3d17..d4a8eaadbeb 100644
--- a/pkg/eks/components/__tests__/CruEKS.test.ts
+++ b/pkg/eks/components/__tests__/CruEKS.test.ts
@@ -146,4 +146,30 @@ describe('eKS provisioning form', () => {
 
     expect(wrapper.vm.nodeGroups.filter((group: EKSNodeGroup) => group.version === '1.24')).toHaveLength(1);
   });
+
+  it('should configure enable network policy at the cluster level not within eksConfig', async() => {
+    const wrapper = shallowMount(CruEKS, {
+      propsData: { value: {}, mode: 'edit' },
+      ...requiredSetup()
+    });
+
+    await setCredential(wrapper);
+
+    const configComponent = wrapper.find('[data-testid="eks-config-section"]');
+
+    configComponent.vm.$emit('update:enableNetworkPolicy', true);
+
+    await wrapper.vm.$nextTick();
+    expect(wrapper.vm.normanCluster.enableNetworkPolicy).toBe(true);
+
+    configComponent.vm.$emit('update:enableNetworkPolicy', false);
+
+    await wrapper.vm.$nextTick();
+    expect(wrapper.vm.normanCluster.enableNetworkPolicy).toBe(false);
+
+    wrapper.setData({ normanCluster: { ...wrapper.vm.normanCluster, enableNetworkPolicy: true } });
+    await wrapper.vm.$nextTick();
+
+    expect(configComponent.props().enableNetworkPolicy).toBe(true);
+  });
 });