55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
import logging
|
|
|
|
from selenium.common.exceptions import TimeoutException
|
|
|
|
from views.base_element import BaseButton
|
|
from views.base_view import BaseView
|
|
|
|
|
|
class PlusButton(BaseButton):
|
|
def __init__(self, driver):
|
|
super(PlusButton, self).__init__(driver)
|
|
self.locator = self.Locator.xpath_selector(
|
|
"//android.widget.TextView[@text='+']")
|
|
|
|
def navigate(self):
|
|
from views.start_new_chat_view import StarNewChatView
|
|
return StarNewChatView(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 HomeView(BaseView):
|
|
def __init__(self, driver):
|
|
super(HomeView, self).__init__(driver)
|
|
|
|
self.plus_button = PlusButton(self.driver)
|
|
self.console_button = ConsoleButton(self.driver)
|
|
|
|
def wait_for_syncing_complete(self):
|
|
logging.info('Waiting for syncing complete:')
|
|
while True:
|
|
try:
|
|
sync = self.find_text_part('Syncing', 10)
|
|
logging.info(sync.text)
|
|
except TimeoutException:
|
|
break
|
|
|
|
def get_chat_with_user(self, username):
|
|
return ChatElement(self.driver, username)
|