mangroveorg/datawinners

# vim: ai ts=4 sts=4 et sw=4 encoding=utf-8
from pages.page import Page
from pages.globalnavigationpage.global_navigation_page import GlobalNavigationPage
from pages.registrationpage.registration_page import  RegistrationPage
from framework.utils.data_fetcher import *
from pages.loginpage.login_locator import *
from testdata.test_data import DATA_WINNER_LOGIN_PAGE
from tests.logintests.login_data import *
from tests.testsettings import UI_TEST_TIMEOUT


def login(driver, credential=VALID_CREDENTIALS, landing_page=None):

    LoginPage(driver).load(landing_page).login_with(credential)
    if not landing_page:
        try:
            driver.wait_for_page_with_title(UI_TEST_TIMEOUT, "Dashboard")
        except Exception as e:
            if not driver.get_title() == "Data Submission":
                raise e
            
    return GlobalNavigationPage(driver)

class LoginPage(Page):
    def __init__(self, driver):
        Page.__init__(self, driver)

    def load(self, landing_page=None):
        url = DATA_WINNER_LOGIN_PAGE
        if landing_page:
            landing_page="/"+landing_page if landing_page[0:1] != "/" else landing_page
            url += "?next=%s"%landing_page
        self.driver.go_to(url)
        return self

    def do_successful_login_with(self, login_credential):
        """
        Function to login into the website with valid credentials

        Args:
        'login_credential' is valid login credentials of the user

        Return GlobalNavigationPage on successful login
        """
        self.driver.find_text_box(EMAIL_TB).enter_text(fetch_(USERNAME, from_(login_credential)))
        self.driver.find_text_box(PASSWORD_TB).enter_text(fetch_(PASSWORD, from_(login_credential)))
        self.driver.find(LOGIN_BTN).click()
        self.driver.wait_for_page_load()
        return GlobalNavigationPage(self.driver)

    def login_with(self, login_credential):
        """
        Function to enter email id and password in the text boxes and click
        on the login button. This function is used for testing error messages
         only
         .
        Args:
        'login_credential' is login credentials of the user e.g. email and password

        Return LoginPage
        """
        self.driver.find_text_box(EMAIL_TB).send_keys(fetch_(USERNAME, from_(login_credential)))
        self.driver.find_text_box(PASSWORD_TB).send_keys(fetch_(PASSWORD, from_(login_credential)))
        self.driver.find(LOGIN_BTN).click()
        self.driver.wait_for_page_load()
        return self

    def get_error_message(self):
        """
        Function to fetch the error messages from error label of the login
        page

        Return error message
        """
        error_message = ""
        locators = self.driver.find_elements_(ERROR_MESSAGE_LABEL)
        if locators:
            for locator in locators:
                error_message = error_message + locator.text
        return error_message.replace("\n", " ")

    def navigate_to_registration_page(self):
        """
        Function to click on register page link which is available on the login page

        Return RegistrationPage
        """
        self.driver.find(CREATE_AN_ACCOUNT_LINK).click()
        return RegistrationPage(self.driver)

    @classmethod
    def navigate_to(cls, driver):
        driver.go_to(DATA_WINNER_LOGIN_PAGE)
        return LoginPage(driver)