From c04fcef472dda19b432c469ef8f88ebbf384fc68 Mon Sep 17 00:00:00 2001
From: Kim Do Young <difbfl4750@gmail.com>
Date: Wed, 25 Sep 2024 19:03:33 +0900
Subject: [PATCH] Close modal and update job list after creating Training Job

Change-Id: Iea60abaffb2e8c980eaf2bb2dded6e9afada972e
Signed-off-by: Kim Do Young <difbfl4750@gmail.com>
---
 src/components/home/create/CreateTrainingJob.js         | 1 +
 src/components/home/form/CreateOrEditTrainingJobForm.js | 2 ++
 src/components/home/status/StatusPageRows.js            | 2 +-
 3 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/components/home/create/CreateTrainingJob.js b/src/components/home/create/CreateTrainingJob.js
index 789e35e..ef4de0f 100644
--- a/src/components/home/create/CreateTrainingJob.js
+++ b/src/components/home/create/CreateTrainingJob.js
@@ -87,6 +87,7 @@ class CreateTrainingJob extends React.Component {
         <CreateOrEditTrainingJobForm
           isCreateTrainingJobForm={true}
           logger={this.logger}
+          onHideCreatePopup={this.props.onHideCreatePopup}
           fetchTrainingJobs={this.fetchTrainingJobs}
         ></CreateOrEditTrainingJobForm>
       </>
diff --git a/src/components/home/form/CreateOrEditTrainingJobForm.js b/src/components/home/form/CreateOrEditTrainingJobForm.js
index 121736b..585e754 100644
--- a/src/components/home/form/CreateOrEditTrainingJobForm.js
+++ b/src/components/home/form/CreateOrEditTrainingJobForm.js
@@ -430,6 +430,8 @@ class CreateTrainingJob extends React.Component {
         if (res.status === 200) {
           alert('Training Job created and training initiated');
           this.resetFrom();
+          this.props.onHideCreatePopup();
+          this.props.fetchTrainingJobs();
         }
       })
       .catch(error => {
diff --git a/src/components/home/status/StatusPageRows.js b/src/components/home/status/StatusPageRows.js
index ae5d1c9..eb361de 100644
--- a/src/components/home/status/StatusPageRows.js
+++ b/src/components/home/status/StatusPageRows.js
@@ -257,7 +257,7 @@ const StatusPageRows = props => {
         </tbody>
       </BTable>
       <Popup show={createPopup} onHide={closeCreatePopup} title='Create Training Job' size='lg'>
-        <CreateTrainingJob logger={logger} fetchTrainingJobs={fetchTrainingJobs} />
+        <CreateTrainingJob logger={logger} onHideCreatePopup={closeCreatePopup} fetchTrainingJobs={fetchTrainingJobs} />
       </Popup>
       <Popup show={editPopup} onHide={closeEditPopup} title='Edit usecase'>
         <CreateOrEditTrainingJobForm
-- 
2.16.6