import time from selenium.common.exceptions import TimeoutException, NoSuchElementException from views.base_element import Button, Text, BaseElement, SilentButton from views.base_view import BaseView from tests import test_dapp_url class ChatButton(Button): def __init__(self, driver, **kwargs): super().__init__(driver, **kwargs) def navigate(self): from views.chat_view import ChatView return ChatView(self.driver) class ChatElement(SilentButton): def __init__(self, driver, username_part): self.username = username_part super().__init__(driver, xpath="//*[@content-desc='chat-name-text'][starts-with(@text,'%s')]/.." % username_part) def navigate(self): from views.chat_view import ChatView return ChatView(self.driver) def click(self): from views.chat_view import ChatView desired_element = ChatView(self.driver).chat_message_input self.click_until_presence_of_element(desired_element=desired_element) return self.navigate() def find_element(self): for i in range(2): try: return super(ChatElement, self).find_element() except NoSuchElementException as e: if i == 0: HomeView(self.driver).reconnect() else: e.msg = 'Device %s: Unable to find chat with user %s' % (self.driver.number, self.username) raise e @property def new_messages_counter(self): class UnreadMessagesCountText(Text): def __init__(self, driver, parent_locator: str): super().__init__(driver, xpath="(%s//android.widget.TextView)[last()]" % parent_locator) return UnreadMessagesCountText(self.driver, self.locator) @property def new_messages_public_chat(self): class UnreadMessagesPublicChat(BaseElement): def __init__(self, driver): super().__init__(driver, accessibility_id="unviewed-messages-public") return UnreadMessagesPublicChat(self.driver) class HomeView(BaseView): def __init__(self, driver): super().__init__(driver) self.plus_button = Button(self.driver, accessibility_id="new-chat-button") self.chat_name_text = Text(self.driver, accessibility_id="chat-name-text") self.start_new_chat_button = ChatButton(self.driver, accessibility_id="start-1-1-chat-button") self.new_group_chat_button = ChatButton(self.driver, accessibility_id="start-group-chat-button") self.join_public_chat_button = ChatButton(self.driver, accessibility_id="join-public-chat-button") self.universal_qr_scanner_button = Button(self.driver, accessibility_id="universal-qr-scanner") self.invite_friends_button = Button(self.driver, accessibility_id="invite-friends-button") self.stop_status_service_button = Button(self.driver, accessibility_id="STOP") # Options on long tap self.chats_menu_invite_friends_button = Button(self.driver, accessibility_id="chats-menu-invite-friends-button") self.delete_chat_button = Button(self.driver, accessibility_id="delete-chat-button") self.clear_history_button = Button(self.driver, accessibility_id="clear-history-button") self.mark_all_messages_as_read_button = Button(self.driver, accessibility_id="mark-all-read-button") def wait_for_syncing_complete(self): self.driver.info('**Waiting for syncing complete:**') while True: try: sync = self.find_text_part('Syncing', 10) self.driver.info(sync.text) except TimeoutException: break def get_chat(self, username): self.driver.info("**Looking for chat '%s'**" % username) return ChatElement(self.driver, username[:25]) def get_username_below_start_new_chat_button(self, username_part): return Text(self.driver, xpath="//*[@content-desc='enter-contact-code-input']/../..//*[starts-with(@text,'%s')]" % username_part) def add_contact(self, public_key, add_in_contacts=True, nickname=''): self.driver.info("**Starting 1-1 chat, add in contacts:%s**" % str(add_in_contacts)) self.plus_button.click_until_presence_of_element(self.start_new_chat_button) contacts_view = self.start_new_chat_button.click() contacts_view.public_key_edit_box.click() contacts_view.public_key_edit_box.send_keys(public_key) one_to_one_chat = self.get_chat_view() contacts_view.confirm_until_presence_of_element(one_to_one_chat.chat_message_input) if add_in_contacts: one_to_one_chat.add_to_contacts.click() if nickname: self.driver.info("**Setting nickname '%s'**" % nickname) one_to_one_chat.chat_options.click() one_to_one_chat.view_profile_button.click() one_to_one_chat.set_nickname(nickname) one_to_one_chat.back_button.click() self.driver.info("**1-1 chat is created successfully!**") return one_to_one_chat def create_group_chat(self, user_names_to_add: list, group_chat_name: str = 'new_group_chat'): self.driver.info("**Creating group chat '%s'**" % group_chat_name) self.plus_button.click() chat_view = self.new_group_chat_button.click() if user_names_to_add: for user_name in user_names_to_add: if len(user_names_to_add) > 5: chat_view.search_by_keyword(user_name[:4]) chat_view.get_username_checkbox(user_name).click() chat_view.search_input.clear() else: chat_view.get_username_checkbox(user_name).click() chat_view.next_button.click() chat_view.chat_name_editbox.send_keys(group_chat_name) chat_view.create_button.click() self.driver.info("**Group chat %s is created successfully!**" % group_chat_name) return chat_view def join_public_chat(self, chat_name: str): self.driver.info("**Creating group chat %s**" % chat_name) self.plus_button.click_until_presence_of_element(self.join_public_chat_button, attempts=5) self.join_public_chat_button.wait_for_visibility_of_element(5) chat_view = self.join_public_chat_button.click() chat_view.chat_name_editbox.click() chat_view.chat_name_editbox.send_keys(chat_name) time.sleep(2) self.confirm_until_presence_of_element(chat_view.chat_message_input) self.driver.info("**Public chat %s is created successfully!**" % chat_name) return chat_view def open_status_test_dapp(self, url=test_dapp_url, allow_all=True): self.driver.info("**Open dapp '%s', allow all:%s**" % (test_dapp_url, str(allow_all))) dapp_view = self.dapp_tab_button.click() dapp_view.open_url(url) status_test_dapp = dapp_view.get_status_test_dapp_view() status_test_dapp.allow_button.wait_for_element(20) if allow_all: status_test_dapp.allow_button.click_until_absense_of_element(status_test_dapp.allow_button) else: status_test_dapp.deny_button.click_until_absense_of_element(status_test_dapp.deny_button) self.driver.info("**Dapp is opened!**") return status_test_dapp def delete_chat_long_press(self, username): self.driver.info("**Deleting chat %s by long press**" % username) self.get_chat(username).long_press_element() self.delete_chat_button.click() self.delete_button.click() def leave_chat_long_press(self, username): self.driver.info("**Leaving chat %s by long press**" % username) self.get_chat(username).long_press_element() from views.chat_view import ChatView ChatView(self.driver).leave_chat_button.click() ChatView(self.driver).leave_button.click() def clear_chat_long_press(self, username): self.driver.info("**Clearing history in chat %s by long press**" % username) self.get_chat(username).long_press_element() self.clear_history_button.click() from views.chat_view import ChatView ChatView(self.driver).clear_button.click()