import allure

import driver
from gui.elements.object import QObject
from gui.objects_map import names


class ToastMessage(QObject):

    def __init__(self):
        super(ToastMessage, self).__init__(names.ephemeral_Notification_List)
        self._toast_message = QObject(names.ephemeralNotificationList_StatusToastMessage)

    @property
    @allure.step('Get toast messages')
    def get_toast_messages(self):
        messages = []
        for obj in driver.findAllObjects(self._toast_message.real_name):
            messages.append(str(obj.primaryText))
        if len(messages) == 0:
            raise LookupError(
                'Toast messages were not found')
        else:
            return messages