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

import allure

import configs.timeouts
import driver
from gui.components.wallet.send_popup import SendPopup
from gui.elements.button import Button
from gui.elements.object import QObject
from gui.elements.scroll import Scroll
from gui.elements.text_edit import TextEdit
from gui.objects_map import names


class ENSSettingsView(QObject):

    def __init__(self):
        super().__init__(names.mainWindow_EnsWelcomeView)
        self._start_button = Button(names.mainWindow_Start_StatusButton)

    @allure.step('Start purhasing ens username')
    def start(self):
        self._start_button.click()
        return ENSSearchView().wait_until_appears()


class ENSSearchView(QObject):

    def __init__(self):
        super(ENSSearchView, self).__init__(names.mainWindow_EnsSearchView)
        self._username_text_field = TextEdit(names.mainWindow_ensUsernameInput_StyledTextField)
        self._next_button = Button(names.mainWindow_ensNextButton_StatusRoundButton)
        self._ens_text_note = QObject(names.ens_StatusBaseText)

    @allure.step('Enter user name to text field')
    def enter_user_name(self, name: str):
        self._username_text_field.text = name
        return self._next_button.wait_until_appears()

    @allure.step('Click next button')
    def click_next_button(self):
        self._next_button.click()
        return ENSTermsAndConditionsView().wait_until_appears()

    @allure.step('Get all text notes from ens search view')
    def ens_text_notes(self):
        return [str(note.text) for note in driver.findAllObjects(self._ens_text_note.real_name)]


class ENSTermsAndConditionsView(QObject):

    def __init__(self):
        super(ENSTermsAndConditionsView, self).__init__(names.mainWindow_EnsTermsAndConditionsView)
        self._agree_terms_checkbox = TextEdit(names.sview_ensAgreeTerms_StatusCheckBox)
        self._register_button = Button(names.mainWindow_Register_StatusButton)

    @allure.step('Get registered button enabled state')
    def is_register_button_enabled(self) -> bool:
        return driver.waitForObjectExists(self._register_button.real_name,
                                          configs.timeouts.UI_LOAD_TIMEOUT_MSEC).enabled

    @allure.step('Register ens username')
    def register_ens_name(self):
        self.check_agree_terms_checkbox()
        driver.waitFor(lambda: self.is_register_button_enabled(), configs.timeouts.UI_LOAD_TIMEOUT_MSEC)
        self._register_button.click()
        return SendPopup().wait_until_appears()

    @allure.step('Check agree terms checkbox')
    def check_agree_terms_checkbox(self):
        self._agree_terms_checkbox.click()


class ENSRegisteredView(QObject):

    def __init__(self):
        super(ENSRegisteredView, self).__init__(names.mainWindow_EnsRegisteredView)