From eddee456b33d1b8ce37c17c1ddf57107ed380ba3 Mon Sep 17 00:00:00 2001
From: Apurva Bakshi <bakshiapurva93@gmail.com>
Date: Wed, 27 Nov 2019 13:24:22 -0500
Subject: [PATCH] Add dataCy attribute to select-user-role and select-team-role
 buttons

---
 awx/ui_next/src/components/AddRole/AddResourceRole.jsx | 2 ++
 awx/ui_next/src/components/AddRole/SelectableCard.jsx  | 3 ++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/awx/ui_next/src/components/AddRole/AddResourceRole.jsx b/awx/ui_next/src/components/AddRole/AddResourceRole.jsx
index 3add8286c9..596b3bbe97 100644
--- a/awx/ui_next/src/components/AddRole/AddResourceRole.jsx
+++ b/awx/ui_next/src/components/AddRole/AddResourceRole.jsx
@@ -187,11 +187,13 @@ class AddResourceRole extends React.Component {
             <SelectableCard
               isSelected={selectedResource === 'users'}
               label={i18n._(t`Users`)}
+              dataCy="add-role-users"
               onClick={() => this.handleResourceSelect('users')}
             />
             <SelectableCard
               isSelected={selectedResource === 'teams'}
               label={i18n._(t`Teams`)}
+              dataCy="add-role-teams"
               onClick={() => this.handleResourceSelect('teams')}
             />
           </div>
diff --git a/awx/ui_next/src/components/AddRole/SelectableCard.jsx b/awx/ui_next/src/components/AddRole/SelectableCard.jsx
index cebc795a0e..475af3d2ce 100644
--- a/awx/ui_next/src/components/AddRole/SelectableCard.jsx
+++ b/awx/ui_next/src/components/AddRole/SelectableCard.jsx
@@ -33,7 +33,7 @@ const Label = styled.div`
 
 class SelectableCard extends Component {
   render() {
-    const { label, onClick, isSelected } = this.props;
+    const { label, onClick, isSelected, dataCy } = this.props;
 
     return (
       <SelectableItem
@@ -41,6 +41,7 @@ class SelectableCard extends Component {
         onKeyPress={onClick}
         role="button"
         tabIndex="0"
+        data-cy={dataCy}
         isSelected={isSelected}
       >
         <Indicator isSelected={isSelected} />