def status_bar_should_have_text(self, text):
theia_status_bar = self.find_it(UI.get_theia_statusbar_locator())
elements_with_commands = self.find_them(UI.get_status_elements(),
parent=theia_status_bar)
for elements_with_command in elements_with_commands:
if elements_with_command.text == text:
highlight(elements_with_command, effect_time=1)
return
raise NoSuchElementException
def get_notifications_from_control_center(self):
status_bar = self.find_it(UI.get_theia_statusbar_locator())
highlight(status_bar)
area_right = self.find_it(UI.get_right_status_bar_area(), parent=status_bar)
highlight(area_right)
status_elements = self.find_them(UI.get_status_elements(), parent=area_right)
show_button = None
for status_element in status_elements:
highlight(status_element)
try:
int_text = int(status_element.text)
show_button = status_element
break
except ValueError:
continue
except Exception:
raise GeneralException(self.get_driver(), call_from=self.get_notifications_from_control_center.__name__)
notifications = list()
if show_button is not None:
highlight(show_button)
notification_center = self.find_it(UI.get_theia_notification_center_locator())
highlight(notification_center)
if constants.THEIA_CLOSED in notification_center.get_attribute(constants.TYPE_CLASS):
self.click_me(show_button)
sleep(1)
while True:
if constants.THEIA_CLOSED in notification_center.get_attribute(constants.TYPE_CLASS):
sleep(1)
continue
break
highlight(notification_center)
notifications = self.find_them(UI.get_theia_notifications_locator(), parent=notification_center)
return notifications
