257 lines
12 KiB
Python
257 lines
12 KiB
Python
import random
|
|
import typing
|
|
|
|
import allure
|
|
|
|
import configs
|
|
import constants.wallet
|
|
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._watch_only_account_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_eth_address(self, value: str):
|
|
self._origin_combobox.click()
|
|
self._watch_only_account_origin_item.click()
|
|
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 != constants.wallet.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()
|