status-desktop/test/e2e/gui/components/wallet/wallet_toast_message.py

24 lines
708 B
Python

import allure
import driver
from gui.elements.object import QObject
class WalletToastMessage(QObject):
def __init__(self):
super(WalletToastMessage, self).__init__('mainWallet_Ephemeral_Notification_List')
self._wallet_toast_message = QObject('ephemeralNotificationList_StatusToastMessage')
@property
@allure.step('Get toast messages')
def get_toast_messages(self):
messages = []
for obj in driver.findAllObjects(self._wallet_toast_message.real_name):
messages.append(str(obj.primaryText))
if len(messages) == 0:
raise LookupError(
'Toast messages were not found')
else:
return messages