status-desktop/test/e2e/gui/components/community/invite_contacts.py

import typing

import allure

import configs.timeouts
import driver
from gui.components.base_popup import BasePopup
from gui.elements.button import Button
from gui.elements.object import QObject
from gui.elements.text_edit import TextEdit
from gui.objects_map import names


class InviteContactsPopup(BasePopup):

    def __init__(self):
        super().__init__()
        self._member_item = QObject(names.o_StatusMemberListItem)
        self._next_button = Button(names.next_StatusButton)
        self._message_text_edit = TextEdit(names.communityProfilePopupInviteMessagePanel_MessageInput_TextEdit)
        self._invited_member_item = QObject(names.o_StatusMemberListItem_2)
        self._send_button = Button(names.send_1_invite_StatusButton)

    @property
    @allure.step('Get contacts')
    def contacts(self) -> typing.List[str]:
        return [str(getattr(user, 'title', '')) for user in driver.findAllObjects(self._member_item.real_name)]

    @property
    @allure.step('Invite contacts')
    def invited_contacts(self) -> typing.List[str]:
        return [str(getattr(user, 'title', '')) for user in driver.findAllObjects(self._invited_member_item.real_name)]

    def invite(self, contacts: typing.List[str], message: str):
        for contact in contacts:
            assert driver.waitFor(lambda: contact in self.contacts, configs.timeouts.UI_LOAD_TIMEOUT_MSEC), \
                f'Contact: {contact} not found in {self.contacts}'

        selected = []
        for member in driver.findAllObjects(self._member_item.real_name):
            if str(getattr(member, 'title', '')) in contacts:
                driver.mouseClick(member)
                selected.append(member.title)

        assert len(contacts) == len(selected), f'Selected contacts: {selected}, expected: {contacts}'

        self._next_button.click()
        self._message_text_edit.text = message

        for contact in contacts:
            assert driver.waitFor(lambda: contact in self.invited_contacts, configs.timeouts.UI_LOAD_TIMEOUT_MSEC), \
                f'Contact: {contact} not found in {self.invited_contacts}'

        self._send_button.click()
        self.wait_until_hidden()