status-desktop/test/e2e/gui/main_window.py

226 lines
8.9 KiB
Python

import logging
import os
import time
import typing
import allure
import configs
import constants
import driver
from constants import UserAccount
from gui.components.community.invite_contacts import InviteContactsPopup
from gui.components.context_menu import ContextMenu
from gui.components.onboarding.before_started_popup import BeforeStartedPopUp
from gui.components.onboarding.beta_consent_popup import BetaConsentPopup
from gui.components.splash_screen import SplashScreen
from gui.components.toast_message import ToastMessage
from gui.components.online_identifier import OnlineIdentifier
from gui.elements.button import Button
from gui.elements.object import QObject
from gui.elements.window import Window
from gui.mocked_keycard_controller import MockedKeycardController
from gui.objects_map import names
from gui.screens.community import CommunityScreen
from gui.screens.community_portal import CommunitiesPortal
from gui.screens.messages import MessagesScreen
from gui.screens.onboarding import AllowNotificationsView, WelcomeToStatusView, BiometricsView, LoginView
from gui.screens.settings import SettingsScreen
from gui.screens.wallet import WalletScreen
from scripts.tools.image import Image
LOG = logging.getLogger(__name__)
class LeftPanel(QObject):
def __init__(self):
super(LeftPanel, self).__init__(names.mainWindow_StatusAppNavBar)
self._profile_button = Button(names.mainWindow_ProfileNavBarButton)
self._messages_button = Button(names.messages_navbar_StatusNavBarTabButton)
self._communities_portal_button = Button(names.communities_Portal_navbar_StatusNavBarTabButton)
self._community_template_button = Button(names.statusCommunityMainNavBarListView_CommunityNavBarButton)
self._settings_button = Button(names.settings_navbar_StatusNavBarTabButton)
self._wallet_button = Button(names.wallet_navbar_StatusNavBarTabButton)
self._community_invite_people_context_item = QObject(names.invite_People_StatusMenuItem)
@property
@allure.step('Get communities names')
def communities(self) -> typing.List[str]:
community_names = []
for obj in driver.findAllObjects(self._community_template_button.real_name):
community_names.append(obj.name)
if len(community_names) == 0:
raise LookupError(
'Communities not found')
else:
return community_names
@property
@allure.step('Get user badge color')
def user_badge_color(self) -> str:
return str(self._profile_button.object.badge.color.name)
@allure.step('Open messages screen')
def open_messages_screen(self) -> MessagesScreen:
self._messages_button.click()
return MessagesScreen().wait_until_appears()
@allure.step('Open online identifier')
def open_online_identifier(self, attempts: int = 2) -> OnlineIdentifier:
time.sleep(0.5)
self._profile_button.click()
try:
return OnlineIdentifier().wait_until_appears()
except Exception as ex:
if attempts:
self.open_online_identifier(attempts - 1)
else:
raise ex
@allure.step('Set user to online')
def set_user_to_online(self):
self.open_online_identifier().set_user_state_online()
@allure.step('Verify: User is online')
def user_is_online(self) -> bool:
return self.user_badge_color == '#4ebc60'
@allure.step('Set user to offline')
def set_user_to_offline(self):
self.open_online_identifier().set_user_state_offline()
@allure.step('Verify: User is offline')
def user_is_offline(self):
return self.user_badge_color == '#7f8990'
@allure.step('Set user to automatic')
def set_user_to_automatic(self):
self.open_online_identifier().set_user_automatic_state()
@allure.step('Verify: User is set to automatic')
def user_is_set_to_automatic(self):
return self.user_badge_color == '#4ebc60'
@allure.step('Open community portal')
def open_communities_portal(self) -> CommunitiesPortal:
self._communities_portal_button.click()
return CommunitiesPortal().wait_until_appears()
def _get_community(self, name: str):
community_names = []
for obj in driver.findAllObjects(self._community_template_button.real_name):
community_names.append(str(obj.name))
if str(obj.name) == str(name):
return obj
raise LookupError(f'Community: {name} not found in {community_names}')
@allure.step('Open community')
def select_community(self, name: str) -> CommunityScreen:
driver.mouseClick(self._get_community(name))
return CommunityScreen().wait_until_appears()
@allure.step('Get community logo')
def get_community_logo(self, name: str) -> Image:
return Image(driver.objectMap.realName(self._get_community(name)))
@allure.step('Open context menu for community')
def open_community_context_menu(self, name: str) -> ContextMenu:
driver.objectMap.realName(self._get_community(name))['name'] = name
self._get_community(name).open_context_menu()
return ContextMenu().wait_until_appears()
@allure.step('Invite people in community')
def invite_people_in_community(self, contacts: typing.List[str], message: str, community_name: str):
driver.mouseClick(self._get_community(community_name), driver.Qt.RightButton)
self._community_invite_people_context_item.click()
InviteContactsPopup().wait_until_appears().invite(contacts, message)
@allure.step('Open settings')
def open_settings(self, attempts: int = 2) -> SettingsScreen:
self._settings_button.click()
time.sleep(0.5)
try:
return SettingsScreen()
except Exception as ex:
if attempts:
self.open_settings(attempts - 1)
else:
raise ex
@allure.step('Open Wallet section')
def open_wallet(self, attempts: int = 2) -> WalletScreen:
self._wallet_button.click()
try:
return WalletScreen()
except Exception as ex:
if attempts:
return self.open_wallet(attempts - 1)
else:
raise ex
class MainWindow(Window):
def __init__(self):
super(MainWindow, self).__init__(names.statusDesktop_mainWindow)
self.left_panel = LeftPanel()
# TODO: we need to handle all the issues with keycard mock var before using keycard window in tests
def prepare(self) -> 'Window':
# MockedKeycardController().wait_until_appears().hide()
return super().prepare()
@allure.step('Sign Up user')
def sign_up(self, user_account: UserAccount = constants.user.user_account_one):
if configs.system.IS_MAC:
AllowNotificationsView().wait_until_appears().allow()
BeforeStartedPopUp().get_started()
wellcome_screen = WelcomeToStatusView().wait_until_appears()
profile_view = wellcome_screen.get_keys().generate_new_keys()
profile_view.set_display_name(user_account.name)
details_view = profile_view.next()
create_password_view = details_view.next()
confirm_password_view = create_password_view.create_password(user_account.password)
confirm_password_view.confirm_password(user_account.password)
if configs.system.IS_MAC:
BiometricsView().wait_until_appears().prefer_password()
SplashScreen().wait_until_appears().wait_until_hidden()
if not configs.system.TEST_MODE:
BetaConsentPopup().confirm()
return self
@allure.step('Log in user')
def log_in(self, user_account: UserAccount):
LoginView().log_in(user_account)
SplashScreen().wait_until_appears().wait_until_hidden()
if not configs.system.TEST_MODE:
BetaConsentPopup().confirm()
return self
@allure.step('Authorize user')
def authorize_user(self, user_account) -> 'MainWindow':
assert isinstance(user_account, UserAccount)
if LoginView().is_visible:
return self.log_in(user_account)
else:
return self.sign_up(user_account)
@allure.step('Create community')
def create_community(self, name, description, intro, outro, logo, banner) -> CommunityScreen:
communities_portal = self.left_panel.open_communities_portal()
create_community_form = communities_portal.open_create_community_popup()
app_screen = create_community_form.create_community(name, description, intro, outro, logo, banner)
return app_screen
@allure.step('Wait for notification and get text')
def wait_for_notification(self, timeout_msec: int = configs.timeouts.UI_LOAD_TIMEOUT_SEC) -> str:
started_at = time.monotonic()
while True:
try:
return ToastMessage().get_toast_messages
except LookupError as err:
LOG.info(err)
assert time.monotonic() - started_at < timeout_msec, str(err)