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

import time
from tests.base_test_case import AbstractTestCase
from views.base_element import Text, Button, EditBox, SilentButton
from views.base_view import BaseView


class OptionsButton(Button):
    def __init__(self, driver):
        super().__init__(driver, xpath="(//android.view.ViewGroup[@content-desc='icon'])[2]")

class AddNewContactButton(Button):
    def __init__(self, driver):
        super().__init__(driver, accessibility_id="add-new-contact-button")

    def navigate(self):
        from views.chat_view import ChatView
        return ChatView(self.driver)


class LogoutButton(SilentButton):
    def __init__(self, driver):
        super().__init__(driver, accessibility_id="log-out-button")

    def click(self):
        self.scroll_to_element().click()
        return self.navigate()


class LogoutDialog(BaseView):
    def __init__(self, driver):
        super().__init__(driver)
        self.logout_button = LogoutDialog.LogoutButton(driver)

    class LogoutButton(SilentButton):
        def __init__(self, driver):
            super().__init__(driver, translation_id="logout", uppercase=True)

        def navigate(self):
            from views.sign_in_view import SignInView
            return SignInView(self.driver)

class ENSusernames(Button):
    def __init__(self, driver):
        super().__init__(driver, translation_id="ens-usernames")

    def navigate(self):
        from views.dapps_view import DappsView
        return DappsView(self.driver)

    def click(self):
        self.scroll_to_element().click()
        return self.navigate()

class AdvancedButton(Button):
    def __init__(self, driver):
        super().__init__(driver, accessibility_id="advanced-button")

    def click(self):
        self.scroll_to_element().click()
        return self.navigate()


class BackupRecoveryPhraseButton(Button):
    def __init__(self, driver):
        super().__init__(driver, accessibility_id="back-up-recovery-phrase-button")

    def click(self):
        self.scroll_to_element().click()
        return self.navigate()


class RecoveryPhraseTable(Text):
    def __init__(self, driver):
        super().__init__(driver, translation_id="your-recovery-phrase",
                         suffix="/following-sibling::android.view.ViewGroup[1]/android.widget.TextView")

class RecoveryPhraseWordNumberText(Text):
    def __init__(self, driver):
        super().__init__(driver, xpath="//*[contains(@text,'#')]")

    @property
    def number(self):
        time.sleep(1)
        return int(self.find_element().text.split('#')[1])


class RecoveryPhraseWordInput(EditBox):
    def __init__(self, driver):
        super(RecoveryPhraseWordInput, self).__init__(driver, xpath="//android.widget.EditText")


class HelpButton(Button):
    def __init__(self, driver):
        super().__init__(driver, accessibility_id="help-button")

    def click(self):
        self.scroll_to_element().click()


class FaqButton(Button):
    def __init__(self, driver):
        super().__init__(driver, accessibility_id="faq-button")

    def navigate(self):
        from views.web_views.base_web_view import BaseWebView
        return BaseWebView(self.driver)


class MailServerElement(Button):
    def __init__(self, driver, server_name):
        super().__init__(driver, xpath="//*[@content-desc='mailserver-item']//*[@text='%s']" % server_name)
        self.server_name = server_name

    def click(self):
        size = self.driver.get_window_size()
        self.driver.swipe(500, size["height"]*0.8, 500, size["height"]*0.05)
        self.find_element().click()


class AboutButton(Button):
    def __init__(self, driver):
        super().__init__(driver, accessibility_id="about-button")

    def click(self):
        self.scroll_to_element().click()
        return self.navigate()


class SyncSettingsButton(Button):
    def __init__(self, driver):
        super().__init__(driver, accessibility_id="sync-settings-button")

    def click(self):
        self.scroll_to_element().click()


class DappPermissionsButton(Button):
    def __init__(self, driver):
        super().__init__(driver, accessibility_id="dapps-permissions-button")

    def click(self):
        self.scroll_to_element().click()


class PrivacyPolicyButton(Button):
    def __init__(self, driver):
        super().__init__(driver, accessibility_id="privacy-policy")

    def navigate(self):
        from views.web_views.base_web_view import BaseWebView
        return BaseWebView(self.driver)


class ProfilePictureElement(Button):
    def __init__(self, driver):
        super().__init__(driver, accessibility_id="chat-icon")


class ProfileView(BaseView):

    def __init__(self, driver):
        super().__init__(driver)
        self.options_button = OptionsButton(self.driver)

        # Header
        self.public_key_text = Text(self.driver, accessibility_id="chat-key")
        self.default_username_text = Text(self.driver, accessibility_id="default-username")
        self.share_my_profile_button = Button(self.driver, accessibility_id="share-header-button")
        self.profile_picture = ProfilePictureElement(self.driver)
        self.edit_picture_button = Button(self.driver, accessibility_id="edit-profile-photo-button")
        self.confirm_edit_button = Button(self.driver, accessibility_id="done-button")
        self.select_from_gallery_button = Button(self.driver, translation_id="profile-pic-pick")
        self.capture_button = Button(self.driver, translation_id="image-source-make-photo")
        self.take_photo_button = Button(self.driver, accessibility_id="take-photo")
        self.crop_photo_button = Button(self.driver, accessibility_id="Crop")
        self.decline_photo_crop = Button(self.driver, accessibility_id="Navigate up")
        self.shutter_button = Button(self.driver, accessibility_id="Shutter")
        self.accept_photo_button = Button(self.driver, accessibility_id="Done")

        # ENS
        self.username_in_ens_chat_settings_text = EditBox(self.driver,
                                                          xpath="//*[@content-desc='chat-icon']/../android.widget.TextView[2]")
        self.ens_usernames_button = ENSusernames(self.driver)
        self.ens_name_in_share_chat_key_text = Text(self.driver, accessibility_id="ens-username")

        # Contacts
        self.contacts_button = Button(self.driver, accessibility_id="contacts-button")
        self.blocked_users_button = Button(self.driver, accessibility_id="blocked-users-list-button")
        self.add_new_contact_button = AddNewContactButton(self.driver)
        self.invite_friends_in_contact_button = Button(self.driver, accessibility_id="invite-friends-button")

        # Privacy and security
        self.privacy_and_security_button = Button(self.driver, accessibility_id="privacy-and-security-settings-button")
        self.accept_new_chats_from = Button(self.driver, accessibility_id="accept-new-chats-from")
        self.accept_new_chats_from_contacts_only = Button(self.driver, translation_id="contacts")
        # Appearance
        self.appearance_button = Button(self.driver, accessibility_id="appearance-settings-button")
        self.show_profile_pictures_of = Button(self.driver, accessibility_id="show-profile-pictures")
        ## Backup recovery phrase
        self.backup_recovery_phrase_button = BackupRecoveryPhraseButton(self.driver)
        self.recovery_phrase_table = RecoveryPhraseTable(self.driver)
        self.recovery_phrase_word_number = RecoveryPhraseWordNumberText(self.driver)
        self.recovery_phrase_word_input = RecoveryPhraseWordInput(self.driver)
        ## Dapps permissions
        self.dapp_permissions_button = DappPermissionsButton(self.driver)
        self.revoke_access_button = Button(self.driver, translation_id="revoke-access")
        ## Delete my profile
        self.delete_my_profile_button = Button(self.driver, translation_id="delete-my-profile")
        self.delete_my_profile_password_input = EditBox(self.driver, xpath="//android.widget.EditText")
        self.delete_profile_button = Button(self.driver, accessibility_id="delete-profile-confirm")

        # Notifications
        self.profile_notifications_button = Button(self.driver, accessibility_id="notifications-settings-button")
        self.profile_notifications_toggle_button = Button(self.driver, accessibility_id="notifications-settings-button")
        self.push_notification_toggle = Button(self.driver,
                                               xpath="//*[@content-desc='notifications-button']//*[@content-desc='switch']")
        self.wallet_push_notifications = Button(self.driver, accessibility_id="notifications-button")

        # Sync settings
        self.sync_settings_button = SyncSettingsButton(self.driver)
        ## Mobile Data
        self.use_mobile_data = Button(self.driver, translation_id="mobile-network-use-mobile",
                                      suffix="/following-sibling::android.widget.Switch[1]")
        self.ask_me_when_on_mobile_network = Button(self.driver, translation_id="mobile-network-ask-me",
                                                    suffix="/following-sibling::android.widget.Switch[1]")
        ##Sync history data
        self.sync_history_for_button = Button(self.driver, accessibility_id="default-sync-period-button")
        ## History nodes
        self.mail_server_button = Button(self.driver, accessibility_id="offline-messages-settings-button")
        self.mail_server_address_input = EditBox(self.driver, translation_id="mailserver-address",
                                                 suffix="/following-sibling::*[1]/android.widget.EditText")
        self.mail_server_connect_button = Button(self.driver, accessibility_id="mailserver-connect-button")
        self.mail_server_auto_selection_button = Button(self.driver, translation_id="mailserver-automatic",
                                                        suffix="/following-sibling::*[1]")
        self.use_history_node_button = Button(self.driver, translation_id="offline-messaging-use-history-nodes",
                                              suffix="/following-sibling::*[1]")
        self.mail_server_delete_button = Button(self.driver, accessibility_id="mailserver-delete-button")
        self.mail_server_confirm_delete_button = Button(self.driver, xpath='//*[@text="%s"]' % self.get_translation_by_key("delete-mailserver").upper())
        ## Device syncing
        self.devices_button = Button(self.driver, accessibility_id="pairing-settings-button")
        self.device_name_input = EditBox(self.driver, accessibility_id="device-name")
        self.go_to_pairing_settings_button = Button(self.driver, translation_id="pairing-go-to-installation",
                                                    uppercase=True)
        self.advertise_device_button = Button(self.driver, accessibility_id="advertise-device")
        self.sync_all_button = Button(self.driver, translation_id="sync-all-devices")

        # Advanced
        self.advanced_button = AdvancedButton(self.driver)
        ## Network
        self.network_settings_button = Button(self.driver, accessibility_id="network-button")
        self.active_network_name = Text(self.driver, xpath="//android.widget.TextView[contains(@text,'with upstream RPC')]")
        self.plus_button = Button(self.driver, xpath="(//android.widget.ImageView[@content-desc='icon'])[2]")
        self.ropsten_chain_button = Button(self.driver, translation_id="ropsten-network")
        self.custom_network_url_input = EditBox(self.driver, translation_id="rpc-url",
                                                suffix="/following-sibling::*[1]/android.widget.EditText")
        self.specify_name_input = EditBox(self.driver, translation_id="name",
                                          suffix="/following-sibling::*[1]/android.widget.EditText")
        self.connect_button = Button(self.driver, accessibility_id="network-connect-button")
        ## Log level
        self.log_level_setting_button = Button(self.driver, accessibility_id="log-level-settings-button")
        ## Fleet
        self.fleet_setting_button = Button(self.driver, accessibility_id="fleet-settings-button")
        ## Bootnodes
        self.bootnodes_button = Button(self.driver, accessibility_id="bootnodes-settings-button")
        self.bootnode_address_input = EditBox(self.driver, accessibility_id="bootnode-address")
        self.enable_bootnodes = Button(self.driver, xpath="//android.widget.Switch")
        self.add_bootnode_button = Button(self.driver, accessibility_id="add-bootnode")

        #Need help
        self.help_button = HelpButton(self.driver)
        self.submit_bug_button = Button(self.driver, accessibility_id="submit-bug-button")
        self.request_a_feature_button = Button(self.driver, accessibility_id="request-a-feature-button")
        self.faq_button = FaqButton(self.driver)

        #About
        self.about_button = AboutButton(self.driver)
        self.privacy_policy_button = PrivacyPolicyButton(self.driver)
        self.app_version_text = Text(self.driver, xpath="//*[@content-desc='app-version']//android.widget.TextView[2]")
        self.node_version_text = Text(self.driver,
                                      xpath="//*[@content-desc='node-version']//android.widget.TextView[2]")

        #Logout
        self.logout_button = LogoutButton(self.driver)
        self.logout_dialog = LogoutDialog(self.driver)
        self.confirm_logout_button = Button(self.driver, translation_id="logout", uppercase=True)

    def switch_network(self, network='Mainnet with upstream RPC'):
        self.driver.info("**Switching network to '%s'**" % network)
        self.advanced_button.click()
        self.network_settings_button.click()
        network_button = Button(self.driver, xpath="//*[@text='%s']" % network)
        network_button.click()
        self.connect_button.click_until_presence_of_element(self.confirm_button)
        self.confirm_button.click_until_absense_of_element(self.confirm_button)
        from views.sign_in_view import SignInView
        SignInView(self.driver).sign_in()

    def open_contact_from_profile(self, username):
        self.driver.info("**Open profile of '%s' via Contacts**" % username)
        self.contacts_button.click()
        self.element_by_text(username).click()
        from views.chat_view import ChatView
        return ChatView(self.driver)

    def add_custom_network(self):
        self.driver.info("**Adding predefined custom network**")
        self.advanced_button.click()
        self.network_settings_button.scroll_to_element()
        self.network_settings_button.click()
        self.plus_button.click_until_presence_of_element(self.ropsten_chain_button)
        self.custom_network_url_input.send_keys('https://ropsten.infura.io/v3/f315575765b14720b32382a61a89341a')
        self.specify_name_input.send_keys('custom_ropsten')
        self.ropsten_chain_button.scroll_to_element()
        self.ropsten_chain_button.click()
        self.ropsten_chain_button.click()
        self.save_button.click()
        self.element_by_text_part('custom_ropsten').scroll_to_element()
        self.element_by_text_part('custom_ropsten').click_until_presence_of_element(self.connect_button)
        self.connect_button.click()
        self.confirm_button.click()
        return self.get_sign_in_view()

    def get_recovery_phrase(self):
        text = [i.text for i in self.recovery_phrase_table.find_elements()]
        return dict(zip(map(int, text[::2]), text[1::2]))

    def backup_recovery_phrase(self):
        self.driver.info("**Backing up seed phrase**")
        self.ok_continue_button.click()
        recovery_phrase = self.get_recovery_phrase()
        self.next_button.click()
        word_number = self.recovery_phrase_word_number.number
        self.recovery_phrase_word_input.set_value(recovery_phrase[word_number])
        self.next_button.click()
        word_number_1 = self.recovery_phrase_word_number.number
        self.recovery_phrase_word_input.set_value(recovery_phrase[word_number_1])
        self.done_button.click()
        self.yes_button.click()
        self.ok_got_it_button.click()
        self.driver.info("**Seed phrase is backed up!**")
        return recovery_phrase

    def edit_profile_picture(self, file_name: str, update_by = "Gallery"):
        self.driver.info("**Setting custom profile image**")
        if not AbstractTestCase().environment == 'sauce':
            raise NotImplementedError('Test case is implemented to run on SauceLabs only')
        self.profile_picture.click()
        self.profile_picture.template = file_name
        if update_by == "Gallery":
            self.select_from_gallery_button.click()
            if self.allow_button.is_element_displayed(sec=5):
                self.allow_button.click()
            image_full_content = self.get_image_in_storage_by_name(file_name)
            if not image_full_content.is_element_displayed(2):
                self.show_roots_button.click()
                for element_text in 'Images', 'DCIM':
                    self.element_by_text(element_text).click()
            image_full_content.click()
        else:
            ## take by Photo
            self.take_photo()
            self.click_system_back_button()
            self.take_photo()
            self.accept_photo_button.click()
        self.crop_photo_button.click()
        self.driver.info("**Custom profile image has been set**")


    def take_photo(self):
        self.take_photo_button.click()
        if self.allow_button.is_element_displayed(sec=5):
            self.allow_button.click()
        if self.allow_all_the_time.is_element_displayed(sec=5):
            self.allow_all_the_time.click()
        if self.element_by_text("NEXT").is_element_displayed(sec=5):
            self.element_by_text("NEXT").click()
        self.shutter_button.click()

    def logout(self):
        self.driver.info("**Logging out**")
        self.logout_button.click()
        self.logout_dialog.logout_button.click()
        self.logout_button.wait_for_invisibility_of_element(30)

    def mail_server_by_name(self, server_name):
        return MailServerElement(self.driver, server_name)

    def get_image_in_storage_by_name(self, image_name=str()):
        return SilentButton(self.driver, xpath="//*[contains(@content-desc,'%s')]" % image_name)

    def get_toggle_device_by_name(self, device_name):
        self.driver.info("**Selecting device %s for sync**" % device_name)
        return SilentButton(self.driver, xpath="//android.widget.TextView[contains(@text,'%s')]/..//android.widget.CheckBox" % device_name)

    def discover_and_advertise_device(self, device_name):
        self.driver.info("**Discover and advertise %s**" % device_name)
        self.sync_settings_button.click()
        self.devices_button.scroll_to_element()
        self.devices_button.click()
        self.device_name_input.set_value(device_name)
        self.continue_button.click_until_presence_of_element(self.advertise_device_button, 2)
        self.advertise_device_button.click()

    def retry_to_connect_to_mailserver(self):
        self.driver.info("**Retrying to connect to mailserver 5 times**")
        i = 0
        while self.element_by_translation_id("mailserver-error-title").is_element_present(20) and i < 5:
            self.element_by_translation_id("mailserver-retry", uppercase=True).click()
            i += 1
            self.just_fyi("retrying to connect: %s attempt" % i)
            time.sleep(10)
        if i == 5:
            self.driver.fail("Failed to connect after %s attempts" % i)

    def connect_existing_status_ens(self, name):
        self.driver.info("**Connect existing ENS: %s**" % name)
        profile = self.profile_button.click()
        profile.switch_network('Mainnet with upstream RPC')
        self.profile_button.click()
        dapp_view = self.ens_usernames_button.click()
        dapp_view.element_by_translation_id("get-started").click()
        dapp_view.ens_name_input.set_value(name)
        dapp_view.check_ens_name.click_until_presence_of_element(self.element_by_translation_id("ens-got-it"))
        dapp_view.element_by_translation_id("ens-got-it").click()
        return dapp_view

    @staticmethod
    def return_mailserver_name(mailserver_name, fleet):
        return mailserver_name + '.' + fleet

    @property
    def current_active_network(self):
        self.advanced_button.click()
        self.active_network_name.scroll_to_element(10, 'up')
        return self.active_network_name.text