def logout(self): welcome = app_welcome_page.Welcome(get_driver(self)) if BasePage.get_url(self).__eq__(welcome.url_page()): welcome.click_to_logo() BasePage.click(self, By.CSS_SELECTOR, AppMain.profile_icon) BasePage.clickJS(self, By.CSS_SELECTOR, AppMain.logout_link) BasePage.wait_page(self) page = ws_page.WSMain(get_driver(self)) wait_title(get_driver(self), page.title_name) return page
def click_to_sign_up(self): BasePage.click(self, By.CSS_SELECTOR, SignUp.sign_up_button) BasePage.wait_page(self) welcome = app_welcome_page.Welcome(get_driver(self)) if BasePage.get_url(self).__eq__(page_url): return SignUp(get_driver(self)) elif welcome.is_url(): page = app_welcome_page.Welcome(get_driver(self)) return page else: return Exception("Unexpected behavior after pressing 'Sign up' button")
def is_url(self): return BasePage.get_url(self).__eq__(self.page_url)