From 098ec639441350ac2bf82e87404300e6d461995e Mon Sep 17 00:00:00 2001
From: Jake McDermott <yo@jakemcdermott.me>
Date: Mon, 15 Mar 2021 12:50:33 -0400
Subject: [PATCH] Add container group flag to add/edit data

---
 .../InstanceGroup/ContainerGroupAdd/ContainerGroupAdd.jsx       | 1 +
 .../InstanceGroup/ContainerGroupAdd/ContainerGroupAdd.test.jsx  | 2 ++
 .../InstanceGroup/ContainerGroupEdit/ContainerGroupEdit.jsx     | 1 +
 .../ContainerGroupEdit/ContainerGroupEdit.test.jsx              | 1 +
 4 files changed, 5 insertions(+)

diff --git a/awx/ui_next/src/screens/InstanceGroup/ContainerGroupAdd/ContainerGroupAdd.jsx b/awx/ui_next/src/screens/InstanceGroup/ContainerGroupAdd/ContainerGroupAdd.jsx
index 6dffae83d3..7549a9d1cc 100644
--- a/awx/ui_next/src/screens/InstanceGroup/ContainerGroupAdd/ContainerGroupAdd.jsx
+++ b/awx/ui_next/src/screens/InstanceGroup/ContainerGroupAdd/ContainerGroupAdd.jsx
@@ -33,6 +33,7 @@ function ContainerGroupAdd() {
         pod_spec_override: values.override
           ? getPodSpecValue(values.pod_spec_override)
           : null,
+        is_container_group: true,
       });
       history.push(`/instance_groups/container_group/${response.id}/details`);
     } catch (error) {
diff --git a/awx/ui_next/src/screens/InstanceGroup/ContainerGroupAdd/ContainerGroupAdd.test.jsx b/awx/ui_next/src/screens/InstanceGroup/ContainerGroupAdd/ContainerGroupAdd.test.jsx
index b8bbb301ab..3b47a19d29 100644
--- a/awx/ui_next/src/screens/InstanceGroup/ContainerGroupAdd/ContainerGroupAdd.test.jsx
+++ b/awx/ui_next/src/screens/InstanceGroup/ContainerGroupAdd/ContainerGroupAdd.test.jsx
@@ -26,6 +26,7 @@ const initialPodSpec = {
         },
       ],
     },
+    is_container_group: true,
   },
 };
 
@@ -80,6 +81,7 @@ describe('<ContainerGroupAdd/>', () => {
     expect(InstanceGroupsAPI.create).toHaveBeenCalledWith({
       ...instanceGroupCreateData,
       credential: 71,
+      is_container_group: true,
     });
     expect(wrapper.find('FormSubmitError').length).toBe(0);
     expect(history.location.pathname).toBe(
diff --git a/awx/ui_next/src/screens/InstanceGroup/ContainerGroupEdit/ContainerGroupEdit.jsx b/awx/ui_next/src/screens/InstanceGroup/ContainerGroupEdit/ContainerGroupEdit.jsx
index 9f4454c0a8..bd56246c18 100644
--- a/awx/ui_next/src/screens/InstanceGroup/ContainerGroupEdit/ContainerGroupEdit.jsx
+++ b/awx/ui_next/src/screens/InstanceGroup/ContainerGroupEdit/ContainerGroupEdit.jsx
@@ -39,6 +39,7 @@ function ContainerGroupEdit({ instanceGroup }) {
         name: values.name,
         credential: values.credential ? values.credential.id : null,
         pod_spec_override: values.override ? values.pod_spec_override : null,
+        is_container_group: true,
       });
       history.push(detailsIUrl);
     } catch (error) {
diff --git a/awx/ui_next/src/screens/InstanceGroup/ContainerGroupEdit/ContainerGroupEdit.test.jsx b/awx/ui_next/src/screens/InstanceGroup/ContainerGroupEdit/ContainerGroupEdit.test.jsx
index 860c6363c5..6d05e43472 100644
--- a/awx/ui_next/src/screens/InstanceGroup/ContainerGroupEdit/ContainerGroupEdit.test.jsx
+++ b/awx/ui_next/src/screens/InstanceGroup/ContainerGroupEdit/ContainerGroupEdit.test.jsx
@@ -147,6 +147,7 @@ describe('<ContainerGroupEdit/>', () => {
       ...updatedInstanceGroup,
       credential: 12,
       pod_spec_override: null,
+      is_container_group: true,
     });
     expect(history.location.pathname).toEqual(
       '/instance_groups/container_group/123/details'