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

import logging
import pathlib
import typing

import allure

import configs
import driver
from gui.components.base_popup import BasePopup
from gui.components.community.color_select_popup import ColorSelectPopup
from gui.components.community.tags_select_popup import TagsSelectPopup
from gui.components.os.open_file_dialogs import OpenFileDialog
from gui.components.picture_edit_popup import PictureEditPopup
from gui.elements.button import Button
from gui.elements.check_box import CheckBox
from gui.elements.object import QObject
from gui.elements.scroll import Scroll
from gui.elements.text_edit import TextEdit
from gui.objects_map import names
from gui.screens.community import CommunityScreen

LOG = logging.getLogger(__name__)


class CreateCommunitiesBanner(BasePopup):

    def __init__(self):
        super().__init__()
        self._crete_community_button = Button(names.create_new_StatusButton)

    def open_create_community_popup(self) -> 'CreateCommunityPopup':
        self._crete_community_button.click()
        return CreateCommunityPopup().wait_until_appears()


class CreateCommunityPopup(BasePopup):

    def __init__(self):
        super().__init__()
        self._scroll = Scroll(names.o_Flickable)
        self._name_text_edit = TextEdit(names.createCommunityNameInput_TextEdit)
        self._description_text_edit = TextEdit(names.createCommunityDescriptionInput_TextEdit)
        self._add_logo_button = Button(names.addButton_StatusRoundButton2)
        self._add_banner_button = Button(names.addButton_StatusRoundButton)
        self._select_color_button = Button(names.StatusPickerButton)
        self._choose_tag_button = Button(names.choose_tags_StatusPickerButton)
        self._community_tags_picker_button = Button(names.communityTagsPicker_TagsPicker)
        self._archive_support_checkbox = CheckBox(names.archiveSupportToggle_StatusCheckBox)
        self._request_to_join_checkbox = CheckBox(names.requestToJoinToggle_StatusCheckBox)
        self._pin_messages_checkbox = CheckBox(names.pinMessagesToggle_StatusCheckBox)
        self._next_button = Button(names.createCommunityNextBtn_StatusButton)
        self._intro_text_edit = TextEdit(names.createCommunityIntroMessageInput_TextEdit)
        self._outro_text_edit = TextEdit(names.createCommunityOutroMessageInput_TextEdit)
        self._create_community_button = Button(names.createCommunityFinalBtn_StatusButton)
        self._cropped_image_logo_item = QObject(names.croppedImageLogo)
        self._cropped_image_banner_item = QObject(names.croppedImageBanner)

    @allure.step('Get next button enabled state')
    def is_next_button_enabled(self) -> bool:
        return driver.waitForObjectExists(self._next_button.real_name, configs.timeouts.UI_LOAD_TIMEOUT_MSEC).enabled

    @allure.step('Get archive support checkbox state')
    def is_archive_checkbox_checked(self) -> bool:
        self._scroll.vertical_scroll_to(self._archive_support_checkbox)
        return self._archive_support_checkbox.is_checked

    @allure.step('Get request to join checkbox state')
    def is_request_to_join_checkbox_checked(self) -> bool:
        self._scroll.vertical_scroll_to(self._request_to_join_checkbox)
        return self._request_to_join_checkbox.is_checked

    @allure.step('Get pin messaged checkbox state')
    def is_pin_messages_checkbox_checked(self) -> bool:
        self._scroll.vertical_scroll_to(self._pin_messages_checkbox)
        return self._pin_messages_checkbox.is_checked

    @allure.step('Get community name')
    def get_name(self) -> str:
        self._scroll.vertical_scroll_to(self._name_text_edit)
        return self._name_text_edit.text

    @allure.step('Set community name')
    def set_name(self, value: str):
        self._scroll.vertical_scroll_to(self._name_text_edit)
        self._name_text_edit.text = value

    @property
    @allure.step('Get community description')
    def description(self) -> str:
        self._scroll.vertical_scroll_to(self._description_text_edit)
        return self._description_text_edit.text

    @allure.step('Set community name')
    def set_description(self, value: str):
        self._scroll.vertical_scroll_to(self._description_text_edit)
        self._description_text_edit.text = value

    @property
    @allure.step('Get community logo')
    def logo(self):
        return NotImplementedError

    def _open_logo_file_dialog(self, attempt: int = 2):
        self._add_logo_button.click()
        try:
            return OpenFileDialog().wait_until_appears()
        except Exception as err:
            if attempt:
                LOG.debug(err)
                return self._open_logo_file_dialog(attempt - 1)
            else:
                raise err

    @allure.step('Set community logo')
    def logo(self, kwargs: dict):
        self._open_logo_file_dialog().open_file(kwargs['fp'])
        PictureEditPopup().wait_until_appears().set_zoom_shift_for_picture(kwargs.get('zoom', None),
                                                                           kwargs.get('shift', None))

    @property
    @allure.step('Get community banner')
    def banner(self):
        raise NotImplementedError

    @allure.step('Set community banner')
    def banner(self, kwargs: dict):
        self._add_banner_button.click()
        OpenFileDialog().wait_until_appears().open_file(kwargs['fp'])
        PictureEditPopup().wait_until_appears().set_zoom_shift_for_picture(kwargs.get('zoom', None),
                                                                           kwargs.get('shift', None))

    @allure.step('Set community logo without file upload dialog')
    def set_logo_without_file_upload_dialog(self, path):
        self._scroll.vertical_scroll_to(self._add_logo_button)
        fileuri = pathlib.Path(str(path)).as_uri()
        self._cropped_image_logo_item.object.cropImage(fileuri)
        return PictureEditPopup()

    @allure.step('Set community banner without file upload dialog')
    def set_banner_without_file_upload_dialog(self, path):
        self._scroll.vertical_scroll_to(self._add_banner_button)
        fileuri = pathlib.Path(str(path)).as_uri()
        self._cropped_image_banner_item.object.cropImage(fileuri)
        return PictureEditPopup()

    @allure.step('Get community color')
    def get_color(self):
        return self._select_color_button.object.bgColor.name

    @allure.step('Set community color')
    def set_color(self, value: str):
        self._scroll.vertical_scroll_to(self._select_color_button)
        self._select_color_button.click()
        ColorSelectPopup().wait_until_appears().select_color(value)

    @allure.step('Get community tags')
    def get_tags(self):
        tags_string = str(self._community_tags_picker_button.object.selectedTags)
        symbols = '[]"'
        for symbol in symbols:
            tags_string = tags_string.replace(symbol, '')
        return tags_string.split(',')

    @allure.step('Set community tags')
    def set_tags(self, values: typing.List[str]):
        self._scroll.vertical_scroll_to(self._choose_tag_button)
        self._choose_tag_button.click()
        TagsSelectPopup().wait_until_appears().select_tags(values)

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

    @allure.step('Set community intro')
    def set_intro(self, value: str):
        self._intro_text_edit.text = value

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

    @allure.step('Set community outro')
    def set_outro(self, value: str):
        self._outro_text_edit.text = value

    @allure.step('Open intro/outro form')
    def open_next_form(self):
        self._next_button.click()

    @allure.step('Select color and verify it was set correctly')
    def verify_color(self, color: str):
        assert self.get_color() == color

    @allure.step('Select tags and verify they were set correctly')
    def verify_tags(self,  tags: typing.List[str]):
        actual_tags = self.get_tags()
        assert tags == actual_tags

    @allure.step('Verify default values of checkboxes')
    def verify_checkboxes_values(self):
        assert not self.is_archive_checkbox_checked()
        assert not self.is_request_to_join_checkbox_checked()
        assert not self.is_pin_messages_checkbox_checked()

    @allure.step('Verify community create popup fields and create community without file upload dialog usage')
    def create_community(self, name: str, description: str, intro: str, outro: str, logo, banner, color: str,
                         tags_to_set: typing.List[str], tags):
        self.set_name(name)
        self.set_description(description)
        self.set_logo_without_file_upload_dialog(logo)
        PictureEditPopup().set_zoom_shift_for_picture(None, None)
        self.set_banner_without_file_upload_dialog(banner)
        PictureEditPopup().set_zoom_shift_for_picture(None, None)
        self.set_color(color)
        self.verify_color(color)
        self.set_tags(tags_to_set)
        self.verify_tags(tags)
        self.verify_checkboxes_values()
        self._next_button.click()
        self.set_intro(intro)
        self.set_outro(outro)
        self._create_community_button.click()
        self.wait_until_hidden()
        return CommunityScreen().wait_until_appears()

    @allure.step('Create simple community without verifications')
    def create_simple_community(self, name: str, description: str, intro: str, outro: str, logo, banner):
        self.set_name(name)
        self.set_description(description)
        self.set_logo_without_file_upload_dialog(logo)
        PictureEditPopup().set_zoom_shift_for_picture(None, None)
        self.set_banner_without_file_upload_dialog(banner)
        PictureEditPopup().set_zoom_shift_for_picture(None, None)
        self._next_button.click()
        self.set_intro(intro)
        self.set_outro(outro)
        self._create_community_button.click()
        self.wait_until_hidden()
        return CommunityScreen().wait_until_appears()