Raise exception when delete modal isn't showing

ColeHiggins2 · Nov 27, 2023 · 36b6255 · 36b6255
1 parent 89bf8d4
commit 36b6255
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/airgun/entities/all_hosts.py b/airgun/entities/all_hosts.py
@@ -1,3 +1,5 @@
+from widgetastic.exceptions import NoSuchElementException
+
 from airgun.entities.base import BaseEntity
 from airgun.navigation import NavigateStep, navigator
 from airgun.utils import retry_navigation
@@ -28,6 +30,8 @@ def delete(self, host_name):
         delete_modal = HostDeleteDialog(self.browser)
         if delete_modal.is_displayed:
             delete_modal.confirm_delete.click()
+        else:
+            raise NoSuchElementException("Delete Modal was not displayed.")
         view = self.navigate_to(self, 'All')
         self.browser.plugin.ensure_page_safe(timeout='5s')
         view.wait_displayed()