Actions

from selenium.webdriver import ActionChains
from selenium.webdriver.support.wait import WebDriverWait

from rita.rita.ui_autotests.Pages.invoice_pages import *
from rita.rita.ui_autotests.Pages.common_functionality_pages import *


invoice = Invoice()
invoice_mod = ModifyInvoice()
invoice_list = InvoiceList()
set_smth = FuctionalButtons()

class Invoice_Actions(object):

    def user_modifies_company_field_in_invoice(self, driver, test_invoice_id, new_company_name):

        previous_company_name = invoice_mod.get_company_name(invoice_id=test_invoice_id)

        invoice_mod.modify_invoice(invoice_id=test_invoice_id)
        invoice_mod.change_company(company_name=new_company_name)
        invoice_mod.save_changes()

        driver.refresh()

        current_company_name = invoice_mod.get_company_name(invoice_id=test_invoice_id)
        return (current_company_name, previous_company_name)


    def user_cancels_invoice_modify(self, driver, test_invoice_id):

        previous_state = invoice_mod.get_state(invoice_id=test_invoice_id)
        invoice.highlight_invoice(invoice_id=test_invoice_id)

        invoice_mod.modify_invoice(invoice_id=test_invoice_id)
        new_state = "paid"
        invoice_mod.change_state(state=new_state)
        time.sleep(2)
        invoice_mod.close_modify()

        driver.refresh()

        current_state = invoice_mod.get_state(invoice_id=test_invoice_id)

        return previous_state, current_state, new_state


    def user_modifies_creator_field_in_invoice(self, driver, test_invoice_id, test_new_creator):

        previous_creator = invoice_mod.get_creator(invoice_id=test_invoice_id)

        invoice_mod.modify_invoice(invoice_id=test_invoice_id)
        invoice_mod.change_creator(new_creator=test_new_creator)
        time.sleep(2)
        invoice_mod.save_changes()

        driver.refresh()

        current_creator = invoice_mod.get_creator(invoice_id=test_invoice_id)

        return previous_creator, current_creator


    def user_modifies_state_field_in_invoice(self, driver, test_invoice_id, new_state):

        previous_state = invoice_mod.get_state(invoice_id=test_invoice_id)
        print(previous_state)
        invoice.highlight_invoice(invoice_id=test_invoice_id)

        invoice_mod.modify_invoice(invoice_id=test_invoice_id)
        invoice_mod.change_state(state=new_state)
        time.sleep(2)

        invoice_mod.save_changes()

        driver.refresh()

        current_state = invoice_mod.get_state(invoice_id=test_invoice_id)
        print(current_state)

        return previous_state, current_state


    def user_modifies_payee_field_in_invoice(self, driver, test_invoice_id, new_payee_value):

        invoice_mod.modify_invoice(invoice_id=test_invoice_id)
        previous_payee = invoice_mod.get_payee()
        print(previous_payee)
        time.sleep(2)
        invoice_mod.change_payee(new_payee_value)
        invoice_mod.save_changes()

        driver.refresh()

        invoice_mod.modify_invoice(invoice_id=test_invoice_id)
        current_payee = invoice_mod.get_payee()
        print(current_payee)
        time.sleep(2)
        invoice_mod.change_payee(previous_payee)
        invoice_mod.save_changes()

        driver.refresh()

        return previous_payee, current_payee

    def extra_spaces_are_skipped_when_modify_invoice(self, driver, test_invoice_id):

        invoice_mod.modify_invoice(invoice_id=test_invoice_id)
        new_payee_value = "                                                    Amélie Dubois                         " \
                          "" \

        time.sleep(2)
        invoice_mod.change_payee(new_payee=new_payee_value)
        time.sleep(2)
        invoice_mod.save_changes()
        time.sleep(2)

        invoice_mod.modify_invoice(invoice_id=test_invoice_id)
        current_payee = invoice_mod.get_payee()

        new_payee_value_without_spaces = "Amélie Dubois"

        time.sleep(2)
        invoice_mod.change_payee(new_payee="to be changed")
        invoice_mod.save_changes()

        driver.refresh()

        invoice_mod.modify_invoice(invoice_id=test_invoice_id)
        print (invoice_mod.get_payee())

        return current_payee, new_payee_value_without_spaces

    def user_opens_invoice_details(self, driver, test_invoice_id):

        invoice.open_invoice(invoice_id=test_invoice_id)

        driver.switch_to_window(driver.window_handles[1])

        return invoice.get_invoice_id_from_open_page()

    def user_downloads_pdf(self, driver, test_invoice_id):

        invoice_list.click_on_pdf_download(invoice_id=test_invoice_id)
        time.sleep(2)

        return invoice_list.get_invoice_number(invoice_id=test_invoice_id)

    def user_accepts_invoice(self, driver, new_invoice_id):

        invoice_list.click_accept_invoice(new_invoice_id)
        wait = WebDriverWait(driver, 10)
        wait.until(EC.presence_of_element_located(by_xpath(invoice_list._get_confirm_modal_path())))

        invoice_list.confirm_accept_invoice()
        time.sleep(10)

        return invoice_mod.get_state(new_invoice_id)

    def user_cancels_invoice_acceptance(self, driver, new_invoice_id):

        invoice_list.click_accept_invoice(new_invoice_id)
        wait = WebDriverWait(driver, 5)
        wait.until(EC.presence_of_element_located(by_xpath(invoice_list._get_confirm_modal_path())))

        invoice_list.cancel_accept_or_clone_invoice()

        return invoice_mod.get_state(new_invoice_id)

    def user_clones_and_void_invoice(self, driver, test_invoice_id):

        invoice_list.click_clone_invoice( invoice_id=test_invoice_id )
        btn = invoice_list.confrim_clone_or_void_invoice()
        void = invoice_list.void_invoice_when_clonning()
        ActionChains( driver ).move_to_element( void ).click().perform()
        ActionChains( driver ).move_to_element( btn ).click().perform()

        return self


    def user_cancels_clone_and_void_invoice(self, driver, test_invoice_id):

        invoice_list.click_clone_invoice( invoice_id=test_invoice_id )
        void = invoice_list.void_invoice_when_clonning()
        ActionChains( driver ).move_to_element( void ).click().perform()
        invoice_list.cancel_accept_or_clone_invoice()
        return self

    def user_voides_invoice(self, driver, test_invoice_id):

        invoice_list.check_state(status="paid", invoice_id=test_invoice_id)
        invoice_list.click_void_invoice(test_invoice_id)
        time.sleep(2)
        btn = invoice_list.confrim_clone_or_void_invoice()
        ActionChains(driver).move_to_element(btn).click().perform()
        time.sleep(2)
        return self