Merge pull request #118 from tpapaioa/flashmessages_msg_count_fix

[RFR] Add NoSuchElementException handling back into FlashMessages.
RedHatQE · Mar 5, 2020 · 4b32fa7 · 4b32fa7
2 parents 9730981 + 6274acf
commit 4b32fa7
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/widgetastic_patternfly/__init__.py b/src/widgetastic_patternfly/__init__.py
@@ -889,7 +889,7 @@ def select_by_visible_text(self, *items):
                 self.logger.info('selecting by visible text: %r', item)
                 try:
                     self.browser.click(self.BY_VISIBLE_TEXT.format(quote(item)),
-                                       parent=self,  force_scroll=True)
+                                       parent=self, force_scroll=True)
                 except NoSuchElementException:
                     try:
                         # Added this as for some views(some tags pages) dropdown is separated from
@@ -2818,7 +2818,12 @@ def __getitem__(self, msg_filter):
 
     @property
     def msg_count(self):
-        return len(self.browser.elements(self.MSG_LOCATOR, parent=self))
+        c = 0
+        try:
+            c = len(self.browser.elements(self.MSG_LOCATOR, parent=self))
+        except NoSuchElementException:
+            pass
+        return c
 
     def messages(self, **msg_filter):
         """Return a generator for all notifications matching the msg_filter.