540 lines
23 KiB
Python
540 lines
23 KiB
Python
import pathlib
|
|
import time
|
|
import typing
|
|
from typing import List
|
|
|
|
import allure
|
|
|
|
import configs
|
|
import driver
|
|
from driver.objects_access import walk_children
|
|
from gui.components.activity_center import ActivityCenter
|
|
from gui.components.context_menu import ContextMenu
|
|
from gui.components.delete_popup import DeleteMessagePopup
|
|
from gui.components.emoji_popup import EmojiPopup
|
|
from gui.components.messaging.clear_chat_history_popup import ClearChatHistoryPopup
|
|
from gui.components.messaging.close_chat_popup import CloseChatPopup
|
|
from gui.components.messaging.edit_group_name_and_image_popup import EditGroupNameAndImagePopup
|
|
from gui.components.messaging.leave_group_popup import LeaveGroupPopup
|
|
from gui.components.messaging.link_preview_options_popup import LinkPreviewOptionsPopup
|
|
from gui.components.messaging.message_context_menu_popup import MessageContextMenuPopup
|
|
from gui.elements.button import Button
|
|
from gui.elements.list import List
|
|
from gui.elements.object import QObject
|
|
from gui.elements.scroll import Scroll
|
|
from gui.elements.text_edit import TextEdit
|
|
from gui.elements.text_label import TextLabel
|
|
from gui.objects_map import messaging_names, communities_names
|
|
from gui.screens.community import CommunityScreen
|
|
from scripts.tools.image import Image
|
|
|
|
|
|
class LeftPanel(QObject):
|
|
|
|
def __init__(self):
|
|
super().__init__(messaging_names.mainWindow_contactColumnLoader_Loader)
|
|
self._start_chat_button = Button(messaging_names.mainWindow_startChatButton_StatusIconTabButton)
|
|
self._search_text_edit = TextEdit(messaging_names.mainWindow_search_edit_TextEdit)
|
|
self._scroll = Scroll(messaging_names.scrollView_Flickable)
|
|
self._chats_list = List(messaging_names.chatList_ListView)
|
|
self._chat_list_item = QObject(messaging_names.scrollView_StatusChatListItem)
|
|
self._chats_scroll = QObject(messaging_names.mainWindow_scrollView_StatusScrollView)
|
|
|
|
@property
|
|
@allure.step('Get chats by chats list')
|
|
def get_chats_names(self) -> typing.List[str]:
|
|
chats_list = []
|
|
for child in walk_children(driver.waitForObjectExists(self._chats_list.real_name)):
|
|
if getattr(child, 'id', '') == 'statusChatListItem':
|
|
chats_list.append(str(child.objectName))
|
|
return chats_list
|
|
|
|
@allure.step('Click chat item')
|
|
def click_chat_by_name(self, chat_name: str):
|
|
self._chat_list_item.real_name['objectName'] = chat_name
|
|
self._chat_list_item.click()
|
|
return ChatView()
|
|
|
|
@allure.step('Click start chat button')
|
|
def start_chat(self):
|
|
self._start_chat_button.click(x=1, y=1)
|
|
return CreateChatView()
|
|
|
|
@allure.step('Open context menu group chat')
|
|
def _open_context_menu_for_chat(self, chat_name: str) -> ContextMenu:
|
|
self._chat_list_item.real_name['objectName'] = chat_name
|
|
self._chat_list_item.right_click()
|
|
return ContextMenu().wait_until_appears()
|
|
|
|
@allure.step('Open leave popup')
|
|
def open_leave_group_popup(self, chat_name: str, attempt: int = 2) -> LeaveGroupPopup:
|
|
try:
|
|
self._open_context_menu_for_chat(chat_name).select('Leave group')
|
|
return LeaveGroupPopup().wait_until_appears()
|
|
except Exception as ex:
|
|
if attempt:
|
|
return self.open_leave_group_popup(chat_name, attempt - 1)
|
|
else:
|
|
raise ex
|
|
|
|
|
|
class ToolBar(QObject):
|
|
|
|
def __init__(self):
|
|
super().__init__(messaging_names.mainWindow_statusToolBar_StatusToolBar)
|
|
self.pinned_message_tooltip = QObject(
|
|
communities_names.statusToolBar_StatusChatInfo_pinText_TruncatedTextWithTooltip)
|
|
self.confirm_button = Button(messaging_names.statusToolBar_Confirm_StatusButton)
|
|
self.status_button = Button(messaging_names.statusToolBar_Cancel_StatusButton)
|
|
self.contact_tag = QObject(messaging_names.statusToolBar_StatusTagItem)
|
|
self.notifications_button = Button(messaging_names.statusToolBar_notificationButton_StatusActivityCenterButton)
|
|
|
|
@property
|
|
@allure.step('Get visibility of pin message tooltip')
|
|
def is_pin_message_tooltip_visible(self) -> bool:
|
|
return self.pinned_message_tooltip.is_visible
|
|
|
|
@allure.step('Click on pin message tooltip')
|
|
def click_pin_message_tooltip(self):
|
|
return self.pinned_message_tooltip.click()
|
|
|
|
@allure.step('Confirm action in toolbar')
|
|
def confirm_action_in_toolbar(self):
|
|
self.confirm_button.click()
|
|
|
|
@allure.step('Remove member by clicking close icon on member tag')
|
|
def click_contact_close_icon(self, member):
|
|
for item in driver.findAllObjects(self.contact_tag.real_name):
|
|
if str(getattr(item, 'text', '')) == str(member):
|
|
for child in walk_children(item):
|
|
if getattr(child, 'objectName', '') == 'close-icon':
|
|
driver.mouseClick(child)
|
|
break
|
|
|
|
@allure.step('Open activity center')
|
|
def open_activity_center(self):
|
|
self.notifications_button.click()
|
|
return ActivityCenter().wait_until_appears()
|
|
|
|
|
|
class Message:
|
|
|
|
def __init__(self, obj):
|
|
self.object = obj
|
|
self.date: typing.Optional[str] = None
|
|
self.time: typing.Optional[str] = None
|
|
self.icon: typing.Optional[Image] = None
|
|
self.from_user: typing.Optional[str] = None
|
|
self.text: typing.Optional[str] = None
|
|
self.delegate_button: typing.Optional[Button] = None
|
|
self.reply_corner: typing.Optional[QObject] = None
|
|
self.link_preview: typing.Optional[QObject] = None
|
|
self.link_preview_title_object: typing.Optional[QObject] = None
|
|
self.link_preview_emoji_hash: typing.Optional[str] = None
|
|
self.image_message: typing.Optional[QObject] = None
|
|
self.banner_image: typing.Optional[QObject] = None
|
|
self.community_invitation: dict = {}
|
|
self.init_ui()
|
|
|
|
def init_ui(self):
|
|
for child in walk_children(self.object):
|
|
if getattr(child, 'objectName', '') == 'StatusDateGroupLabel':
|
|
self.date = str(child.text)
|
|
elif getattr(child, 'id', '') == 'title':
|
|
self.community_invitation['name'] = str(child.text)
|
|
elif getattr(child, 'id', '') == 'description':
|
|
self.community_invitation['description'] = str(child.text)
|
|
elif getattr(child, 'id', '') == 'titleLayout':
|
|
self.link_preview_title_object = child
|
|
elif getattr(child, 'objectName', '') == 'linkPreviewEmojiHash':
|
|
self.link_preview_emoji_hash = str(child.publicKey)
|
|
else:
|
|
match getattr(child, 'id', ''):
|
|
case 'profileImage':
|
|
self.icon = Image(driver.objectMap.realName(child))
|
|
case 'primaryDisplayName':
|
|
self.from_user = str(child.text)
|
|
case 'timestampText':
|
|
self.time = str(child.text)
|
|
case 'chatText':
|
|
self.text = str(child.text)
|
|
case 'replyCorner':
|
|
self.reply_corner = QObject(real_name=driver.objectMap.realName(child))
|
|
case 'delegate':
|
|
self.delegate_button = Button(real_name=driver.objectMap.realName(child))
|
|
case 'linksMessageView':
|
|
self.link_preview = QObject(real_name=driver.objectMap.realName(child))
|
|
case 'imageMessage':
|
|
self.image_message = child
|
|
case 'bannerImage':
|
|
self.banner_image = QObject(real_name=driver.objectMap.realName(child))
|
|
|
|
@allure.step('Open community invitation')
|
|
def open_community_invitation(self):
|
|
driver.waitFor(lambda: self.delegate_button.is_visible, configs.timeouts.UI_LOAD_TIMEOUT_MSEC)
|
|
self.delegate_button.click()
|
|
return CommunityScreen().wait_until_appears()
|
|
|
|
@allure.step('Hover message')
|
|
def hover_message(self):
|
|
self.delegate_button.hover()
|
|
return MessageQuickActions()
|
|
|
|
@allure.step('Get color of message background')
|
|
def get_message_color(self) -> str:
|
|
return self.delegate_button.object.background.color.name
|
|
|
|
@property
|
|
@allure.step('Get user name in pinned message details')
|
|
def user_name_in_pinned_message(self) -> str:
|
|
return str(self.delegate_button.object.pinnedBy)
|
|
|
|
@property
|
|
@allure.step('Get info text in pinned message details')
|
|
def pinned_info_text(self) -> str:
|
|
return str(self.delegate_button.object.pinnedMsgInfoText)
|
|
|
|
@property
|
|
@allure.step('Get message pinned state')
|
|
def message_is_pinned(self) -> bool:
|
|
return self.delegate_button.object.isPinned
|
|
|
|
@allure.step('Get title of link preview')
|
|
def get_link_preview_title(self) -> str:
|
|
for child in walk_children(self.link_preview_title_object):
|
|
if getattr(child, 'objectName', '') == 'linkPreviewTitle':
|
|
return str(child.text)
|
|
|
|
@allure.step('Get link domain from message')
|
|
def get_link_domain(self) -> str:
|
|
return self.delegate_button.object.linkData.domain
|
|
|
|
@allure.step('Open context menu for message')
|
|
def open_context_menu_for_message(self):
|
|
QObject(real_name=driver.objectMap.realName(self.object)).right_click()
|
|
return MessageContextMenuPopup().wait_until_appears()
|
|
|
|
@allure.step('Get emoji reactions pathes')
|
|
def get_emoji_reactions_pathes(self):
|
|
reactions_pathes = []
|
|
for child in walk_children(self.object):
|
|
if getattr(child, 'id', '') == 'reactionDelegate':
|
|
for item in walk_children(child):
|
|
if getattr(item, 'objectName', '') == 'emojiReaction':
|
|
reactions_pathes.append(item.source.path)
|
|
return reactions_pathes
|
|
|
|
|
|
class ChatView(QObject):
|
|
|
|
def __init__(self):
|
|
super().__init__(messaging_names.mainWindow_ChatColumnView)
|
|
self._message_list_item = QObject(messaging_names.chatLogView_chatMessageViewDelegate_MessageView)
|
|
self._deleted_message = QObject(messaging_names.chatMessageViewDelegate_deletedMessage_RowLayout)
|
|
|
|
@allure.step('Get messages')
|
|
def messages(self, index: int) -> typing.List[Message]:
|
|
_messages = []
|
|
time.sleep(1)
|
|
# message_list_item has different indexes if we run multiple instances, so we pass index
|
|
if index is not None:
|
|
self._message_list_item.real_name['index'] = index
|
|
for item in driver.findAllObjects(self._message_list_item.real_name):
|
|
if getattr(item, 'isMessage', False):
|
|
_messages.append(Message(item))
|
|
return _messages
|
|
|
|
@allure.step('Get deleted message state')
|
|
def get_deleted_message_state(self):
|
|
return self._deleted_message.exists
|
|
|
|
def find_message_by_text(self, message_text: str, index: int):
|
|
message = None
|
|
started_at = time.monotonic()
|
|
while message is None:
|
|
for _message in self.messages(index):
|
|
if message_text in _message.text:
|
|
message = _message
|
|
break
|
|
if time.monotonic() - started_at > configs.timeouts.MESSAGING_TIMEOUT_SEC:
|
|
raise LookupError(f'Message not found')
|
|
return message
|
|
|
|
@allure.step('Accept community invitation')
|
|
def accept_community_invite(self, community: str, index: int) -> 'CommunityScreen':
|
|
message = None
|
|
started_at = time.monotonic()
|
|
while message is None:
|
|
for _message in self.messages(index):
|
|
if _message.community_invitation.get('name', '') == community:
|
|
message = _message
|
|
break
|
|
if time.monotonic() - started_at > 80:
|
|
raise LookupError(f'Community invitation was not found')
|
|
|
|
return message.open_community_invitation()
|
|
|
|
|
|
class CreateChatView(QObject):
|
|
|
|
def __init__(self):
|
|
super().__init__(messaging_names.mainWindow_CreateChatView)
|
|
self._confirm_button = Button(messaging_names.createChatView_confirmBtn)
|
|
self._cancel_button = Button(messaging_names.mainWindow_Cancel_StatusButton)
|
|
self._create_chat_contacts_list = List(messaging_names.createChatView_contactsList)
|
|
|
|
@property
|
|
@allure.step('Get contacts')
|
|
def contacts(self) -> typing.List[str]:
|
|
return self._create_chat_contacts_list.get_values('title')
|
|
|
|
@allure.step('Select contact in the list')
|
|
def select_contact(self, contact: str):
|
|
assert driver.waitFor(lambda: contact in self.contacts), f'Contact: {contact} not found in {self.contacts}'
|
|
self._create_chat_contacts_list.select(contact, 'title')
|
|
|
|
@allure.step('Create chat by adding contacts from contact list')
|
|
def create_chat(self, members):
|
|
for member in members[0:]:
|
|
time.sleep(0.2)
|
|
self.select_contact(member)
|
|
self._confirm_button.click()
|
|
return ChatMessagesView().wait_until_appears()
|
|
|
|
|
|
class ChatMessagesView(QObject):
|
|
|
|
def __init__(self):
|
|
super().__init__(messaging_names.mainWindow_ChatMessagesView)
|
|
self._group_chat_message_item = TextLabel(messaging_names.chatLogView_Item)
|
|
self._group_name_label = TextLabel(messaging_names.statusChatInfoButton)
|
|
self._more_button = Button(messaging_names.moreOptionsButton_StatusFlatRoundButton)
|
|
self._edit_menu_item = QObject(messaging_names.edit_name_and_image_StatusMenuItem)
|
|
self._leave_group_item = QObject(messaging_names.leave_group_StatusMenuItem)
|
|
self._add_remove_item = QObject(messaging_names.add_remove_from_group_StatusMenuItem)
|
|
self._clear_history_item = QObject(messaging_names.clear_History_StatusMenuItem)
|
|
self._close_chat_item = QObject(messaging_names.close_Chat_StatusMenuItem)
|
|
self._chat_input = QObject(messaging_names.mainWindow_statusChatInput_StatusChatInput)
|
|
self._message_input_area = QObject(messaging_names.inputScrollView_messageInputField_TextArea)
|
|
self._message_field = TextEdit(messaging_names.inputScrollView_Message_PlaceholderText)
|
|
self._emoji_button = Button(messaging_names.mainWindow_statusChatInputEmojiButton_StatusFlatRoundButton)
|
|
self._image_button = Button(messaging_names.mainWindow_imageBtn_StatusFlatRoundButton)
|
|
self._link_preview_title = QObject(messaging_names.mainWindow_linkPreviewTitleText_StatusBaseText)
|
|
self._link_preview_preview_subtitle = QObject(messaging_names.mainWindow_linkPreviewSubtitleText_StatusBaseText)
|
|
self._link_preview_show_preview = QObject(messaging_names.mainWindow_titleText_StatusBaseText)
|
|
self._link_preview_show_description = QObject(messaging_names.mainWindow_subtitleText_StatusBaseText)
|
|
self._link_preview_card = QObject(messaging_names.mainWindow_settingsCard_LinkPreviewSettingsCard)
|
|
self._options_combobox = QObject(messaging_names.mainWindow_optionsComboBox_ComboBox)
|
|
self._close_preview_button = QObject(messaging_names.mainWindow_closeLinkPreviewButton_StatusFlatRoundButton)
|
|
|
|
@property
|
|
@allure.step('Get group name')
|
|
def group_name(self) -> str:
|
|
return self._group_name_label.text
|
|
|
|
@property
|
|
@allure.step('Get group welcome message')
|
|
def group_welcome_message(self) -> str:
|
|
for delegate in walk_children(self._group_chat_message_item.object):
|
|
if getattr(delegate, 'id', '') == 'msgDelegate':
|
|
for item in walk_children(delegate):
|
|
if getattr(item, 'id', '') == 'descText':
|
|
return str(item.text)
|
|
|
|
@property
|
|
@allure.step('Get gray text from message area')
|
|
def gray_text_from_message_area(self) -> str:
|
|
return driver.waitForObjectExists(self._message_input_area.real_name,
|
|
configs.timeouts.UI_LOAD_TIMEOUT_MSEC).placeholderText
|
|
|
|
@property
|
|
@allure.step('Get enabled state of message area')
|
|
def is_message_area_enabled(self) -> bool:
|
|
return driver.waitForObjectExists(self._message_input_area.real_name,
|
|
configs.timeouts.UI_LOAD_TIMEOUT_MSEC).enabled
|
|
|
|
@allure.step('Click more options button')
|
|
def open_more_options(self):
|
|
self._more_button.click()
|
|
|
|
@allure.step('Choose edit group name option')
|
|
def open_edit_group_name_form(self):
|
|
time.sleep(2)
|
|
self.open_more_options()
|
|
time.sleep(2)
|
|
self._edit_menu_item.click()
|
|
return EditGroupNameAndImagePopup().wait_until_appears()
|
|
|
|
@allure.step('Choose leave group option')
|
|
def leave_group(self):
|
|
time.sleep(2)
|
|
self.open_more_options()
|
|
time.sleep(2)
|
|
self._leave_group_item.click()
|
|
return LeaveGroupPopup().wait_until_appears()
|
|
|
|
@allure.step('Send message to group chat')
|
|
def send_message_to_group_chat(self, message: str):
|
|
self.type_message(message)
|
|
self.confirm_sending_message()
|
|
|
|
@allure.step('Type text to message field')
|
|
def type_message(self, message: str):
|
|
self._message_field.type_text(message)
|
|
|
|
@allure.step('Confirm sending message')
|
|
def confirm_sending_message(self):
|
|
self._message_input_area.click()
|
|
for i in range(2):
|
|
driver.nativeType('<Return>')
|
|
|
|
@allure.step('Click options combobox')
|
|
def click_options(self):
|
|
self._options_combobox.click()
|
|
return LinkPreviewOptionsPopup().wait_until_appears()
|
|
|
|
@allure.step('Close link preview popup by clicking preview bubble area')
|
|
def close_link_preview_popup(self):
|
|
self._link_preview_card.click()
|
|
LinkPreviewOptionsPopup().wait_until_hidden()
|
|
return self
|
|
|
|
@allure.step('Get text of title of link preview bubble')
|
|
def get_link_preview_bubble_title(self) -> str:
|
|
return str(self._link_preview_title.object.text)
|
|
|
|
@allure.step('Get text of description of link preview bubble')
|
|
def get_link_preview_bubble_description(self) -> str:
|
|
return str(self._link_preview_preview_subtitle.object.text)
|
|
|
|
@allure.step('Get text of title of show link preview bubble')
|
|
def get_show_link_preview_bubble_title(self) -> str:
|
|
return str(self._link_preview_show_preview.object.text)
|
|
|
|
@allure.step('Get text of description of show link preview bubble')
|
|
def get_show_link_preview_bubble_description(self) -> str:
|
|
return str(self._link_preview_show_description.object.text)
|
|
|
|
@allure.step('Get close button visibility state')
|
|
def does_close_button_exist(self) -> bool:
|
|
return self._close_preview_button.is_visible
|
|
|
|
@allure.step('Send emoji to chat')
|
|
def send_emoji_to_chat(self, emoji: str):
|
|
self._emoji_button.click()
|
|
EmojiPopup().wait_until_appears().select(emoji)
|
|
self.send_message()
|
|
|
|
@allure.step('Send image to chat')
|
|
def send_image_to_chat(self, path):
|
|
self.choose_image(path)
|
|
self.send_message()
|
|
|
|
@allure.step('Choose image')
|
|
def choose_image(self, path):
|
|
fileuri = pathlib.Path(str(path)).as_uri()
|
|
self._chat_input.object.selectImageString(fileuri)
|
|
|
|
@allure.step('Confirm sending message')
|
|
def send_message(self):
|
|
for i in range(2):
|
|
driver.nativeType('<Return>')
|
|
|
|
@allure.step('Remove member from chat')
|
|
def remove_member_from_chat(self, member):
|
|
time.sleep(2)
|
|
self.open_more_options()
|
|
time.sleep(2)
|
|
self._add_remove_item.click()
|
|
tool_bar = ToolBar().wait_until_appears()
|
|
tool_bar.click_contact_close_icon(member)
|
|
time.sleep(1)
|
|
tool_bar.confirm_action_in_toolbar()
|
|
time.sleep(1)
|
|
|
|
@allure.step('Clear chat history option')
|
|
def clear_history(self):
|
|
time.sleep(2)
|
|
self.open_more_options()
|
|
time.sleep(2)
|
|
self._clear_history_item.click()
|
|
ClearChatHistoryPopup().wait_until_appears().confirm_clearing_chat()
|
|
|
|
@allure.step('Close chat')
|
|
def close_chat(self):
|
|
time.sleep(2)
|
|
self.open_more_options()
|
|
time.sleep(2)
|
|
self._close_chat_item.click()
|
|
CloseChatPopup().wait_until_appears().confirm_closing_chat()
|
|
|
|
|
|
class MessageQuickActions(QObject):
|
|
def __init__(self):
|
|
super().__init__(messaging_names.chatMessageViewDelegate_StatusMessageQuickActions)
|
|
self._pin_button = Button(
|
|
messaging_names.chatMessageViewDelegate_pin_icon_StatusIcon)
|
|
self._unpin_button = Button(messaging_names.chatMessageViewDelegate_unpin_icon_StatusIcon)
|
|
self._edit_button = Button(messaging_names.chatMessageViewDelegate_editMessageButton_StatusFlatRoundButton)
|
|
self._delete_button = Button(
|
|
messaging_names.chatMessageViewDelegate_chatDeleteMessageButton_StatusFlatRoundButton)
|
|
self._reply_button = Button(messaging_names.chatMessageViewDelegate_reply_icon_StatusIcon)
|
|
self._edit_message_field = TextEdit(messaging_names.edit_inputScrollView_messageInputField_TextArea)
|
|
self._reply_area = QObject(messaging_names.mainWindow_replyArea_StatusChatInputReplyArea)
|
|
self._save_text_button = Button(messaging_names.chatMessageViewDelegate_Save_StatusButton)
|
|
self._message_input_area = TextEdit(messaging_names.inputScrollView_messageInputField_TextArea)
|
|
|
|
@allure.step('Click pin button')
|
|
def pin_message(self):
|
|
self._pin_button.click()
|
|
|
|
@allure.step('Click unpin button')
|
|
def unpin_message(self):
|
|
self._unpin_button.click()
|
|
|
|
@allure.step('Edit message and save changes')
|
|
def edit_message(self, text: str):
|
|
self._edit_button.click()
|
|
self._edit_message_field.type_text(text)
|
|
self._save_text_button.click()
|
|
|
|
@allure.step('Delete message')
|
|
def delete_message(self):
|
|
self._delete_button.click()
|
|
DeleteMessagePopup().delete()
|
|
|
|
@allure.step('Reply to own message')
|
|
def reply_own_message(self, text: str):
|
|
self._reply_button.click()
|
|
assert self._reply_area.exists
|
|
self._message_input_area.type_text(text)
|
|
for i in range(2):
|
|
driver.nativeType('<Return>')
|
|
|
|
@allure.step('Delete button is visible')
|
|
def is_delete_button_visible(self) -> bool:
|
|
return self._delete_button.is_visible
|
|
|
|
|
|
class Members(QObject):
|
|
|
|
def __init__(self):
|
|
super().__init__(messaging_names.mainWindow_userListPanel_StatusListView)
|
|
self._member_item = QObject(messaging_names.groupUserListPanel_StatusMemberListItem)
|
|
|
|
@property
|
|
@allure.step('Get group members')
|
|
def members(self) -> typing.List[str]:
|
|
return [str(member.title) for member in driver.findAllObjects(self._member_item.real_name)]
|
|
|
|
|
|
class MessagesScreen(QObject):
|
|
|
|
def __init__(self):
|
|
super().__init__(messaging_names.mainWindow_chatView_ChatView)
|
|
self.left_panel = LeftPanel()
|
|
self.tool_bar = ToolBar()
|
|
self.chat = ChatView()
|
|
self.right_panel = Members()
|
|
self.group_chat = ChatMessagesView()
|