import random import typing import allure import configs from constants.wallet import * import driver from gui.screens.settings_wallet import * from gui.components.base_popup import BasePopup from gui.components.emoji_popup import EmojiPopup from gui.components.wallet.authenticate_popup import AuthenticatePopup from gui.components.wallet.back_up_your_seed_phrase_popup import BackUpYourSeedPhrasePopUp from gui.elements.button import Button from gui.elements.check_box import CheckBox from gui.elements.object import QObject from gui.elements.scroll import Scroll from gui.elements.text_edit import TextEdit GENERATED_PAGES_LIMIT = 20 class AccountPopup(BasePopup): def __init__(self): super(AccountPopup, self).__init__() self._scroll = Scroll('o_Flickable') self._name_text_edit = TextEdit('mainWallet_AddEditAccountPopup_AccountName') self._emoji_button = Button('mainWallet_AddEditAccountPopup_AccountEmojiPopupButton') self._color_radiobutton = QObject('color_StatusColorRadioButton') # origin self._origin_combobox = QObject('mainWallet_AddEditAccountPopup_SelectedOrigin') self._watched_address_origin_item = QObject("mainWallet_AddEditAccountPopup_OriginOptionWatchOnlyAcc") self._new_master_key_origin_item = QObject('mainWallet_AddEditAccountPopup_OriginOptionNewMasterKey') self._existing_origin_item = QObject('addAccountPopup_OriginOption_StatusListItem') self._use_keycard_button = QObject('mainWallet_AddEditAccountPopup_MasterKey_GoToKeycardSettingsOption') # derivation self._address_text_edit = TextEdit('mainWallet_AddEditAccountPopup_AccountWatchOnlyAddress') self._add_account_button = Button('mainWallet_AddEditAccountPopup_PrimaryButton') self._edit_derivation_path_button = Button('mainWallet_AddEditAccountPopup_EditDerivationPathButton') self._derivation_path_combobox_button = Button('mainWallet_AddEditAccountPopup_PreDefinedDerivationPathsButton') self._derivation_path_list_item = QObject('mainWallet_AddEditAccountPopup_derivationPath') self._reset_derivation_path_button = Button('mainWallet_AddEditAccountPopup_ResetDerivationPathButton') self._derivation_path_text_edit = TextEdit('mainWallet_AddEditAccountPopup_DerivationPathInput') self._address_combobox_button = Button('mainWallet_AddEditAccountPopup_GeneratedAddressComponent') self._non_eth_checkbox = CheckBox('mainWallet_AddEditAccountPopup_NonEthDerivationPathCheckBox') @allure.step('Set name for account') def set_name(self, value: str): self._name_text_edit.text = value return self @allure.step('Set color for account') def set_color(self, value: str): if 'radioButtonColor' in self._color_radiobutton.real_name.keys(): del self._color_radiobutton.real_name['radioButtonColor'] colors = [str(item.radioButtonColor) for item in driver.findAllObjects(self._color_radiobutton.real_name)] assert value in colors, f'Color {value} not found in {colors}' self._color_radiobutton.real_name['radioButtonColor'] = value self._color_radiobutton.click() return self @allure.step('Set emoji for account') def set_emoji(self, value: str): self._emoji_button.click() EmojiPopup().wait_until_appears().select(value) return self @allure.step('Set eth address for account added from context menu') def set_eth_address(self, value: str): self._address_text_edit.text = value return self @allure.step('Set eth address for account added from plus button') def set_origin_watched_address(self, value: str): self._origin_combobox.click() self._watched_address_origin_item.click() assert self._origin_combobox.__getattr__('title') == WalletOrigin.WATCHED_ADDRESS_ORIGIN.value self._address_text_edit.text = value return self @allure.step('Set private key for account') def set_origin_private_key(self, value: str): self._origin_combobox.click() self._new_master_key_origin_item.click() AddNewAccountPopup().wait_until_appears().import_private_key(value) return self @allure.step('Set new seed phrase for account') def set_origin_new_seed_phrase(self, value: str): self._origin_combobox.click() self._new_master_key_origin_item.click() AddNewAccountPopup().wait_until_appears().generate_new_master_key(value) return self @allure.step('Set seed phrase') def set_origin_seed_phrase(self, value: typing.List[str]): self._origin_combobox.click() self._new_master_key_origin_item.click() AddNewAccountPopup().wait_until_appears().import_new_seed_phrase(value) return self @allure.step('Set derivation path for account') def set_derivation_path(self, value: str, index: int, password: str): self._edit_derivation_path_button.hover().click() AuthenticatePopup().wait_until_appears().authenticate(password) if value in [_.value for _ in constants.wallet.DerivationPath]: self._derivation_path_combobox_button.click() self._derivation_path_list_item.real_name['title'] = value self._derivation_path_list_item.click() del self._derivation_path_list_item.real_name['title'] self._address_combobox_button.click() GeneratedAddressesList().wait_until_appears().select(index) if value != DerivationPath.ETHEREUM.value: self._scroll.vertical_scroll_to(self._non_eth_checkbox) self._non_eth_checkbox.set(True) else: self._derivation_path_text_edit.type_text(str(index)) return self @allure.step('Click continue in keycard settings') def continue_in_keycard_settings(self): self._origin_combobox.click() self._new_master_key_origin_item.click() self._use_keycard_button.click() return self @allure.step('Save added account') def save(self): self._add_account_button.wait_until_appears().click() return self class EditAccountFromSettingsPopup(BasePopup): def __init__(self): super(EditAccountFromSettingsPopup, self).__init__() self._change_name_button = Button('editWalletSettings_renameButton') self._account_name_input = TextEdit('editWalletSettings_AccountNameInput') self._emoji_selector = QObject('editWalletSettings_EmojiSelector') self._color_radiobutton = QObject('editWalletSettings_ColorSelector') self._emoji_item = QObject('editWalletSettings_EmojiItem') @allure.step('Click Change name button') def click_change_name_button(self): self._change_name_button.click() @allure.step('Type in name for account') def type_in_account_name(self, value: str): self._account_name_input.text = value return self @allure.step('Select random color for account') def select_random_color_for_account(self): if 'radioButtonColor' in self._color_radiobutton.real_name.keys(): del self._color_radiobutton.real_name['radioButtonColor'] colors = [str(item.radioButtonColor) for item in driver.findAllObjects(self._color_radiobutton.real_name)] self._color_radiobutton.real_name['radioButtonColor'] =\ random.choice([color for color in colors if color != '#2a4af5']) # exclude status default color self._color_radiobutton.click() return self @allure.step('Click emoji button') def select_random_emoji_for_account(self): self._emoji_selector.click() EmojiPopup().wait_until_appears() emojis = [str(item.objectName) for item in driver.findAllObjects(self._emoji_item.real_name)] value = ((random.choice(emojis)).split('_', 1))[1] EmojiPopup().wait_until_appears().select(value) return self class AddNewAccountPopup(BasePopup): def __init__(self): super(AddNewAccountPopup, self).__init__() self._import_private_key_button = Button('mainWallet_AddEditAccountPopup_MasterKey_ImportPrivateKeyOption') self._private_key_text_edit = TextEdit('mainWallet_AddEditAccountPopup_PrivateKey') self._private_key_name_text_edit = TextEdit('mainWallet_AddEditAccountPopup_PrivateKeyName') self._continue_button = Button('mainWallet_AddEditAccountPopup_PrimaryButton') self._import_seed_phrase_button = Button('mainWallet_AddEditAccountPopup_MasterKey_ImportSeedPhraseOption') self._generate_master_key_button = Button('mainWallet_AddEditAccountPopup_MasterKey_GenerateSeedPhraseOption') self._seed_phrase_12_words_button = Button("mainWallet_AddEditAccountPopup_12WordsButton") self._seed_phrase_18_words_button = Button("mainWallet_AddEditAccountPopup_18WordsButton") self._seed_phrase_24_words_button = Button("mainWallet_AddEditAccountPopup_24WordsButton") self._seed_phrase_word_text_edit = TextEdit('mainWallet_AddEditAccountPopup_SPWord') self._seed_phrase_phrase_key_name_text_edit = TextEdit( 'mainWallet_AddEditAccountPopup_ImportedSeedPhraseKeyName') @allure.step('Import private key') def import_private_key(self, private_key: str) -> str: self._import_private_key_button.click() self._private_key_text_edit.text = private_key self._private_key_name_text_edit.text = private_key[:5] self._continue_button.click() return private_key[:5] @allure.step('Import new seed phrase') def import_new_seed_phrase(self, seed_phrase_words: list) -> str: self._import_seed_phrase_button.click() if len(seed_phrase_words) == 12: self._seed_phrase_12_words_button.click() elif len(seed_phrase_words) == 18: self._seed_phrase_18_words_button.click() elif len(seed_phrase_words) == 24: self._seed_phrase_24_words_button.click() else: raise RuntimeError("Wrong amount of seed words", len(seed_phrase_words)) for count, word in enumerate(seed_phrase_words, start=1): self._seed_phrase_word_text_edit.real_name['objectName'] = f'statusSeedPhraseInputField{count}' self._seed_phrase_word_text_edit.text = word seed_phrase_name = ''.join([word[0] for word in seed_phrase_words[:10]]) self._seed_phrase_phrase_key_name_text_edit.text = seed_phrase_name self._continue_button.click() return seed_phrase_name @allure.step('Generate new seed phrase') def generate_new_master_key(self, name: str): self._generate_master_key_button.click() BackUpYourSeedPhrasePopUp().wait_until_appears().generate_seed_phrase(name) class GeneratedAddressesList(QObject): def __init__(self): super(GeneratedAddressesList, self).__init__('statusDesktop_mainWindow_overlay_popup2') self._address_list_item = QObject('addAccountPopup_GeneratedAddress') self._paginator_page = QObject('page_StatusBaseButton') @property @allure.step('Load generated addresses list') def is_paginator_load(self) -> bool: try: return str(driver.findAllObjects(self._paginator_page.real_name)[0].text) == '1' except IndexError: return False @allure.step('Wait until appears {0}') def wait_until_appears(self, timeout_msec: int = configs.timeouts.UI_LOAD_TIMEOUT_MSEC): if 'text' in self._paginator_page.real_name: del self._paginator_page.real_name['text'] assert driver.waitFor(lambda: self.is_paginator_load, timeout_msec), 'Generated address list not load' return self @allure.step('Select address in list') def select(self, index: int): self._address_list_item.real_name['objectName'] = f'AddAccountPopup-GeneratedAddress-{index}' selected_page_number = 1 while selected_page_number != GENERATED_PAGES_LIMIT: if self._address_list_item.is_visible: self._address_list_item.click() self._paginator_page.wait_until_hidden() break else: selected_page_number += 1 self._paginator_page.real_name['text'] = selected_page_number self._paginator_page.click()