status-mobile/test/appium/views/send_transaction_view.py

from tests import common_password
from views.base_element import Text, SilentButton
from views.base_element import Button, EditBox
from views.base_view import BaseView

class AmountEditBox(EditBox, Button):
    def __init__(self, driver):
        super(AmountEditBox, self).__init__(driver, accessibility_id="amount-input")

    def set_value(self, value):
        EditBox.set_value(self, value)
        self.driver.press_keycode(66)


class ChooseRecipientButton(Button):
    def __init__(self, driver):
        super(ChooseRecipientButton, self).__init__(driver, accessibility_id="choose-recipient-button")

    def click(self):
        self.click_until_presence_of_element(Button(self.driver, translation_id="my-accounts"))
        return self.navigate()


class GasPriceInput(EditBox):
    def __init__(self, driver):
        super(GasPriceInput, self).__init__(driver, prefix="(", translation_id="gas-limit",
                                            suffix="/..//android.widget.EditText)[2]")

    @property
    def text(self):
        return self.find_element().text


class UpdateFeeButton(Button):
    def __init__(self, driver):
        super(UpdateFeeButton, self).__init__(driver, translation_id="update")

    def click(self):
        for _ in range(3):
            self.find_element().click()
            if not self.is_element_displayed():
                return self.navigate()


class ValidationErrorOnSendTransaction(Button):
    def __init__(self, driver, field):
        super(ValidationErrorOnSendTransaction, self).__init__(driver, xpath="//*[@text='%s']/../*[@content-desc='icon']" % field)


class NotEnoughEthForGas(Text):
    def __init__(self, driver):
        super().__init__(driver, translation_id="wallet-insufficient-gas")


class ValidationWarnings(object):
    def __init__(self, driver):
        self.not_enough_eth_for_gas = NotEnoughEthForGas(driver)


class SignWithKeycardButton(Button):
    def __init__(self, driver):
        super(SignWithKeycardButton, self).__init__(driver, xpath="//*[contains(@text,'%s')]" % self.get_translation_by_key("sign-with"))

    def navigate(self):
        from views.keycard_view import KeycardView
        return KeycardView(self.driver)

    def click(self):
        from views.keycard_view import KeycardView
        self.click_until_presence_of_element(KeycardView(self.driver).one_button)
        return self.navigate()


class SendTransactionView(BaseView):
    def __init__(self, driver):
        super(SendTransactionView, self).__init__(driver)

        self.chose_recipient_button = ChooseRecipientButton(self.driver)
        self.accounts_button = Button(self.driver, translation_id="my-accounts")
        self.enter_recipient_address_button = Button(self.driver, accessibility_id="choose-recipient-recipient-code")
        self.scan_qr_code_button = Button(self.driver, accessibility_id="scan-contact-code-button")
        self.enter_recipient_address_input = EditBox(self.driver, accessibility_id="recipient-address-input")
        self.first_recipient_button = Button(self.driver, accessibility_id="chat-icon")
        self.enter_recipient_address_text = Text(self.driver, xpath="//*[@content-desc='choose-recipient-button']//android.widget.TextView")

        self.recent_recipients_button = Button(self.driver, translation_id="recent-recipients")
        self.amount_edit_box = AmountEditBox(self.driver)
        self.set_max_button = Button(self.driver, translation_id="set-max")
        self.validation_error_element = Text(self.driver, xpath="//*[@content-desc='custom-gas-fee']/../android.view.ViewGroup//*[@content-desc='icon']")

        self.network_fee_button = Button(self.driver, accessibility_id="custom-gas-fee")
        self.transaction_fee_button = Button(self.driver, accessibility_id="transaction-fee-button")
        self.transaction_fee_total_value = Text(self.driver, translation_id="wallet-transaction-total-fee", suffix="//following::android.widget.TextView[1]")
        self.gas_limit_input = EditBox(self.driver, prefix="(", translation_id="gas-limit", suffix="/..//android.widget.EditText)[1]")
        self.gas_price_input = GasPriceInput(self.driver)
        self.total_fee_input = EditBox(self.driver, translation_id="wallet-transaction-total-fee", suffix="/following-sibling::android.widget.TextView")
        self.update_fee_button = UpdateFeeButton(self.driver)


        self.sign_transaction_button = Button(self.driver, accessibility_id="send-transaction-bottom-sheet")
        self.sign_with_keycard_button = SignWithKeycardButton(self.driver)
        self.sign_with_password = Button(self.driver, translation_id="sign-with-password")
        self.sign_button = Button(self.driver, translation_id="transactions-sign")
        self.sign_in_phrase_text = Text(self.driver, accessibility_id="signing-phrase-text")
        self.enter_password_input = EditBox(self.driver, accessibility_id="enter-password-input")
        self.got_it_button = Button(self.driver, accessibility_id="got-it-button")

        self.select_asset_button = Button(self.driver, accessibility_id="choose-asset-button")
        self.asset_text = Text(self.driver, xpath="//*[@content-desc='choose-asset-button']//android.widget.TextView")
        self.recipient_text = Text(self.driver, xpath="//*[@content-desc='choose-recipient-button']//android.widget.TextView")

        self.share_button = Button(self.driver, accessibility_id="share-address-button")

        self.onboarding_message = Text(self.driver, translation_id="this-is-you-signing")
        self.validation_warnings = ValidationWarnings(self.driver)
        self.eth_asset_in_select_asset_bottom_sheet_button = Button(self.driver, accessibility_id=":ETH-asset-value")

        # Elements for commands in 1-1 chat
        self.select_button = Button(self.driver, accessibility_id="select-account-bottom-sheet")
        self.request_transaction_button = Button(self.driver, accessibility_id="request-transaction-bottom-sheet")

        # Elements on set recipient screen
        self.recipient_add_to_favorites = Button(self.driver, accessibility_id="participant-add-to-favs")
        self.recipient_done = Button(self.driver, accessibility_id="participant-done")
        self.new_favorite_name_input = EditBox(self.driver, accessibility_id="fav-name")
        self.new_favorite_add_favorite = Button(self.driver, accessibility_id="add-fav")


    def set_recipient_address(self, address):
        self.driver.info("**Setting recipient address to %s**" % address)
        self.chose_recipient_button.click()
        self.enter_recipient_address_input.set_value(address)
        self.enter_recipient_address_input.click()
        self.done_button.click_until_absense_of_element(self.done_button)

    def sign_transaction(self, sender_password: str = common_password, keycard=False, default_gas_price=False):
        self.driver.info("**Signing transaction (keycard:%s, default_gas_price:%s)**" % (str(keycard), str(default_gas_price)))
        if self.sign_in_phrase.is_element_displayed():
            self.set_up_wallet_when_sending_tx()
        if not default_gas_price:
            self.network_fee_button.click()
            default_gas_price = self.gas_price_input.text
            self.gas_price_input.clear()
            self.gas_price_input.set_value(str(int(float(default_gas_price))+30))
            self.update_fee_button.click()
        if keycard:
            keycard_view = self.sign_with_keycard_button.click()
            keycard_view.enter_default_pin()
        else:
            self.sign_with_password.click_until_presence_of_element(self.enter_password_input)
            self.enter_password_input.send_keys(sender_password)
            self.sign_button.click_until_absense_of_element(self.sign_button)
        self.ok_button.wait_for_element(120)
        if self.element_by_text_part('Transaction failed').is_element_displayed():
            self.driver.fail('Transaction failed')
        self.driver.info("**Transaction is signed!**")
        self.ok_button.click()

    def get_transaction_fee_total(self):
        self.driver.info("**Getting transaction fee**")
        self.network_fee_button.click_until_presence_of_element(self.gas_limit_input)
        fee_value = self.transaction_fee_total_value.text.split()[0]
        self.update_fee_button.click()
        return fee_value

    @staticmethod
    def get_formatted_recipient_address(address):
        return address[:6] + '…' + address[-4:]

    def get_username_in_transaction_bottom_sheet_button(self, username_part):
        self.driver.info("**Getting username by '%s' in transaction fee bottom sheet**" % username_part)
        return SilentButton(self.driver, xpath="//*[@content-desc='amount-input']/..//*[starts-with(@text,'%s')]" % username_part)

    def get_account_in_select_account_bottom_sheet_button(self, account_name):
        self.driver.info("**Getting account by '%s' in transaction fee bottom sheet**" % account_name)
        return SilentButton(self.driver, translation_id="select-account", suffix="/..//*[starts-with(@text,'%s')]" % account_name)

    def get_validation_icon(self, field='Network fee'):
        return ValidationErrorOnSendTransaction(self.driver, field)

    def get_values_from_send_transaction_bottom_sheet(self, gas=False):
        self.driver.info("**Getting values from send transaction bottom sheet**")
        data = {
            'amount': self.amount_edit_box.text,
            'asset': self.asset_text.text,
            'address':  self.enter_recipient_address_text.text
        }
        if gas:
            self.sign_transaction_button.click_until_presence_of_element(self.sign_with_password)
            self.network_fee_button.click_until_presence_of_element(self.gas_limit_input)
            data['gas_limit'] = self.gas_limit_input.text
            data['gas_price'] = self.gas_price_input.text
            self.cancel_button.click()
        return data

    def add_to_favorites(self, name):
        self.driver.info("**Adding '%s' to favorite recipients**" % name)
        self.recipient_add_to_favorites.click()
        self.new_favorite_name_input.set_value(name)
        self.new_favorite_add_favorite.click()