adding ansible role as per given input (SatelliteQE#1542)

airgun/entities/host_new.py

@@ -542,18 +542,16 @@ def bulk_override_repo_sets(self, entity_name, repo_type, action):
         view.flash.assert_no_error()
         view.flash.dismiss()
 
-    def add_single_ansible_role(self, entity_name):
+    def add_single_ansible_role(self, entity_name, role=None):
         view = self.navigate_to(self, 'NewDetails', entity_name=entity_name)
         view.wait_displayed()
         self.browser.plugin.ensure_page_safe()
         wait_for(lambda: view.ansible.roles.edit.is_displayed, timeout=5)
         view.ansible.roles.edit.click()
         wait_for(lambda: EditAnsibleRolesView(self.browser).addAnsibleRole.is_displayed, timeout=10)
         edit_view = EditAnsibleRolesView(self.browser)
-        actions = [edit_view.addAnsibleRole, edit_view.selectRoles, edit_view.confirm]
-        for action in actions:
-            wait_for(lambda: edit_view.is_displayed, timeout=5)
-            action.click()
+        edit_view.addAnsibleRole.select_and_move([role])
+        edit_view.confirm.click()
 
     def get_ansible_roles(self, entity_name):
         view = self.navigate_to(self, 'NewDetails', entity_name=entity_name)

airgun/views/host_new.py

@@ -5,6 +5,7 @@
 from widgetastic_patternfly4 import (
     Button,
     Dropdown,
+    DualListSelector,
     Pagination as PF4Pagination,
     Select,
     Tab,
@@ -800,9 +801,7 @@ def read(self):
 class EditAnsibleRolesView(View):
     """Edit Ansible Roles Modal"""
 
-    addAnsibleRole = Text(
-        './/span[contains(text(),"RedHatInsights.insights-client") or contains(text(),"theforeman.foreman_scap_client")]'
-    )
+    addAnsibleRole = DualListSelector('//div[@class = "pf-c-dual-list-selector"]')
     confirm = Button(locator='.//button[@aria-label="submit ansible roles"]')
     hostAssignedAnsibleRoles = Text(
         './/button[@class="pf-c-dual-list-selector__item"]/span[1]//span[2]'