added changes in host_new entities (#1347) (#1348)

airgun/entities/host_new.py

@@ -498,17 +498,18 @@ def get_ansible_roles_modal(self, entity_name):
  self.browser.plugin.ensure_page_safe()
  return view.table.read()
 
- def remove_ansible_role(self, entity_name):
+ def remove_single_ansible_role(self, entity_name):
  view = self.navigate_to(self, 'NewDetails', entity_name=entity_name)
  view.wait_displayed()
  self.browser.plugin.ensure_page_safe()
  view.ansible.roles.edit.click()
  wait_for(lambda: view.ansible.roles.edit.click(), timeout=5)
- edit_role = EditAnsibleRolesView(self.browser)
- actions = [edit_role.hostAssignedAnsibleRoles, edit_role.unselectRoles, edit_role.confirm]
+ edit_view = EditAnsibleRolesView(self.browser)
+ edit_view.wait_displayed()
+ actions = [edit_view.hostAssignedAnsibleRoles, edit_view.unselectRoles, edit_view.confirm]
  for action in actions:
- wait_for(lambda: edit_role.is_displayed, timeout=5)
  action.click()
+ wait_for(lambda: view.ansible.roles.noRoleAssign.is_displayed, timeout=5)
 
  def enable_tracer(self, entity_name):
  view = self.navigate_to(self, 'NewDetails', entity_name=entity_name)