status-desktop/test/e2e/gui/screens/wallet.py

315 lines
14 KiB
Python
Raw Normal View History

2024-03-22 07:49:16 +00:00
import time
import typing
import allure
import pyperclip
import configs
import constants.user
import driver
from driver.objects_access import walk_children
from gui.components.context_menu import ContextMenu
from gui.components.wallet.add_saved_address_popup import AddressPopup, EditSavedAddressPopup
from gui.components.wallet.asset_context_menu_popup import AssetContextMenuPopup
from gui.components.wallet.bridge_popup import BridgePopup
from gui.components.wallet.confirmation_popup import ConfirmationPopup
from gui.components.wallet.receive_popup import ReceivePopup
from gui.components.wallet.remove_wallet_account_popup import RemoveWalletAccountPopup
2023-12-05 13:23:02 +00:00
from gui.components.wallet.send_popup import SendPopup
from gui.components.wallet.wallet_account_popups import AccountPopup
from gui.elements.button import Button
2024-03-22 07:49:16 +00:00
from gui.elements.list import List
from gui.elements.object import QObject
from gui.elements.text_label import TextLabel
from gui.objects_map import wallet_names, settings_names, names
class WalletScreen(QObject):
def __init__(self):
super().__init__(wallet_names.mainWindow_WalletLayout)
2023-12-05 13:23:02 +00:00
self.left_panel: LeftPanel = LeftPanel()
class LeftPanel(QObject):
def __init__(self):
super(LeftPanel, self).__init__(wallet_names.mainWallet_LeftTab)
self._saved_addresses_button = Button(wallet_names.mainWallet_Saved_Addresses_Button)
self._wallet_account_item = QObject(wallet_names.walletAccount_StatusListItem)
self._add_account_button = Button(wallet_names.mainWallet_Add_Account_Button)
self._all_accounts_button = Button(wallet_names.mainWallet_All_Accounts_Button)
self._all_accounts_balance = TextLabel(wallet_names.mainWallet_All_Accounts_Balance)
2023-12-05 13:23:02 +00:00
@allure.step('Get total balance visibility state')
def is_total_balance_visible(self) -> bool:
return self._all_accounts_balance.is_visible
@property
@allure.step('Get all accounts from list')
def accounts(self) -> typing.List[constants.user.account_list_item]:
if 'title' in self._wallet_account_item.real_name.keys():
del self._wallet_account_item.real_name['title']
accounts = []
for account_item in driver.findAllObjects(self._wallet_account_item.real_name):
try:
name = str(account_item.title)
# TODO: to fix properly with account data class implementation
# color = str(account_item.asset.color.name).lower()
# emoji = ''
# for child in walk_children(account_item):
# if hasattr(child, 'emojiId'):
# emoji = str(child.emojiId)
# break
accounts.append(constants.user.account_list_item(name, None, None))
except (AttributeError, RuntimeError):
continue
return accounts
@allure.step('Get total balance value from All accounts')
def get_total_balance_value(self):
return self._all_accounts_balance.text
@allure.step('Choose saved addresses on left wallet panel')
def open_saved_addresses(self) -> 'SavedAddressesView':
self._saved_addresses_button.click()
return SavedAddressesView().wait_until_appears()
@allure.step('Select account from list')
def select_account(self, account_name: str) -> 'WalletAccountView':
2024-09-03 14:49:42 +00:00
account_items = self.accounts
existing_accounts_names = [account.name for account in account_items]
if account_name in existing_accounts_names:
self._wallet_account_item.real_name['title'] = account_name
time.sleep(0.5)
self._wallet_account_item.click()
return WalletAccountView().wait_until_appears()
@allure.step('Open context menu from left wallet panel')
def _open_context_menu(self) -> ContextMenu:
2024-03-13 08:44:28 +00:00
super(LeftPanel, self).right_click()
return ContextMenu().wait_until_appears()
2024-08-29 13:25:59 +00:00
@allure.step('Open context menu for specific account')
def _open_context_menu_for_account(self, account_name: str) -> ContextMenu:
2024-08-29 13:25:59 +00:00
account_items = self.accounts
existing_accounts_names = [account.name for account in account_items]
if account_name in existing_accounts_names:
self._wallet_account_item.real_name['title'] = account_name
self._wallet_account_item.click()
time.sleep(0.5)
self._wallet_account_item.right_click()
return ContextMenu()
else:
raise LookupError(f'{account_name} is not present in {account_items}')
@allure.step("Select Hide/Include in total balance from context menu for account")
def hide_include_in_total_balance_from_context_menu(self, account_name: str):
2024-08-29 13:25:59 +00:00
self._open_context_menu_for_account(account_name).hide_include_in_total_balance.click()
@allure.step('Open account popup for editing from context menu')
def open_edit_account_popup_from_context_menu(self, account_name: str) -> AccountPopup:
2024-08-29 13:25:59 +00:00
context_menu = self._open_context_menu_for_account(account_name)
context_menu.edit_from_context.click()
return AccountPopup().verify_edit_account_popup_present()
@allure.step('Open account popup')
def open_add_account_popup(self, attempt: int = 2):
self._add_account_button.click()
try:
return AccountPopup().wait_until_appears()
except AssertionError as err:
if attempt:
self.open_add_account_popup(attempt - 1)
else:
raise err
@allure.step('Select add watched address from context menu')
def select_add_watched_address_from_context_menu(self) -> AccountPopup:
2024-08-29 13:25:59 +00:00
self._open_context_menu().add_watched_address_from_context.click()
return AccountPopup().wait_until_appears()
@allure.step('Delete account from the list from context menu')
def delete_account_from_context_menu(self, account_name: str, attempt: int = 2) -> RemoveWalletAccountPopup:
try:
2024-08-29 13:25:59 +00:00
self._open_context_menu_for_account(account_name).delete_from_context.click()
return RemoveWalletAccountPopup().wait_until_appears()
2023-12-19 13:12:24 +00:00
except Exception as ex:
if attempt:
return self.delete_account_from_context_menu(account_name, attempt - 1)
else:
2023-12-19 13:12:24 +00:00
raise ex
@allure.step('Copy address for the account in the context menu')
def copy_account_address_in_context_menu(self, account_name: str):
2024-08-29 13:25:59 +00:00
self._open_context_menu_for_account(account_name).copy_address_from_context.click()
return str(pyperclip.paste())
class SavedAddressesView(QObject):
def __init__(self):
super(SavedAddressesView, self).__init__(wallet_names.mainWindow_SavedAddressesView)
self._add_new_address_button = Button(wallet_names.mainWallet_Saved_Addresses_Add_Buttton)
self._address_list_item = QObject(wallet_names.savedAddressView_Delegate)
self._addresses_area = QObject(wallet_names.savedAddresses_area)
self._addresses_list_view = QObject(wallet_names.mainWallet_Saved_Addresses_List)
self._send_button = Button(wallet_names.send_StatusRoundButton)
self._open_menu_button = Button(wallet_names.savedAddressView_Delegate_menuButton)
self._saved_address_item = QObject(wallet_names.savedAddressView_Delegate)
@property
@allure.step('Get saved addresses wallet_names')
def address_names(self):
address_names = []
for child in walk_children(self._addresses_list_view.object):
if getattr(child, 'id', '') == 'savedAddressDelegate':
address_names.append(str(child.name))
return address_names
@allure.step('Get saved addresses list')
def get_saved_addresses_list(self):
addresses = [str(address.name) for address in driver.findAllObjects(self._saved_address_item.real_name)]
return addresses
@allure.step('Open add new address popup')
def open_add_saved_address_popup(self, attempt=2) -> 'AddressPopup':
self._add_new_address_button.click()
try:
return AddressPopup().wait_until_appears()
except AssertionError as err:
if attempt:
self.open_add_saved_address_popup(attempt - 1)
else:
raise err
@allure.step('Open edit address popup for saved address')
def open_edit_address_popup(self, name: str) -> 'EditSavedAddressPopup':
2024-08-29 13:25:59 +00:00
self.right_click(name).edit_saved_address_from_context.click()
return EditSavedAddressPopup()
@allure.step('Delete saved address from the list')
def delete_saved_address(self, address_name):
2024-08-29 13:25:59 +00:00
self.right_click(address_name).delete_saved_address_from_context.click()
assert ConfirmationPopup().get_confirmation_text().startswith('Are you sure you want to remove')
ConfirmationPopup().confirm()
@allure.step('Open context menu in saved address')
2024-03-13 08:44:28 +00:00
def right_click(self, name) -> ContextMenu:
self._open_menu_button.real_name['objectName'] = 'savedAddressView_Delegate_menuButton' + '_' + name
self._address_list_item.real_name['objectName'] = 'savedAddressView_Delegate' + '_' + name
self._address_list_item.hover()
self._open_menu_button.click()
return ContextMenu().wait_until_appears()
class WalletAccountView(QObject):
def __init__(self):
super(WalletAccountView, self).__init__(settings_names.mainWindow_StatusSectionLayout_ContentItem)
self._account_name_text_label = TextLabel(wallet_names.mainWallet_Account_Name)
self._addresses_panel = QObject(names.mainWallet_Address_Panel)
self._send_button = Button(wallet_names.mainWindow_Send_Button)
self._receive_button = Button(wallet_names.mainWindow_Receive_Button)
self._bridge_button = Button(wallet_names.mainWindow_Bridge_Button)
self._filter_button = Button(wallet_names.filterButton_StatusFlatButton)
self._assets_combobox = List(wallet_names.cmbTokenOrder_SortOrderComboBox)
self._assets_tab_button = Button(wallet_names.rightSideWalletTabBar_Assets_StatusTabButton)
self._collectibles_tab_button = Button(wallet_names.rightSideWalletTabBar_Collectibles_StatusTabButton)
self._asset_item_delegate = QObject(wallet_names.itemDelegate)
self._asset_item = QObject(wallet_names.assetView_TokenListItem_TokenDelegate)
self._arrow_icon = QObject(wallet_names.arrow_icon_StatusIcon)
@property
@allure.step('Get name of account')
def name(self) -> str:
return self._account_name_text_label.text
@property
@allure.step('Get address of account')
def address(self) -> str:
return str(self._addresses_panel.object.value)
@allure.step('Wait until appears {0}')
def wait_until_appears(self, timeout_msec: int = configs.timeouts.UI_LOAD_TIMEOUT_MSEC):
self._account_name_text_label.wait_until_appears(timeout_msec)
return self
2023-12-05 13:23:02 +00:00
@allure.step('Open send popup')
def open_send_popup(self) -> SendPopup:
self._send_button.click()
return SendPopup().wait_until_appears()
2024-03-22 07:49:16 +00:00
@allure.step('Open bridge popup')
def open_bridge_popup(self) -> BridgePopup:
self._bridge_button.click()
return BridgePopup().wait_until_appears()
@allure.step('Open receive popup')
def open_receive_popup(self) -> ReceivePopup:
self._receive_button.click()
return ReceivePopup().wait_until_appears()
2024-03-22 07:49:16 +00:00
@allure.step('Open assets tab')
def open_assets_tab(self):
self._assets_tab_button.click()
return self
@allure.step('Open collectibles tab')
def open_collectibles_tab(self):
self._collectibles_tab_button.click()
return self
@allure.step('Click filter button')
def click_filter_button(self):
self._filter_button.click()
return self
@allure.step('Get value from combobox')
def get_combobox_value(self) -> str:
return str(self._assets_combobox.object.displayText)
@allure.step('Choose sort by value')
def choose_sort_by_value(self, sort_by_value: str):
self._assets_combobox.click()
driver.mouseClick(self.get_sort_by_item_object(sort_by_value))
@allure.step('Get sort by item')
def get_sort_by_item_object(self, sort_by_value: str):
for item in driver.findAllObjects(self._asset_item_delegate.real_name):
if getattr(item, 'text', '') == sort_by_value:
return item
@allure.step('Click the arrow button')
def click_arrow_button(self, arrow_name: str, occurrence: int):
self._assets_combobox.click()
self._arrow_icon.real_name['objectName'] = arrow_name
if occurrence > 1:
self._arrow_icon.real_name['occurrence'] = occurrence
self._arrow_icon.click()
@allure.step('Get list of assets')
def get_list_of_assets(self) -> typing.List:
time.sleep(1)
token_list_items = []
for item in driver.findAllObjects(self._asset_item.real_name):
token_list_items.append(item)
sorted(token_list_items, key=lambda item: item.y)
return token_list_items
2024-05-13 07:09:56 +00:00
@allure.step('Open asset context menu')
def open_asset_context_menu(self, index: int):
QObject(real_name=driver.objectMap.realName(self.get_list_of_assets()[index])).right_click()
return AssetContextMenuPopup().wait_until_appears()
2024-05-13 07:09:56 +00:00
@allure.step('Get list of collectibles')
def get_list_of_collectibles(self) -> typing.List:
time.sleep(1)
token_list_items = []
for item in driver.findAllObjects(self._collectible_item.real_name):
token_list_items.append(item)
sorted(token_list_items, key=lambda item: item.x)
return token_list_items