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

26 lines
1.1 KiB
Python

import driver
from gui.elements.object import QObject
from gui.elements.text_edit import TextEdit
from gui.objects_map import names
class TokenSelectorPopup(QObject):
def __init__(self):
super().__init__(names.tokenSelectorPanel_TokenSelectorNew)
self.token_selector_panel = QObject(names.tokenSelectorPanel_TokenSelectorNew)
self.tab_bar = QObject(names.tokensTabBar_StatusTabBar)
self.assets_tab = QObject(names.tokenSelectorPanel_AssetsTab)
self.collectibles_tab = QObject(names.tokenSelectorPanel_CollectiblesTab)
self.asset_list_item = QObject(names.tokenSelectorAssetDelegate_template)
self.amount_to_send_field = TextEdit(names.amountInput_TextEdit)
def select_asset_from_list(self, asset_name: str):
self.assets_tab.click()
assets_list = driver.findAllObjects(self.asset_list_item.real_name)
assert assets_list, f'Assets are not displayed'
for item in assets_list:
if getattr(item, 'tokensKey', '') == asset_name:
QObject(item).click()
break
return self