102 lines
3.5 KiB
Python
102 lines
3.5 KiB
Python
from tests import info
|
|
import time
|
|
from selenium.common.exceptions import TimeoutException, NoSuchElementException
|
|
from views.base_element import BaseButton, BaseText
|
|
from views.base_view import BaseView
|
|
|
|
|
|
class PlusButton(BaseButton):
|
|
def __init__(self, driver):
|
|
super(PlusButton, self).__init__(driver)
|
|
self.locator = self.Locator.xpath_selector(
|
|
"//*[@text='+']")
|
|
|
|
def navigate(self):
|
|
from views.start_new_chat_view import StartNewChatView
|
|
return StartNewChatView(self.driver)
|
|
|
|
|
|
class ConsoleButton(BaseButton):
|
|
def __init__(self, driver):
|
|
super(ConsoleButton, self).__init__(driver)
|
|
self.locator = self.Locator.xpath_selector(
|
|
"//*[@text='Console']")
|
|
|
|
|
|
class ChatElement(BaseButton):
|
|
def __init__(self, driver, username):
|
|
super(ChatElement, self).__init__(driver)
|
|
self.locator = self.Locator.xpath_selector("//*[@text='%s']" % username)
|
|
|
|
def navigate(self):
|
|
from views.chat_view import ChatView
|
|
return ChatView(self.driver)
|
|
|
|
|
|
class FirstChatElementTitle(BaseText):
|
|
def __init__(self, driver):
|
|
super(FirstChatElementTitle, self).__init__(driver)
|
|
self.locator = self.Locator.xpath_selector('(//android.widget.ScrollView//android.widget.TextView)[1]')
|
|
|
|
|
|
class HomeView(BaseView):
|
|
def __init__(self, driver):
|
|
super(HomeView, self).__init__(driver)
|
|
|
|
self.plus_button = PlusButton(self.driver)
|
|
self.console_button = ConsoleButton(self.driver)
|
|
self.first_chat_element_title = FirstChatElementTitle(self.driver)
|
|
|
|
def wait_for_syncing_complete(self):
|
|
info('Waiting for syncing complete:')
|
|
while True:
|
|
try:
|
|
sync = self.find_text_part('Syncing', 10)
|
|
info(sync.text)
|
|
except TimeoutException:
|
|
break
|
|
|
|
def get_chat_with_user(self, username):
|
|
return ChatElement(self.driver, username)
|
|
|
|
def get_back_to_home_view(self):
|
|
counter = 0
|
|
while not self.home_button.is_element_present():
|
|
try:
|
|
if counter >= 5:
|
|
return
|
|
self.back_button.click()
|
|
except (NoSuchElementException, TimeoutException):
|
|
counter += 1
|
|
|
|
def add_contact(self, public_key):
|
|
start_new_chat = self.plus_button.click()
|
|
start_new_chat.add_new_contact.click()
|
|
start_new_chat.public_key_edit_box.send_keys(public_key)
|
|
start_new_chat.confirm()
|
|
|
|
def create_group_chat(self, user_names_to_add: list, group_chat_name: str = 'new_group_chat'):
|
|
start_new_chat = self.plus_button.click()
|
|
start_new_chat.new_group_chat_button.click()
|
|
for user_name in user_names_to_add:
|
|
user_contact = start_new_chat.get_username_checkbox(user_name)
|
|
user_contact.scroll_to_element()
|
|
user_contact.click()
|
|
start_new_chat.next_button.click()
|
|
start_new_chat.chat_name_editbox.send_keys(group_chat_name)
|
|
start_new_chat.confirm_button.click()
|
|
|
|
def join_public_chat(self, chat_name: str):
|
|
start_new_chat = self.plus_button.click()
|
|
start_new_chat.join_public_chat_button.click()
|
|
start_new_chat.chat_name_editbox.send_keys(chat_name)
|
|
start_new_chat.confirm()
|
|
|
|
def get_public_key(self):
|
|
profile_view = self.profile_button.click()
|
|
profile_view.share_my_contact_key_button.click()
|
|
time.sleep(4)
|
|
public_key = profile_view.get_text_from_qr()
|
|
profile_view.cross_icon.click()
|
|
return public_key
|