From f189cac2870c190e872a6ea3734a04a763c2a20d Mon Sep 17 00:00:00 2001
From: Ilank <63646693+ilan7empest@users.noreply.github.com>
Date: Mon, 4 Dec 2023 16:05:49 +0700
Subject: [PATCH] Fix [UI] add the 'data-testid' for 'Add volume' and 'Add node
 selector' buttons (#2087)

---
 .../JobWizardSteps/JobWizardResources/JobWizardResources.js    | 1 +
 src/elements/FormVolumesTable/FormVolumesTable.js              | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/components/JobWizard/JobWizardSteps/JobWizardResources/JobWizardResources.js b/src/components/JobWizard/JobWizardSteps/JobWizardResources/JobWizardResources.js
index defc3d778c..c0b8cecb77 100644
--- a/src/components/JobWizard/JobWizardSteps/JobWizardResources/JobWizardResources.js
+++ b/src/components/JobWizard/JobWizardSteps/JobWizardResources/JobWizardResources.js
@@ -66,6 +66,7 @@ const JobWizardResources = ({ formState, frontendSpec, stepIsActive }) => {
       <div className="form-row form-table-title">Node selection</div>
       <div className="form-row">
         <FormKeyValueTable
+          actionButtonId="add-node-selector"
           addNewItemLabel="Add node selector"
           exitEditModeTriggerItem={stepIsActive}
           fieldsPath={`${RESOURCES_STEP}.nodeSelectorTable`}
diff --git a/src/elements/FormVolumesTable/FormVolumesTable.js b/src/elements/FormVolumesTable/FormVolumesTable.js
index 7cc3ebe8fc..cf5644ee3b 100644
--- a/src/elements/FormVolumesTable/FormVolumesTable.js
+++ b/src/elements/FormVolumesTable/FormVolumesTable.js
@@ -85,6 +85,7 @@ const FormVolumesTable = ({ disabled, exitEditModeTriggerItem, fieldsPath, formS
                 hidden={editingItem?.ui?.isNew}
                 fields={fields}
                 fieldsPath={fieldsPath}
+                id="add-volume"
                 label="Add volume"
                 onClick={(...addRowArgs) =>
                   addNewRow(...addRowArgs, {
@@ -111,7 +112,7 @@ const FormVolumesTable = ({ disabled, exitEditModeTriggerItem, fieldsPath, formS
 
 FormVolumesTable.defaultProps = {
   disabled: false,
-  exitEditModeTriggerItem: null,
+  exitEditModeTriggerItem: null
 }
 
 FormVolumesTable.propTypes = {