From 6a3216443829381da2692ddf3fa88329096a7855 Mon Sep 17 00:00:00 2001
From: mabashian <mabashia@redhat.com>
Date: Mon, 8 Mar 2021 16:11:22 -0500
Subject: [PATCH] Adds support for ouiaId's on copy buttons

---
 awx/ui_next/src/components/CopyButton/CopyButton.jsx          | 4 ++++
 .../NotificationTemplateList/NotificationTemplateList.jsx     | 2 +-
 .../NotificationTemplateList/NotificationTemplateListItem.jsx | 3 +++
 3 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/awx/ui_next/src/components/CopyButton/CopyButton.jsx b/awx/ui_next/src/components/CopyButton/CopyButton.jsx
index 2856c69c0c..1f5dd9cff4 100644
--- a/awx/ui_next/src/components/CopyButton/CopyButton.jsx
+++ b/awx/ui_next/src/components/CopyButton/CopyButton.jsx
@@ -17,6 +17,7 @@ function CopyButton({
   onCopyFinish,
   errorMessage,
   i18n,
+  ouiaId,
 }) {
   const { isLoading, error: copyError, request: copyItemToAPI } = useRequest(
     copyItem
@@ -35,6 +36,7 @@ function CopyButton({
     <>
       <Button
         id={id}
+        ouiaId={ouiaId}
         isDisabled={isLoading || isDisabled}
         aria-label={i18n._(t`Copy`)}
         variant="plain"
@@ -62,10 +64,12 @@ CopyButton.propTypes = {
   onCopyFinish: PropTypes.func.isRequired,
   errorMessage: PropTypes.string.isRequired,
   isDisabled: PropTypes.bool,
+  ouiaId: PropTypes.string,
 };
 
 CopyButton.defaultProps = {
   isDisabled: false,
+  ouiaId: null,
 };
 
 export default withI18n()(CopyButton);
diff --git a/awx/ui_next/src/screens/NotificationTemplate/NotificationTemplateList/NotificationTemplateList.jsx b/awx/ui_next/src/screens/NotificationTemplate/NotificationTemplateList/NotificationTemplateList.jsx
index 2e8b9a0c80..ea06a49c03 100644
--- a/awx/ui_next/src/screens/NotificationTemplate/NotificationTemplateList/NotificationTemplateList.jsx
+++ b/awx/ui_next/src/screens/NotificationTemplate/NotificationTemplateList/NotificationTemplateList.jsx
@@ -227,7 +227,7 @@ function NotificationTemplatesList({ i18n }) {
         {i18n._(t`Failed to delete one or more notification template.`)}
         <ErrorDetail error={deletionError} />
       </AlertModal>
-      <AlertGroup isToast>
+      <AlertGroup ouiaId="notification-template-alerts" isToast>
         {testToasts
           .filter(notification => notification.status !== 'pending')
           .map(notification => (
diff --git a/awx/ui_next/src/screens/NotificationTemplate/NotificationTemplateList/NotificationTemplateListItem.jsx b/awx/ui_next/src/screens/NotificationTemplate/NotificationTemplateList/NotificationTemplateListItem.jsx
index 71d32e9821..025e4b7465 100644
--- a/awx/ui_next/src/screens/NotificationTemplate/NotificationTemplateList/NotificationTemplateListItem.jsx
+++ b/awx/ui_next/src/screens/NotificationTemplate/NotificationTemplateList/NotificationTemplateListItem.jsx
@@ -119,6 +119,7 @@ function NotificationTemplateListItem({
         <ActionsTd dataLabel={i18n._(t`Actions`)}>
           <ActionItem visible tooltip={i18n._(t`Test notification`)}>
             <Button
+              ouiaId={`notification-test-button-${template.id}`}
               aria-label={i18n._(t`Test Notification`)}
               variant="plain"
               onClick={sendTestNotification}
@@ -132,6 +133,7 @@ function NotificationTemplateListItem({
             tooltip={i18n._(t`Edit`)}
           >
             <Button
+              ouiaId={`notification-edit-button-${template.id}`}
               aria-label={i18n._(t`Edit Notification Template`)}
               variant="plain"
               component={Link}
@@ -145,6 +147,7 @@ function NotificationTemplateListItem({
             tooltip={i18n._(t`Copy Notification Template`)}
           >
             <CopyButton
+              ouiaId={`notification-copy-button-${template.id}`}
               copyItem={copyTemplate}
               isCopyDisabled={isCopyDisabled}
               onCopyStart={handleCopyStart}