import base64 import random import re import string import time from datetime import datetime from appium.webdriver.common.touch_action import TouchAction from selenium.common.exceptions import NoSuchElementException, TimeoutException from support.device_apps import start_web_browser from tests import common_password, pytest_config_global, transl from views.base_element import Button, BaseElement, EditBox, Text, CheckBox class BackButton(Button): def __init__(self, driver): super().__init__(driver, accessibility_id="back-button") def click(self, times_to_click: int = 1): for _ in range(times_to_click): self.find_element().click() return self.navigate() class AllowButton(Button): def __init__(self, driver): super().__init__(driver, translation_id="allow", uppercase=True) def click(self, times_to_click=3): try: for _ in range(times_to_click): self.find_element().click() except NoSuchElementException: pass class UnreadMessagesCountText(Text): def __init__(self, driver, parent_locator: str): super().__init__(driver, xpath="(%s//android.widget.TextView)[last()]" % parent_locator) class TabButton(Button): @property def counter(self): return UnreadMessagesCountText(self.driver, parent_locator='//*[@content-desc="%s"]' % self.accessibility_id) class HomeButton(TabButton): def __init__(self, driver): super().__init__(driver, xpath="//*[contains(@content-desc,'tab, 1 out of 5')]") def navigate(self): from views.home_view import HomeView return HomeView(self.driver) def click(self, desired_view='home'): from views.chat_view import ChatView from views.home_view import HomeView if desired_view == 'home': ChatView(self.driver).get_back_to_home_view() element = HomeView(self.driver).plus_button if not element.is_element_displayed(): self.click_until_presence_of_element(element) elif desired_view == 'chat': element = ChatView(self.driver).chat_message_input self.click_until_presence_of_element(element) elif desired_view == 'other_user_profile': element = ChatView(self.driver).profile_nickname self.click_until_presence_of_element(element) return self.navigate() class CommunitiesTab(TabButton): def __init__(self, driver): super().__init__(driver, accessibility_id="communities-stack-tab") class ChatsTab(TabButton): def __init__(self, driver): super().__init__(driver, accessibility_id="chats-stack-tab") def navigate(self): from views.home_view import HomeView return HomeView(self.driver) class WalletTab(TabButton): def __init__(self, driver): super().__init__(driver, accessibility_id="wallet-stack-tab") class BrowserTab(TabButton): def __init__(self, driver): super().__init__(driver, accessibility_id="browser-stack-tab") class DappTabButton(TabButton): def __init__(self, driver): super().__init__(driver, xpath="//*[contains(@content-desc,'tab, 2 out of 5')]") def navigate(self): from views.dapps_view import DappsView return DappsView(self.driver) def click(self, desired_element_text=None): if desired_element_text is None: super().click() elif desired_element_text == 'webview': self.find_element().click() else: base_view = BaseView(self.driver) self.click_until_presence_of_element(base_view.element_by_text_part(desired_element_text)) return self.navigate() class WalletButton(TabButton): def __init__(self, driver): super().__init__(driver, xpath="//*[contains(@content-desc,'tab, 3 out of 5')]") def navigate(self): from views.wallet_view import WalletView return WalletView(self.driver) def click(self): from views.wallet_view import WalletView self.click_until_presence_of_element(WalletView(self.driver).multiaccount_more_options) return self.navigate() class ProfileButton(TabButton): def __init__(self, driver): super().__init__(driver, accessibility_id="open-profile") def navigate(self): from views.profile_view import ProfileView return ProfileView(self.driver) def click(self, desired_element_text='privacy'): if not self.is_element_displayed(): ChatsTab(self.driver).click() from views.profile_view import ProfileView if desired_element_text == 'privacy': self.click_until_presence_of_element(ProfileView(self.driver).privacy_and_security_button) else: base_view = BaseView(self.driver) self.click_until_presence_of_element(base_view.element_by_text_part(desired_element_text)) return self.navigate() class StatusButton(TabButton): def __init__(self, driver): super().__init__(driver, xpath="//*[contains(@content-desc,'tab, 4 out of 5')]") def navigate(self): from views.chat_view import ChatView return ChatView(self.driver) class SendMessageButton(Button): def __init__(self, driver): super().__init__(driver, accessibility_id="send-message-button") class AssetButton(Button): def __init__(self, driver, asset_name): super().__init__(driver, xpath="(//*[@content-desc=':%s-asset-value'])[1]" % asset_name) self.asset_name = asset_name @property def name(self): return self.asset_name + self.__class__.__name__ def click(self): self.wait_for_element().click() class OpenInStatusButton(Button): def __init__(self, driver): super().__init__(driver, translation_id="browsing-open-in-status") def click(self): self.wait_for_visibility_of_element() # using sleep is wrong, but implicit wait for element can't help in particular case time.sleep(3) self.swipe_to_web_element() self.wait_for_element().click() class EnterQRcodeEditBox(EditBox): def __init__(self, driver): super().__init__(driver, translation_id="type-a-message") def scan_qr(self, value): self.set_value(value) base_view = BaseView(self.driver) base_view.ok_button.click() def click(self): self.wait_for_element().click() self.wait_for_invisibility_of_element() class AirplaneModeButton(Button): def __init__(self, driver): super().__init__(driver, accessibility_id="Airplane mode") def open_quick_action_menu(self): action = TouchAction(self.driver) action.press(None, 200, 0).move_to(None, 200, 300).perform() def click(self): counter = 0 desired_element = AirplaneModeButton(self.driver) while not desired_element.is_element_displayed() and counter <= 3: try: self.open_quick_action_menu() desired_element.wait_for_element(5) except (NoSuchElementException, TimeoutException): counter += 1 else: self.driver.info("%s element not found" % desired_element.name) super(AirplaneModeButton, self).click() self.driver.press_keycode(4) class SignInPhraseText(Text): def __init__(self, driver): super().__init__(driver, translation_id="this-is-you-signing", suffix="//following-sibling::*[2]/android.widget.TextView") @property def list(self): return self.text.split() class BaseView(object): def __init__(self, driver): self.driver = driver self.send_message_button = SendMessageButton(self.driver) self.send_contact_request_button = Button(self.driver, translation_id="send-request") # Old UI Tabs self.home_button = HomeButton(self.driver) self.wallet_button = WalletButton(self.driver) self.profile_button = ProfileButton(self.driver) self.dapp_tab_button = DappTabButton(self.driver) self.status_button = StatusButton(self.driver) # New UI Tabs self.communities_tab = CommunitiesTab(self.driver) self.chats_tab = ChatsTab(self.driver) self.browser_tab = BrowserTab(self.driver) self.wallet_tab = WalletTab(self.driver) # Floating screens (introduced by https://github.com/status-im/status-mobile/pull/16438) self.chat_floating_screen = BaseElement(self.driver, accessibility_id=":chat-floating-screen") self.community_floating_screen = BaseElement(self.driver, accessibility_id=":community-overview-floating-screen") self.jump_to_button = Button(self.driver, accessibility_id="jump-to") self.yes_button = Button(self.driver, xpath="//*[@text='YES' or @text='GOT IT']") self.no_button = Button(self.driver, translation_id="no") self.back_button = BackButton(self.driver) self.allow_button = AllowButton(self.driver) self.allow_all_the_time = Button(self.driver, xpath="//*[@text='Allow all the time']") self.deny_button = Button(self.driver, translation_id="deny", uppercase=True) self.continue_button = Button(self.driver, translation_id="continue", uppercase=True) self.ok_button = Button(self.driver, xpath="//*[@text='OK' or @text='Ok']") self.next_button = Button(self.driver, translation_id="next") self.add_button = Button(self.driver, translation_id="add") self.save_button = Button(self.driver, translation_id="save") self.done_button = Button(self.driver, translation_id="done") self.delete_button = Button(self.driver, translation_id="delete", uppercase=True) self.ok_continue_button = Button(self.driver, xpath="//*[@text='OK, CONTINUE' or @text='Okay, continue']") self.discard_button = Button(self.driver, xpath="//*[@text='DISCARD']") self.confirm_button = Button(self.driver, translation_id='confirm', uppercase=True) self.cross_icon = Button(self.driver, xpath="(//android.widget.ImageView[@content-desc='icon'])[1]") self.close_sticker_view_icon = Button(self.driver, xpath="//androidx.appcompat.widget.LinearLayoutCompat") self.native_close_button = Button(self.driver, id="android:id/aerr_close") self.close_button = Button(self.driver, accessibility_id="back-button") self.navigate_up_button = Button(self.driver, accessibility_id="Navigate Up") self.show_roots_button = Button(self.driver, accessibility_id="Show roots") self.get_started_button = Button(self.driver, translation_id="get-started") self.ok_got_it_button = Button(self.driver, translation_id="ok-got-it") self.cross_icon_inside_welcome_screen_button = Button(self.driver, accessibility_id='hide-home-button') self.status_in_background_button = Button(self.driver, xpath="//*[contains(@content-desc,'Status')]") self.cancel_button = Button(self.driver, translation_id="cancel", uppercase=True) self.search_input = EditBox(self.driver, accessibility_id="search-input") self.share_button = Button(self.driver, accessibility_id="share-my-contact-code-button") self.qr_code_image = Button(self.driver, accessibility_id="qr-code-image") self.sign_in_phrase = SignInPhraseText(self.driver) self.toast_content_element = BaseElement(self.driver, accessibility_id="toast-content") # checkboxes and toggles self.checkbox_button = CheckBox(self.driver, accessibility_id="checkbox-off") # external browser self.open_in_status_button = OpenInStatusButton(self.driver) self.apps_button = Button(self.driver, accessibility_id="Apps") self.status_app_icon = Button(self.driver, translation_id="status") self.airplane_mode_button = AirplaneModeButton(self.driver) self.enter_qr_edit_box = EnterQRcodeEditBox(self.driver) self.element_types = { 'base': BaseElement, 'button': Button, 'edit_box': EditBox, 'text': Text } @property def status_account_name(self): return self.get_translation_by_key('main-account') def accept_agreements(self): iterations = int() self.close_native_device_dialog("Messages has stopped") self.close_native_device_dialog("YouTube") while iterations <= 1 and (self.ok_button.is_element_displayed(2) or self.continue_button.is_element_displayed(2)): for button in self.ok_button, self.continue_button: try: button.wait_for_element(3) button.click() except (NoSuchElementException, TimeoutException): pass iterations += 1 @staticmethod def get_translation_by_key(translation_id): return transl[translation_id] def rooted_device_continue(self): try: self.continue_button.wait_for_element(3) self.continue_button.click() except (NoSuchElementException, TimeoutException): pass def close_native_device_dialog(self, alert_text_part): element = self.element_by_text_part(alert_text_part) if element.is_element_displayed(1): self.driver.info("Closing '%s' alert..." % alert_text_part) self.native_close_button.click() @property def logcat(self): logcat = self.driver.get_log("logcat") if len(logcat) > 1000: return str([i for i in logcat if not ('appium' in str(i).lower() or ':1.000000.' in str(i).lower())]) raise TimeoutError('Logcat is empty') def confirm(self): self.driver.info("Tap 'Confirm' on native keyboard") self.driver.press_keycode(66) def confirm_until_presence_of_element(self, desired_element, attempts=3): counter = 0 while not desired_element.is_element_displayed(1) and counter <= attempts: try: self.confirm() self.driver.info("Wait for '%s'" % desired_element.name) desired_element.wait_for_element(5) return except TimeoutException: counter += 1 def just_fyi(self, some_str): self.driver.info('# STEP: %s' % some_str, device=False) self.driver.execute_script("sauce:context=STEP: %s" % some_str) def hide_keyboard_if_shown(self): if self.driver.is_keyboard_shown(): self.click_system_back_button() def click_system_back_button(self, times=1): self.driver.info('Click system back button') for _ in range(times): self.driver.press_keycode(4) def click_system_back_button_until_element_is_shown(self, attempts=3, element='home'): counter = 0 if self.chat_floating_screen.is_element_displayed(2) or self.community_floating_screen.is_element_displayed(2): self.driver.press_keycode(4) if element == 'home': element = self.chats_tab elif element == 'chat': chat = self.get_chat_view() element = chat.chat_message_input while not element.is_element_displayed(1) and counter <= attempts: self.driver.press_keycode(4) try: element.wait_for_element(2) return self except (NoSuchElementException, TimeoutException): counter += 1 else: self.driver.info("Could not reach %s element by pressing back" % element.name) def get_app_from_background(self): self.driver.info('Get Status back from Recent apps') self.driver.press_keycode(187) self.status_in_background_button.click() def put_app_to_background_and_back(self, time_in_background=1): self.driver.info('Put app to background and back') self.driver.press_keycode(187) time.sleep(time_in_background) self.status_in_background_button.click() def click_system_home_button(self): self.driver.info('Press system Home button') self.driver.press_keycode(3) def put_app_to_background(self): self.driver.info('App to background') self.driver.press_keycode(187) def cut_text(self): self.driver.info('Cut text') self.driver.press_keycode(277) def copy_text(self): self.driver.info('Copy text') self.driver.press_keycode(278) def paste_text(self): self.driver.info('Paste text') self.driver.press_keycode(279) def send_as_keyevent(self, keyevent): self.driver.info("Sending as keyevent `%s`" % keyevent) 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], ':': [74, 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) for i in keyevent: if i.isalpha() and i.isupper(): keycode, metastate = keys[i.lower()], 64 # META_SHIFT_LEFT_ON Constant Value: 64. Example: i='n' -> 'N' elif 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 element_by_text(self, text, element_type='button'): element = self.element_types[element_type](self.driver) element.locator = '//*[@text="%s"]' % text return element def element_by_text_part(self, text, element_type='button'): element = self.element_types[element_type](self.driver) element.locator = '//*[contains(@text, "' + text + '")]' return element def element_starts_with_text(self, text, element_type='button'): element = self.element_types[element_type](self.driver, xpath="//*[starts-with(@text,'%s')]" % text) return element def element_by_translation_id(self, translation_id, element_type='button', uppercase=False): element = self.element_types[element_type](self.driver, translation_id=translation_id, uppercase=uppercase) return element def wait_for_element_starts_with_text(self, text, wait_time=60): element = Button(self.driver, xpath="//*[starts-with(@text,'%s')]" % text) return element.wait_for_element(wait_time) def swipe_by_custom_coordinates(self, x_start, y_start, x_end, y_end): """Uses percentage values based on device width/height""" self.driver.info("Swiping based on custom coordinates relative to device height/width") size = self.driver.get_window_size() self.driver.swipe(size["width"] * x_start, size["height"] * y_start, size["width"] * x_end, size["height"] * y_end) def swipe_up(self): self.driver.info("Swiping up") size = self.driver.get_window_size() self.driver.swipe(size["width"] * 0.5, size["height"] * 0.8, size["width"] * 0.5, size["height"] * 0.2) def swipe_down(self): self.driver.info("Swiping down") size = self.driver.get_window_size() self.driver.swipe(size["width"] * 0.5, size["height"] * 0.2, size["width"] * 0.5, size["height"] * 0.8) def swipe_left(self): self.driver.info("Swiping left") size = self.driver.get_window_size() self.driver.swipe(size["width"] * 0.8, size["height"] * 0.8, size["width"] * 0.2, size["height"] * 0.8) def swipe_right(self): self.driver.info("Swiping right") size = self.driver.get_window_size() self.driver.swipe(size["width"] * 0.2, size["height"] * 0.8, size["width"] * 0.8, size["height"] * 0.8) def switch_to_mobile(self, before_login=False, sync=False): self.driver.info("Turning on mobile data, syncing is %s" % str(sync)) self.driver.set_network_connection(4) if before_login is False: from views.home_view import HomeView home = HomeView(self.driver) if sync is True: home.continue_syncing_button.wait_and_click() else: home.stop_syncing_button.wait_and_click() def pull_to_refresh(self, wait_sec=20): self.driver.info("Pull to refresh view") self.driver.swipe(500, 500, 500, 1000) time.sleep(wait_sec) def get_status_test_dapp_view(self): from views.web_views.status_test_dapp import StatusTestDAppView return StatusTestDAppView(self.driver) def get_dapp_view(self): from views.dapps_view import DappsView return DappsView(self.driver) 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_community_view(self): from views.chat_view import CommunityView return CommunityView(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_transaction_view(self): from views.transactions_view import TransactionsView return TransactionsView(self.driver) def get_wallet_view(self): from views.wallet_view import WalletView return WalletView(self.driver) def get_webview_view(self): from views.web_views.base_web_view import BaseWebView return BaseWebView(self.driver) @staticmethod def get_unique_amount(): return '0.000%s' % datetime.now().strftime('%-d%-H%-M%-S').strip('0') @staticmethod def get_random_chat_name(): return ''.join(random.choice(string.ascii_lowercase) for _ in range(7)) @staticmethod def get_random_message(): message = 'test message:' return message + ''.join(random.choice(string.ascii_lowercase) for _ in range(10)) def get_back_to_home_view(self, times_to_click_on_back_btn=3): counter = 0 while self.back_button.is_element_displayed(2) or self.navigate_up_button.is_element_displayed(2): try: if counter >= times_to_click_on_back_btn: break if self.back_button.is_element_displayed(2): self.back_button.click_until_absense_of_element(self.back_button) else: self.navigate_up_button.click_until_absense_of_element(self.navigate_up_button) counter += 1 except (NoSuchElementException, TimeoutException): continue return self def relogin(self, password=common_password): try: profile_view = self.profile_button.click() except (NoSuchElementException, TimeoutException): self.get_back_to_home_view() profile_view = self.profile_button.click() profile_view.logout() sign_in_view = self.get_sign_in_view() sign_in_view.sign_in(password) def jump_to_messages_home(self): self.hide_keyboard_if_shown() self.jump_to_button.click() self.chats_tab.click_until_presence_of_element(self.jump_to_button) def jump_to_communities_home(self): self.hide_keyboard_if_shown() self.jump_to_button.click() self.communities_tab.click_until_presence_of_element(self.jump_to_button) def jump_to_card_by_text(self, text: str): self.hide_keyboard_if_shown() self.jump_to_button.click() self.element_by_text(text).click() def reopen_app(self, password=common_password, sign_in=True): self.driver.close_app() self.driver.launch_app() if sign_in: sign_in_view = self.get_sign_in_view() sign_in_view.sign_in(password) def close_share_popup(self): self.driver.info("Closing share popup") TouchAction(self.driver).tap(None, 255, 104, 1).perform() time.sleep(3) def get_public_key(self): self.driver.info("Get public key") profile_view = self.get_profile_view() self.profile_button.click_until_presence_of_element(profile_view.default_username_text) profile_view.share_my_profile_button.click() profile_view.public_key_text.wait_for_visibility_of_element(20) public_key = profile_view.public_key_text.text self.click_system_back_button() return public_key def tap_mutual_cr_switcher(self): profile_view = self.profile_button.click() profile_view.advanced_button.scroll_and_click() profile_view.mutual_contact_request_switcher.scroll_and_click() profile_view.click_system_back_button() def share_via_messenger(self): self.driver.info("Sharing via messenger", device=False) self.element_by_text('Messages').wait_for_visibility_of_element(40) self.element_by_text('Messages').click_until_presence_of_element(self.element_by_text('New message')) self.element_by_text('New message').wait_and_click() self.send_as_keyevent('+0100100101') self.confirm() def click_upon_push_notification_by_text(self, text): element = self.element_by_text_part(text) self.driver.info("Click on PN with text: '%s'" % element.exclude_emoji(text)) element.click() return self.get_chat_view() def find_values_in_logcat(self, **kwargs): logcat = self.logcat items_in_logcat = list() for key, value in kwargs.items(): self.driver.info("Checking in logcat for: `%s`" % value) escaped_value = re.escape(value) if re.findall(r'\W%s$|\W%s\W' % (escaped_value, escaped_value), logcat): items_in_logcat.append('%s in logcat!!!' % key.capitalize()) return items_in_logcat def find_values_in_geth(self, *args): from tests.base_test_case import AbstractTestCase b64_log = self.driver.pull_file(AbstractTestCase().geth_path) file = base64.b64decode(b64_log) result = False for value in args: self.driver.info('Checking in geth for: `%s`' % value) if re.findall('%s*' % value, file.decode("utf-8")): self.driver.info('%s was found in geth.log' % value) result = True return result def asset_by_name(self, asset_name): return AssetButton(self.driver, asset_name) def open_notification_bar(self): self.driver.open_notifications() def toggle_airplane_mode(self): self.driver.info("Toggling airplane mode") self.airplane_mode_button.click() self.close_native_device_dialog("MmsService") def set_device_to_offline(self): # setting network connection to data only and switching off wifi self.driver.set_network_connection(2) self.driver.toggle_wifi() def set_network_to_cellular_only(self): self.driver.set_network_connection(4) def toggle_mobile_data(self): self.driver.info("Toggling mobile data") self.driver.start_activity(app_package='com.android.settings', app_activity='.Settings') network_and_internet = self.element_by_text('Network & internet') network_and_internet.wait_for_visibility_of_element() network_and_internet.click() toggle = Button(self.driver, accessibility_id='Wi‑Fi') toggle.wait_for_visibility_of_element() toggle.click() self.driver.back() self.driver.back() def open_universal_web_link(self, deep_link): start_web_browser(self.driver) self.driver.info('Open web link via web browser: `%s`' % deep_link) self.driver.get(deep_link) def upgrade_app(self): self.driver.info("Upgrading apk to apk_upgrade") self.driver.install_app(pytest_config_global['apk_upgrade'], replace=True) self.app = self.driver.launch_app() def search_by_keyword(self, keyword): self.driver.info('Search for `%s`' % keyword) self.search_input.click() self.search_input.send_keys(keyword) def set_up_wallet_when_sending_tx(self): self.driver.info("Setting up wallet") phrase = self.sign_in_phrase.text self.ok_got_it_button.wait_and_click(20) return phrase def get_empty_dapp_tab(self): from views.web_views.base_web_view import BaseWebView web_view = BaseWebView(self.driver) web_view.options_button.click() if web_view.new_tab_button.is_element_displayed(): web_view.new_tab_button.click() return web_view def get_test_assets(self, token=False, keycard=False): from views.home_view import HomeView status_test_dapp = HomeView(self.driver).open_status_test_dapp() status_test_dapp.wait_for_d_aap_to_load() self.just_fyi("Requesting test assets in dapp") status_test_dapp.assets_button.click() if token: send_tx = status_test_dapp.request_stt_button.click() send_tx.sign_transaction(keycard=keycard) wallet = self.wallet_button.click() wallet.wait_balance_is_changed(asset='STT', scan_tokens=True) else: status_test_dapp.request_eth_button.click() status_test_dapp.ok_button.wait_and_click() wallet = self.wallet_button.click() wallet.wait_balance_is_changed() return wallet def donate_leftovers(self, keycard=False): self.just_fyi("Send leftovers from test accounts") wallet = self.wallet_button.click() self.wallet_button.click() send_transaction = wallet.send_transaction_from_main_screen.click() send_transaction.set_max_button.click() send_transaction.confirm() send_transaction.chose_recipient_button.click() send_transaction.set_recipient_address('0xE2363E6e91d1a29d82C2c695fa8fa2e3Fa5d55eA') send_transaction.sign_transaction_button.click() send_transaction.sign_transaction(keycard=keycard) def tap_by_coordinates(self, x, y): action = TouchAction(self.driver) action.press(None, x, y).release().perform() # Method-helper def write_page_source_to_file(self, full_path_to_file): string_source = self.driver.page_source source = open(full_path_to_file, "a+") source.write(string_source) def wait_for_current_package_to_be(self, expected_package_name: str, timeout: int = 10): start_time = time.time() while time.time() - start_time <= timeout: package = self.driver.current_package if package == expected_package_name: return time.sleep(1) raise TimeoutException("Driver current package is '%s' after %s seconds" % (package, timeout))