from rita.rita.ui_autotests.actions.invoice_actions import Invoice_Actions
from rita.rita.ui_autotests.Pages.invoice_pages import Invoice, ModifyInvoice, InvoiceList, Login
from rita.rita.ui_autotests.Pages.common_functionality_pages import FuctionalButtons
from selenium.webdriver.chrome.options import Options
import selenium.webdriver.support.ui as ui

import time
import unittest
import os
import sys

from selene.tools import get_driver, set_driver
from selene.elements import *
from selenium import webdriver


invoice_actions = Invoice_Actions()
invoice = Invoice()
invoice_mod = ModifyInvoice()
invoice_list = InvoiceList()
test_invoice_id = "5900a860ffdf34019c52a648"
new_invoice_id = "58fdb9a8981661175925e58b"
clone_invoice_id = "58fdb7a0981661175925e54a"
new_invoice_id2 = "58fdb925981661175925e575"
set_smth = FuctionalButtons()
new_company_name = "rest, best"

def test_company_modifying_in_invoice(driver1):


    modify_company = invoice_actions.user_modifies_company_field_in_invoice(driver1, test_invoice_id,
                                                                             new_company_name )
    current_value = modify_company[0]
    previous_value = modify_company[1]

    print(driver1.current_url )
    driver1.get_screenshot_as_file( 'screen1.png' )
    assert previous_value != current_value

def test_cancel_invoice_modify(driver2):

    modify_smth = invoice_actions.user_cancels_invoice_modify(driver2, test_invoice_id)

    previous_value = modify_smth[0]
    current_value = modify_smth[1]
    set_value = modify_smth[2]

    driver2.get_screenshot_as_file('screen1.png')

    assert previous_value == current_value
    assert current_value != set_value


def test_creator_modifying_in_invoice(driver2):

    modify_creator = invoice_actions.user_modifies_creator_field_in_invoice(driver2, test_invoice_id, test_new_creator="Carol")

    previous_value = modify_creator[0]
    current_value = modify_creator[1]
    driver2.get_screenshot_as_file('screen1.png')

    assert previous_value != current_value

    invoice_actions.user_modifies_creator_field_in_invoice(driver2, test_invoice_id, test_new_creator="Agnessa Stone")

def test_state_modifying_in_invoice(driver2):

    new_state = "paid"
    modify_state = invoice_actions.user_modifies_state_field_in_invoice(driver2, test_invoice_id, new_state)

    previous_value = modify_state[0]
    current_value = modify_state[1]

    driver2.get_screenshot_as_file('screen1.png')

    assert current_value == new_state
    assert previous_value != current_value

    invoice_actions.user_modifies_state_field_in_invoice(driver2, test_invoice_id, new_state=previous_value)


def test_payee_modifying_in_invoice(driver2):

    new_payee_value = "John Dou"

    modify_state = invoice_actions.user_modifies_payee_field_in_invoice(driver2, test_invoice_id, new_payee_value)

    previous_value = modify_state[0]
    current_value = modify_state[1]

    driver2.get_screenshot_as_file('screen1.png')

    assert current_value == new_payee_value
    assert previous_value != current_value

def test_skip_extra_spaces_when_modify_invoice(driver2):

    extra_spaces = invoice_actions.extra_spaces_are_skipped_when_modify_invoice(driver2, test_invoice_id)

    previous_value = extra_spaces[0]
    current_value = extra_spaces[1]

    driver2.get_screenshot_as_file('screen1.png')

    assert previous_value == current_value


def test_open_invoice_details(driver2):

    open_invoice = invoice_actions.user_opens_invoice_details(driver2, test_invoice_id)
    print(open_invoice)

    driver2.get_screenshot_as_file('screen1.png')

    assert test_invoice_id == open_invoice


def test_download_pdf(driver):

    invoice_number = invoice_actions.user_downloads_pdf(driver2, test_invoice_id)

    driver.get_screenshot_as_file('screen1.png')

    invoice_list.check_that_pdf_is_downloaded(invoice_number)

    invoice_list.check_if_file_exists(invoice_number)
    invoice_list.delete_downloaded_pdf(invoice_number)


def test_accept_invoice(driver2):

    invoice_list.check_state(status="pending", invoice_id=new_invoice_id)

    accepted_invoice = invoice_actions.user_accepts_invoice(driver2, new_invoice_id)

    driver2.get_screenshot_as_file('screen1.png')

    invoice_list.check_state(accepted_invoice, new_invoice_id)

    invoice_list.check_success_message_displaying(invoice_id=new_invoice_id)

    invoice_actions.user_modifies_state_field_in_invoice(driver2, test_invoice_id, new_state = "pending")


def test_cancel_invoice_acceptance(driver2):

    invoice_list.check_state(status="pending", invoice_id=new_invoice_id)

    cancel_invoice_accept = invoice_actions.user_cancels_invoice_acceptance(driver2, new_invoice_id)

    invoice_list.cancel_accept_or_clone_invoice()
    driver2.get_screenshot_as_file('screen1.png')

    invoice_list.check_state(cancel_invoice_accept, new_invoice_id)


def test_void_invoice_when_cloning(driver2):

    invoice_actions.user_clones_and_void_invoice(driver2, clone_invoice_id)
    driver2.get_screenshot_as_file('screen1.png')

    invoice_list.check_clone_success_message_displaying()
    invoice_actions.user_modifies_state_field_in_invoice(driver2, clone_invoice_id, new_state="paid")

#  TODO for some reason clone is not working
#
# def test_clone_invoice(driver):
#
#     invoice_actions.user_clones_and_void_invoice(driver, clone_invoice_id)
#     invoice_list.check_clone_success_message_displaying()


def test_cancel_invoice_cloning(driver2):

    invoice_list.check_state(status="paid", invoice_id=clone_invoice_id)
    invoice_actions.user_cancels_clone_and_void_invoice(driver2, clone_invoice_id)
    driver2.get_screenshot_as_file('screen1.png')

    invoice_list.check_state(status="paid", invoice_id=clone_invoice_id)


def test_void_invoice(driver3):

    invoice_actions.user_voides_invoice(driver3, clone_invoice_id)
    invoice_list.check_state(status="void", invoice_id=clone_invoice_id)
    driver3.get_screenshot_as_file('screen1.png')

    invoice_list.check_success_void_message_displaying(invoice_id=clone_invoice_id)
    invoice_actions.user_modifies_state_field_in_invoice(driver3, clone_invoice_id, new_state="paid")