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

import allure

import driver
from driver.objects_access import wait_for_template
from gui.components.base_popup import BasePopup
from gui.elements.button import Button
from gui.elements.object import QObject
from gui.elements.text_edit import TextEdit
from gui.elements.text_label import TextLabel
from gui.objects_map import names


class SendPopup(BasePopup):

    def __init__(self):
        super().__init__()
        self._tab_item_template = QObject(names.tab_Status_template)
        self._search_field = TextEdit(names.search_TextEdit)
        self._asset_list_item = QObject(names.o_TokenBalancePerChainDelegate_template)
        self._amount_text_edit = TextEdit(names.amountInput_TextEdit)
        self._paste_button = Button(names.paste_StatusButton)
        self._ens_address_text_edit = TextEdit(names.ens_or_address_TextEdit)
        self._my_accounts_tab = Button(names.accountSelectionTabBar_My_Accounts_StatusTabButton)
        self._account_list_item = QObject(names.status_account_WalletAccountListItem_template)
        self._arbitrum_network = QObject(names.arbitrum_StatusListItem)
        self._mainnet_network = QObject(names.mainnet_StatusListItem)
        self._fiat_fees_label = TextLabel(names.fiatFees_StatusBaseText)
        self._send_button = Button(names.send_StatusFlatButton)

    def _select_asset(self, asset: str):
        for item in driver.findAllObjects(self._asset_list_item.real_name):
            if str(getattr(item, 'title', '')) == asset:
                driver.mouseClick(item)
            else:
                raise LookupError(f"Chosen asset didn't appear")

    def _open_tab(self, name: str):
        assets_tab = wait_for_template(self._tab_item_template.real_name, name, 'text')
        driver.mouseClick(assets_tab)

    @allure.step('Send {2} {3} to {1}')
    def send(self, address: str, amount: int, asset: str):
        self._open_tab('Assets')
        self._search_field.type_text(asset)
        self._select_asset(asset)
        assert driver.waitFor(lambda: self._amount_text_edit.is_visible, timeout_msec=6000)
        self._amount_text_edit.text = str(amount)
        self._ens_address_text_edit.type_text(address)
        assert driver.waitFor(lambda: self._send_button.is_visible, timeout_msec=6000)
        self.click_send()

    @allure.step('Click send button')
    def click_send(self):
        self._send_button.click()

    def is_arbitrum_network_identified(self) -> bool:
        return self._arbitrum_network.is_visible

    def is_mainnet_network_identified(self) -> bool:
        return self._mainnet_network.is_visible

    def get_fiat_fees(self) -> str:
        return self._fiat_fees_label.text