329 lines
12 KiB
Python
329 lines
12 KiB
Python
import time
|
|
import base64
|
|
import zbarlight
|
|
from tests import info, common_password
|
|
from eth_keys import datatypes
|
|
from selenium.common.exceptions import NoSuchElementException, TimeoutException
|
|
from PIL import Image
|
|
from datetime import datetime
|
|
from io import BytesIO
|
|
from views.base_element import BaseButton, BaseElement, BaseEditBox, BaseText
|
|
|
|
|
|
class BackButton(BaseButton):
|
|
def __init__(self, driver):
|
|
super(BackButton, self).__init__(driver)
|
|
self.locator = self.Locator.accessibility_id('back-button')
|
|
|
|
def click(self, times_to_click: int = 1):
|
|
for _ in range(times_to_click):
|
|
self.find_element().click()
|
|
info('Tap on %s' % self.name)
|
|
return self.navigate()
|
|
|
|
|
|
class AllowButton(BaseButton):
|
|
def __init__(self, driver):
|
|
super(AllowButton, self).__init__(driver)
|
|
self.locator = self.Locator.xpath_selector("//*[@text='Allow' or @text='ALLOW']")
|
|
|
|
def click(self):
|
|
try:
|
|
for _ in range(3):
|
|
self.find_element().click()
|
|
info('Tap on %s' % self.name)
|
|
except NoSuchElementException:
|
|
pass
|
|
|
|
|
|
class DenyButton(BaseButton):
|
|
def __init__(self, driver):
|
|
super(DenyButton, self).__init__(driver)
|
|
self.locator = self.Locator.xpath_selector("//*[@text='Deny']")
|
|
|
|
|
|
class DeleteButton(BaseButton):
|
|
def __init__(self, driver):
|
|
super(DeleteButton, self).__init__(driver)
|
|
self.locator = self.Locator.xpath_selector("//*[@text='DELETE']")
|
|
|
|
|
|
class YesButton(BaseButton):
|
|
def __init__(self, driver):
|
|
super(YesButton, self).__init__(driver)
|
|
self.locator = self.Locator.xpath_selector("//*[@text='YES']")
|
|
|
|
|
|
class NoButton(BaseButton):
|
|
def __init__(self, driver):
|
|
super(NoButton, self).__init__(driver)
|
|
self.locator = self.Locator.xpath_selector("//*[@text='No']")
|
|
|
|
|
|
class OkButton(BaseButton):
|
|
def __init__(self, driver):
|
|
super(OkButton, self).__init__(driver)
|
|
self.locator = self.Locator.xpath_selector("//*[@text='OK']")
|
|
|
|
|
|
class ContinueButton(BaseButton):
|
|
def __init__(self, driver):
|
|
super(ContinueButton, self).__init__(driver)
|
|
self.locator = self.Locator.xpath_selector("//*[@text='CONTINUE' or @text='Continue']")
|
|
|
|
|
|
class HomeButton(BaseButton):
|
|
def __init__(self, driver):
|
|
super(HomeButton, self).__init__(driver)
|
|
self.locator = self.Locator.accessibility_id('home-tab-button')
|
|
|
|
def navigate(self):
|
|
from views.home_view import HomeView
|
|
return HomeView(self.driver)
|
|
|
|
|
|
class WalletButton(BaseButton):
|
|
def __init__(self, driver):
|
|
super(WalletButton, self).__init__(driver)
|
|
self.locator = self.Locator.accessibility_id('wallet-tab-button')
|
|
|
|
def navigate(self):
|
|
from views.wallet_view import WalletView
|
|
return WalletView(self.driver)
|
|
|
|
|
|
class ProfileButton(BaseButton):
|
|
def __init__(self, driver):
|
|
super(ProfileButton, self).__init__(driver)
|
|
self.locator = self.Locator.accessibility_id('profile-tab-button')
|
|
|
|
def navigate(self):
|
|
from views.profile_view import ProfileView
|
|
return ProfileView(self.driver)
|
|
|
|
|
|
class SaveButton(BaseButton):
|
|
def __init__(self, driver):
|
|
super(SaveButton, self).__init__(driver)
|
|
self.locator = self.Locator.xpath_selector(
|
|
"//android.widget.TextView[@text='SAVE']")
|
|
|
|
|
|
class NextButton(BaseButton):
|
|
def __init__(self, driver):
|
|
super(NextButton, self).__init__(driver)
|
|
self.locator = self.Locator.xpath_selector(
|
|
"//android.widget.TextView[@text='NEXT']")
|
|
|
|
|
|
class DoneButton(BaseButton):
|
|
def __init__(self, driver):
|
|
super(DoneButton, self).__init__(driver)
|
|
self.locator = self.Locator.xpath_selector(
|
|
"//android.widget.TextView[@text='DONE']")
|
|
|
|
|
|
class AppsButton(BaseButton):
|
|
def __init__(self, driver):
|
|
super(AppsButton, self).__init__(driver)
|
|
self.locator = self.Locator.accessibility_id("Apps")
|
|
|
|
|
|
class StatusAppIcon(BaseButton):
|
|
def __init__(self, driver):
|
|
super(StatusAppIcon, self).__init__(driver)
|
|
self.locator = self.Locator.xpath_selector(
|
|
"//*[@text='Status']")
|
|
|
|
|
|
class SendMessageButton(BaseButton):
|
|
def __init__(self, driver):
|
|
super(SendMessageButton, self).__init__(driver)
|
|
self.locator = self.Locator.accessibility_id("send-message-button")
|
|
|
|
def click(self):
|
|
self.find_element().click()
|
|
info('Tap on %s' % self.name)
|
|
|
|
|
|
class ConnectionStatusText(BaseText):
|
|
def __init__(self, driver):
|
|
super(ConnectionStatusText, self).__init__(driver)
|
|
self.locator = self.Locator.xpath_selector(
|
|
"//*[@content-desc='connection-status-text']/android.widget.TextView")
|
|
|
|
|
|
class BaseView(object):
|
|
def __init__(self, driver):
|
|
self.driver = driver
|
|
|
|
self.send_message_button = SendMessageButton(self.driver)
|
|
self.home_button = HomeButton(self.driver)
|
|
self.wallet_button = WalletButton(self.driver)
|
|
self.profile_button = ProfileButton(self.driver)
|
|
|
|
self.yes_button = YesButton(self.driver)
|
|
self.no_button = NoButton(self.driver)
|
|
self.back_button = BackButton(self.driver)
|
|
self.allow_button = AllowButton(self.driver)
|
|
self.deny_button = DenyButton(self.driver)
|
|
self.continue_button = ContinueButton(self.driver)
|
|
self.ok_button = OkButton(self.driver)
|
|
self.next_button = NextButton(self.driver)
|
|
self.save_button = SaveButton(self.driver)
|
|
self.done_button = DoneButton(self.driver)
|
|
self.delete_button = DeleteButton(self.driver)
|
|
self.connection_status = ConnectionStatusText(self.driver)
|
|
|
|
self.apps_button = AppsButton(self.driver)
|
|
self.status_app_icon = StatusAppIcon(self.driver)
|
|
|
|
self.element_types = {
|
|
'base': BaseElement,
|
|
'button': BaseButton,
|
|
'edit_box': BaseEditBox,
|
|
'text': BaseText
|
|
}
|
|
|
|
def accept_agreements(self):
|
|
iterations = int()
|
|
from views.sign_in_view import CreateAccountButton, PasswordInput
|
|
while iterations <= 3 and not (CreateAccountButton(self.driver).is_element_displayed() or PasswordInput(
|
|
self.driver).is_element_displayed()):
|
|
for button in self.ok_button, self.continue_button:
|
|
try:
|
|
button.wait_for_element(15)
|
|
button.click()
|
|
except (NoSuchElementException, TimeoutException):
|
|
pass
|
|
iterations += 1
|
|
|
|
@property
|
|
def logcat(self):
|
|
return self.driver.get_log("logcat")
|
|
|
|
def confirm(self):
|
|
info("Tap 'Confirm' on native keyboard")
|
|
self.driver.press_keycode(66)
|
|
|
|
def send_as_keyevent(self, string):
|
|
keys = {'0': 7, '1': 8, '2': 9, '3': 10, '4': 11, '5': 12, '6': 13, '7': 14, '8': 15, '9': 16,
|
|
|
|
',': 55, '-': 69, '+': 81, '.': 56, '/': 76, '\\': 73, ';': 74, ' ': 62,
|
|
'[': 71, ']': 72, '=': 70, '\n': 66, '_': [69, 5],
|
|
|
|
'a': 29, 'b': 30, 'c': 31, 'd': 32, 'e': 33, 'f': 34, 'g': 35, 'h': 36, 'i': 37, 'j': 38,
|
|
'k': 39, 'l': 40, 'm': 41, 'n': 42, 'o': 43, 'p': 44, 'q': 45, 'r': 46, 's': 47, 't': 48,
|
|
'u': 49, 'v': 50, 'w': 51, 'x': 52, 'y': 53, 'z': 54}
|
|
time.sleep(3)
|
|
info("Enter '%s' using native keyboard" % string)
|
|
for i in string:
|
|
if type(keys[i]) is list:
|
|
keycode, metastate = keys[i][0], keys[i][1]
|
|
else:
|
|
keycode, metastate = keys[i], None
|
|
self.driver.press_keycode(keycode=keycode, metastate=metastate)
|
|
|
|
def find_full_text(self, text, wait_time=60):
|
|
info("Looking for full text: '%s'" % text)
|
|
element = BaseElement(self.driver)
|
|
element.locator = element.Locator.text_selector(text)
|
|
return element.wait_for_element(wait_time)
|
|
|
|
def find_text_part(self, text, wait_time=60):
|
|
info("Looking for a text part: '%s'" % text)
|
|
element = BaseElement(self.driver)
|
|
element.locator = element.Locator.text_part_selector(text)
|
|
return element.wait_for_element(wait_time)
|
|
|
|
def element_by_text(self, text, element_type='button'):
|
|
info("Looking for an element by text: '%s'" % text)
|
|
element = self.element_types[element_type](self.driver)
|
|
element.locator = element.Locator.text_selector(text)
|
|
return element
|
|
|
|
def element_by_text_part(self, text, element_type='base'):
|
|
info("Looking for an element by text part: '%s'" % text)
|
|
element = self.element_types[element_type](self.driver)
|
|
element.locator = element.Locator.text_part_selector(text)
|
|
return element
|
|
|
|
def element_starts_with_text(self, text, element_type='base'):
|
|
info("Looking for full text: '%s'" % text)
|
|
element = self.element_types[element_type](self.driver)
|
|
element.locator = element.Locator.xpath_selector("//*[starts-with(@text,'%s')]" % text)
|
|
return element
|
|
|
|
def wait_for_element_starts_with_text(self, text, wait_time=60):
|
|
info("Looking for full text: '%s'" % text)
|
|
element = BaseElement(self.driver)
|
|
element.locator = element.Locator.xpath_selector("//*[starts-with(@text,'%s')]" % text)
|
|
return element.wait_for_element(wait_time)
|
|
|
|
def swipe_down(self):
|
|
self.driver.swipe(500, 500, 500, 1000)
|
|
|
|
def get_home_view(self):
|
|
from views.home_view import HomeView
|
|
return HomeView(self.driver)
|
|
|
|
def get_chat_view(self):
|
|
from views.chat_view import ChatView
|
|
return ChatView(self.driver)
|
|
|
|
def get_sign_in_view(self):
|
|
from views.sign_in_view import SignInView
|
|
return SignInView(self.driver)
|
|
|
|
def get_send_transaction_view(self):
|
|
from views.send_transaction_view import SendTransactionView
|
|
return SendTransactionView(self.driver)
|
|
|
|
def get_base_web_view(self):
|
|
from views.web_views.base_web_view import BaseWebView
|
|
return BaseWebView(self.driver)
|
|
|
|
def get_profile_view(self):
|
|
from views.profile_view import ProfileView
|
|
return ProfileView(self.driver)
|
|
|
|
def get_wallet_view(self):
|
|
from views.wallet_view import WalletView
|
|
return WalletView(self.driver)
|
|
|
|
def get_unique_amount(self):
|
|
return '0.0%s' % datetime.now().strftime('%-m%-d%-H%-M%-S').strip('0')
|
|
|
|
def get_text_from_qr(self):
|
|
image = Image.open(BytesIO(base64.b64decode(self.driver.get_screenshot_as_base64())))
|
|
image.load()
|
|
try:
|
|
return str(zbarlight.scan_codes('qrcode', image)[0])[2:][:132]
|
|
except IndexError:
|
|
raise BaseException('No data in QR code')
|
|
|
|
def public_key_to_address(self, public_key):
|
|
raw_public_key = bytearray.fromhex(public_key.replace('0x04', ''))
|
|
return datatypes.PublicKey(raw_public_key).to_address()[2:]
|
|
|
|
def get_back_to_home_view(self):
|
|
counter = 0
|
|
while not self.home_button.is_element_displayed(2):
|
|
try:
|
|
if counter >= 5:
|
|
return
|
|
self.back_button.click()
|
|
except (NoSuchElementException, TimeoutException):
|
|
counter += 1
|
|
|
|
def relogin(self, password=common_password):
|
|
self.get_back_to_home_view()
|
|
profile_view = self.profile_button.click()
|
|
profile_view.logout_button.click()
|
|
profile_view.confirm_logout_button.click()
|
|
sign_in_view = self.get_sign_in_view()
|
|
sign_in_view.click_account_by_position(0)
|
|
sign_in_view.password_input.send_keys(password)
|
|
sign_in_view.sign_in_button.click()
|
|
sign_in_view.home_button.wait_for_visibility_of_element()
|