status-desktop/test/e2e/gui/screens/community.py

import time
import typing

import allure
from allure_commons._allure import step

import configs
import driver
from constants import UserChannel, ColorCodes
from driver.objects_access import walk_children
from gui.components.community.community_category_popup import NewCategoryPopup, EditCategoryPopup, CategoryPopup
from gui.components.community.community_channel_popups import EditChannelPopup, NewChannelPopup
from gui.components.community.invite_contacts import InviteContactsPopup
from gui.components.community.welcome_community import WelcomeCommunityPopup
from gui.components.context_menu import ContextMenu
from gui.components.delete_popup import DeletePopup, DeleteCategoryPopup
from gui.components.profile_popup import ProfilePopupFromMembers
from gui.elements.button import Button
from gui.elements.list import List
from gui.elements.object import QObject
from gui.elements.text_label import TextLabel
from gui.objects_map import names, communities_names, messaging_names
from gui.screens.community_settings import CommunitySettingsScreen
from scripts.tools.image import Image


class CommunityScreen(QObject):

    def __init__(self):
        super().__init__(communities_names.mainWindow_communityLoader_Loader)
        self.left_panel = LeftPanel()
        self.tool_bar = ToolBar()
        self.chat = Chat()
        self.right_panel = Members()

    @allure.step('Wait until appears {0}')
    def wait_until_appears(self, timeout_msec: int = configs.timeouts.UI_LOAD_TIMEOUT_MSEC):
        self.left_panel.wait_until_appears(timeout_msec)
        return self

    @allure.step('Create channel')
    def create_channel(self, name: str, description: str, emoji: str = None):
        self.left_panel.open_create_channel_popup().create(name, description, emoji).save()

    @allure.step('Edit channel')
    def edit_channel(self, channel, name: str, description: str, emoji: str = None):
        self.left_panel.select_channel(channel)
        self.tool_bar.open_edit_channel_popup().edit(name, description, emoji)

    @allure.step('Delete channel')
    def delete_channel(self, name: str):
        self.left_panel.select_channel(name)
        self.tool_bar.open_delete_channel_popup().delete()

    @allure.step('Verify channel')
    def verify_channel(
            self, name: str, description: str, emoji):
        with step('Channel is correct in channels list'):
            channel = self.left_panel.get_channel_parameters(name)
            assert channel.name == name

        with step('Channel is correct in community toolbar'):
            assert self.tool_bar.channel_name == name
            assert self.tool_bar.channel_description == description
            if emoji is not None:
                assert self.tool_bar.channel_emoji == emoji

        with step('Verify channel in chat'):
            assert self.chat.channel_name == name
            if emoji is not None:
                assert self.chat.channel_emoji == emoji

    @allure.step('Create category')
    def create_category(self, name: str, general_checkbox: bool):
        self.left_panel.open_create_category_popup().create(name, general_checkbox)

    @allure.step('Delete category from the list')
    def delete_category(self):
        self.left_panel.open_more_options()
        self.left_panel.open_delete_category_popup().delete()

    @allure.step('Edit category')
    def edit_category(self):
        self.left_panel.open_more_options()
        self.left_panel.open_edit_category_popup()
        return EditCategoryPopup()

    @allure.step('Verify category in the list')
    def verify_category(self, category_name: str):
        category = self.left_panel.find_category_in_list(category_name)
        assert category.category_name == category_name


class ToolBar(QObject):

    def __init__(self):
        super().__init__(communities_names.mainWindow_statusToolBar_StatusToolBar)
        self._more_options_button = Button(communities_names.statusToolBar_chatToolbarMoreOptionsButton)
        self._options_list = List(names.o_StatusListView)
        self._edit_channel_context_item = QObject(communities_names.edit_Channel_StatusMenuItem)
        self._channel_icon = QObject(communities_names.statusToolBar_statusSmartIdenticonLetter_StatusLetterIdenticon)
        self._channel_name = TextLabel(
            communities_names.statusToolBar_statusChatInfoButtonNameText_TruncatedTextWithTooltip)
        self._channel_description = TextLabel(communities_names.statusToolBar_TruncatedTextWithTooltip)
        self._delete_channel_context_item = QObject(communities_names.delete_Channel_StatusMenuItem)
        self._channel_header = QObject(communities_names.statusToolBar_chatInfoBtnInHeader_StatusChatInfoButton)

    @property
    @allure.step('Get channel emoji')
    def channel_emoji(self):
        return self._channel_header.object.asset.emoji

    @property
    @allure.step('Get channel color')
    def channel_color(self) -> str:
        return str(self._channel_header.object.asset.color.name).lower()

    @property
    @allure.step('Get channel name')
    def channel_name(self) -> str:
        return self._channel_name.text

    @property
    @allure.step('Get channel description')
    def channel_description(self) -> str:
        return self._channel_description.text

    @allure.step('Open edit channel popup')
    def open_edit_channel_popup(self):
        self.open_more_options_dropdown()
        self._edit_channel_context_item.click()
        return EditChannelPopup().wait_until_appears()

    @allure.step('Open delete channel popup')
    def open_delete_channel_popup(self):
        self.open_more_options_dropdown()
        self._delete_channel_context_item.click()
        return DeletePopup().wait_until_appears()

    @allure.step('Open more options dropdown')
    def open_more_options_dropdown(self):
        self._more_options_button.click()
        return ContextMenu()

    @allure.step('Get visibility state of edit item')
    def is_edit_item_visible(self) -> bool:
        return self._edit_channel_context_item.exists

    @allure.step('Get visibility state of delete item')
    def is_delete_item_visible(self) -> bool:
        return self._delete_channel_context_item.exists


class CategoryItem:

    def __init__(self, obj):
        self.object = obj
        self._category_name: typing.Optional[Image] = None
        self._add_category_button: typing.Optional[Button] = None
        self._more_button: typing.Optional[Button] = None
        self._arrow_button: typing.Optional[Button] = None
        self._arrow_icon: typing.Optional[QObject] = None
        self.init_ui()

    def __repr__(self):
        return self.category_name

    def init_ui(self):
        for child in walk_children(self.object):
            if str(getattr(child, 'id', '')) == 'statusChatListCategoryItem':
                self.category_name = str(child.text)
            elif str(getattr(child, 'id', '')) == 'addButton':
                self._add_channel_button = Button(real_name=driver.objectMap.realName(child))
            elif str(getattr(child, 'id', '')) == 'menuButton':
                self._more_button = Button(real_name=driver.objectMap.realName(child))
            elif str(getattr(child, 'id', '')) == 'toggleButton':
                self._arrow_button = Button(real_name=driver.objectMap.realName(child))
            elif str(getattr(child, 'objectName', '')) == 'chevron-down-icon':
                self._arrow_icon = QObject(real_name=driver.objectMap.realName(child))

    @allure.step('Click arrow button')
    def click_arrow_button(self):
        self._arrow_button.click()

    @allure.step('Get arrow button rotation value')
    def get_arrow_icon_rotation_value(self) -> int:
        return self._arrow_icon.object.rotation


class LeftPanel(QObject):

    def __init__(self):
        super().__init__(communities_names.mainWindow_communityColumnView_CommunityColumnView)
        self._community_info_button = Button(communities_names.mainWindow_communityHeaderButton_StatusChatInfoButton)
        self._community_logo = QObject(communities_names.mainWindow_identicon_StatusSmartIdenticon)
        self._name_text_label = TextLabel(
            communities_names.mainWindow_statusChatInfoButtonNameText_TruncatedTextWithTooltip)
        self._members_text_label = TextLabel(communities_names.mainWindow_Members_TruncatedTextWithTooltip)
        self._general_channel_item = QObject(communities_names.scrollView_general_StatusChatListItem)
        self._add_channels_button = Button(communities_names.add_channels_StatusButton)
        self._channel_icon_template = QObject(communities_names.channel_identicon_StatusSmartIdenticon)
        self._channel_or_category_button = Button(
            communities_names.mainWindow_createChannelOrCategoryBtn_StatusBaseText)
        self._create_channel_menu_item = Button(communities_names.create_channel_StatusMenuItem)
        self._create_category_menu_item = Button(communities_names.create_category_StatusMenuItem)
        self._join_community_button = Button(communities_names.mainWindow_Join_Community_StatusButton)

        self.communityChatListAndCategories = QObject(communities_names.communityChatListAndCategories)
        self.channelAndCategoriesListItems = QObject(communities_names.channelAndCategoriesListItems)
        self.chatListItems = QObject(communities_names.chatListItems)
        self.chatListItemDropAreaItem = QObject(communities_names.chatListItemDropAreaItem)
        self.categoryItemDropAreaItem = QObject(communities_names.categoryListItemDropAreaItem)

        self._category_list_item = QObject(communities_names.categoryItem_StatusChatListCategoryItem)
        self._create_category_button = Button(communities_names.add_categories_StatusFlatButton)
        self._delete_category_item = QObject(communities_names.delete_Category_StatusMenuItem)
        self._edit_category_item = QObject(communities_names.edit_Category_StatusMenuItem)
        self._add_channel_inside_category_item = QObject(
            communities_names.scrollView_addButton_StatusChatListCategoryItemButton)
        self._more_button = Button(communities_names.scrollView_menuButton_StatusChatListCategoryItemButton)
        self._arrow_button = Button(communities_names.scrollView_toggleButton_StatusChatListCategoryItemButton)
        self._add_members_button = Button(names.scrollView_Add_members_StatusButton)

    @property
    @allure.step('Get community logo')
    def logo(self) -> Image:
        return self._community_logo.image

    @property
    @allure.step('Get community name')
    def name(self) -> str:
        return self._name_text_label.text

    @property
    @allure.step('Get community members label')
    def members(self) -> str:
        return self._members_text_label.text

    @property
    @allure.step('Get Join button visible attribute')
    def is_join_community_visible(self) -> bool:
        return self._join_community_button.exists

    @property
    @allure.step('Get channels')
    def channels(self) -> typing.List[UserChannel]:
        time.sleep(0.5)
        channels_list = []
        for obj in driver.findAllObjects(self.chatListItemDropAreaItem.real_name):
            container = driver.objectMap.realName(obj)
            self._channel_icon_template.real_name['container'] = container
            channels_list.append(UserChannel(
                str(obj.objectName),
                obj.item.selected,
                obj.item.visible
            ))
        return channels_list

    @property
    @allure.step('Get categories')
    def categories_items(self) -> typing.List[CategoryItem]:
        return [CategoryItem(item) for item in driver.findAllObjects(self.categoryItemDropAreaItem.real_name)]

    @allure.step('Get arrow button rotation value')
    def get_arrow_icon_rotation_value(self, category_name) -> int:
        category = self.find_category_in_list(category_name)
        return int(category.get_arrow_icon_rotation_value())

    @allure.step('Get channel params')
    def get_channel_parameters(self, name) -> UserChannel:
        for channel in self.channels:
            if channel.name == name:
                return channel
        raise LookupError(f'Channel not found in {self.channels}')

    @allure.step('Open community settings')
    def open_community_settings(self):
        self._community_info_button.click()
        return CommunitySettingsScreen().wait_until_appears()

    @allure.step('Open create channel popup')
    def open_create_channel_popup(self) -> NewChannelPopup:
        self._channel_or_category_button.click()
        self._create_channel_menu_item.click()
        return NewChannelPopup()

    @allure.step('Get presence state of create channel or category button')
    def does_create_channel_or_category_button_exist(self) -> bool:
        return self._channel_or_category_button.exists

    @allure.step('Get visibility state of add channels button')
    def is_add_channels_button_visible(self) -> bool:
        return self._add_channels_button.is_visible

    @allure.step('Select channel')
    def select_channel(self, name: str):
        time.sleep(3)
        for obj in driver.findAllObjects(self.chatListItemDropAreaItem.real_name):
            if str(obj.objectName) == name:
                driver.mouseClick(obj)
                return obj
        raise LookupError('Channel not found')

    @allure.step('Open general channel context menu')
    def open_general_channel_context_menu(self):
        self._general_channel_item.right_click()
        return ContextMenu()

    @allure.step('Open category context menu')
    def open_category_context_menu(self):
        self.categoryItemDropAreaItem.right_click()

    @allure.step('Open create category popup')
    def open_create_category_popup(self, attempts: int = 2) -> NewCategoryPopup:
        self._channel_or_category_button.click()
        try:
            self._create_category_menu_item.click()
            return NewCategoryPopup()
        except Exception as ex:
            if attempts:
                self.open_create_category_popup(attempts - 1)
            else:
                raise ex

    @allure.step('Open join community popup')
    def open_welcome_community_popup(self):
        self._join_community_button.wait_until_appears(configs.timeouts.UI_LOAD_TIMEOUT_MSEC)
        self._join_community_button.click()
        return WelcomeCommunityPopup().wait_until_appears()

    @allure.step('Find category')
    def find_category_in_list(
            self, category_name: str):
        time.sleep(1)
        categories = self.categories_items
        for _category in categories:
            if _category.category_name == category_name:
                category = _category
                return category
            raise LookupError(f'Category: {category_name} not found in {categories}')

    def click_category(self, category_name: str):
        driver.mouseClick(self.find_category_in_list(category_name).object)

    @allure.step('Open more options')
    def open_more_options(self, attempts: int = 2):
        self._arrow_button.click()
        try:
            self._more_button.click()
        except LookupError as err:
            if attempts:
                return self._more_button.click(attempts - 1)
            else:
                raise err
        return self

    @allure.step('Get visibility state of delete item')
    def is_delete_item_visible(self) -> bool:
        return self._delete_category_item.is_visible

    @allure.step('Get visibility state of edit item')
    def is_edit_item_visible(self) -> bool:
        return self._edit_category_item.is_visible

    @allure.step('Open delete category popup')
    def open_delete_category_popup(self) -> DeleteCategoryPopup:
        self._delete_category_item.click()
        return DeleteCategoryPopup().wait_until_appears()

    @allure.step('Open edit category popup')
    def open_edit_category_popup(self) -> EditCategoryPopup:
        self._edit_category_item.click()
        return CategoryPopup().wait_until_appears()

    @allure.step('Open new channel popup inside category')
    def open_new_channel_popup_in_category(self) -> NewChannelPopup:
        self._arrow_button.click()
        self._add_channel_inside_category_item.click()
        return NewChannelPopup().wait_until_appears()

    @allure.step('Get channel or category index in the list')
    def get_channel_or_category_index(self, name: str) -> int:
        for child in walk_children(self.chatListItems.object):
            if child.objectName == name:
                return child.visualIndex

    @allure.step('Right click on left panel')
    def right_click_on_panel(self):
        super(LeftPanel, self).right_click()

    @allure.step('Invite people to community')
    def invite_people_to_community(self, contacts: typing.List[str], message: str, attempts: int = 2):
        self._add_members_button.click()
        try:
            InviteContactsPopup().wait_until_appears().invite(contacts, message)
        except AssertionError as err:
            if attempts:
                self.invite_people_to_community(contacts, message, attempts - 1)
            else:
                raise err


class Chat(QObject):

    def __init__(self):
        super().__init__(communities_names.mainWindow_ChatColumnView)
        self._channel_icon = QObject(
            communities_names.chatMessageViewDelegate_channelIdentifierSmartIdenticon_StatusSmartIdenticon)
        self._channel_name_label = TextLabel(
            communities_names.chatMessageViewDelegate_channelIdentifierNameText_StyledText)
        self._channel_welcome_label = TextLabel(communities_names.chatMessageViewDelegate_Welcome)
        self._channel_identifier_view = QObject(messaging_names.chatMessageViewDelegate_ChannelIdentifierView)

    @property
    @allure.step('Get channel emoji')
    def channel_emoji(self):
        return self._channel_identifier_view.object.chatEmoji

    @property
    @allure.step('Get channel color')
    def channel_color(self) -> str:
        return str(self._channel_identifier_view.object.chatColor).lower()

    @property
    @allure.step('Get channel name')
    def channel_name(self) -> str:
        return self._channel_name_label.text

    @property
    @allure.step('Get channel welcome note')
    def channel_welcome_note(self) -> str:
        return self._channel_welcome_label.text


class Members(QObject):

    def __init__(self):
        super().__init__(communities_names.mainWindow_userListPanel_StatusListView)
        self._member_item = QObject(communities_names.userListPanel_StatusMemberListItem)
        self._user_badge_color = QObject(communities_names.statusBadge_StatusBadge)
        self._banned_tab_button = Button(communities_names.membersTabBar_Banned_StatusTabButton)
        self._all_members_tab_button = Button(communities_names.membersTabBar_All_Members_StatusTabButton)

    @property
    @allure.step('Get all members')
    def members(self) -> typing.List[str]:
        return [str(member.userName) for member in driver.findAllObjects(self._member_item.real_name)]

    @allure.step('Open banned tab')
    def click_banned_button(self):
        self._banned_tab_button.click()

    @allure.step('Open all members tab')
    def click_all_members_button(self):
        self._all_members_tab_button.click()

    @allure.step('Click member by name')
    def click_member(self, member_name: str):
        for member in driver.findAllObjects(self._member_item.real_name):
            if getattr(member, 'userName', '') == member_name:
                driver.mouseClick(member)
                break
        return ProfilePopupFromMembers().wait_until_appears()

    @allure.step('Verify member is offline by index')
    def member_state(self, member_name: str) -> bool:
        for member in driver.findAllObjects(self._member_item.real_name):
            if getattr(member, 'userName', '') == member_name:
                for child in walk_children(member):
                    if getattr(child, 'id', '') == 'statusBadge':
                        return child.color.name