mangroveorg/datawinners

# vim: ai ts=4 sts=4 et sw=4 encoding=utf-8
from framework.utils.common_utils import generateId, get_epoch_last_ten_digit
from framework.utils.text_box_web_element import TextBox

from pages.page import Page
from pages.registerconfirmationpage.registration_confirmation_page import RegistrationConfirmationPage
from framework.utils.data_fetcher import *
from pages.registrationpage.registration_locator import *
from tests.registrationtests.registration_data import *
from tests.testsettings import UI_TEST_TIMEOUT
from tests.upgradetests.upgrade_trial_data import ORGANIZATION_SECTOR as ORGANIZATION_SECTOR_FOR_UPGRADE
import time

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

    def successful_registration_with(self, registration_data):
        registration_data = dict(registration_data) # create a copy so we don't modify in place
        email = registration_data[EMAIL] + generateId() + "@ngo.com"
        registration_data[EMAIL] = email
        registration_data[ADMIN_MOBILE_NUMBER] = str(get_epoch_last_ten_digit())
        self.register_with(registration_data)
        return RegistrationConfirmationPage(self.driver), email

    def register_with(self, registration_data):
        self.agree_terms_and_conditions()
        for key, value in registration_data.items():
            if key in [ORGANIZATION_SECTOR, ORGANIZATION_COUNTRY, ORGANIZATION_SECTOR_FOR_UPGRADE]:
                self.driver.find_drop_down(by_css(".registration_form select[name=%s]" % key)).set_selected(value)
            elif key in [PAY_MONTHLY, WIRE_TRANSFER, PRO, PRO_SMS]:
                self.driver.find_radio_button(by_css(".registration_form input[value='%s']" % key)).click()
            else:
                el = self.driver.find(by_css(".registration_form input[name=%s]" % key))
                if el.get_attribute("value") != "":
                    el.clear()
                el.send_keys(value)

        self.click_submit_button()
        return self

    def agree_terms_and_conditions(self):
        self.driver.find(AGREE_TERMS_CB).click()

    def get_error_message(self):
        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 click_submit_button(self):
        self.driver.find(ORGANIZATION_REGISTER_BTN).click()