import unittest
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By

username =
password =


class TestCreateDelivery(unittest.TestCase):

    def setUp(self):
        self.driver = webdriver.Firefox()
        driver = self.driver
        driver.get("secret")
        self.assertIn("Login | Pharmacy", driver.title)
        elem = driver.find_element_by_id("username")
        elem.send_keys(username)
        elem = driver.find_element_by_id("password")
        elem.send_keys(password)
        elem = driver.find_element_by_id("Login")
        elem.send_keys(Keys.RETURN)
        WebDriverWait(driver, 10).until(EC.title_is("Home"))  # give 10 seconds for login to process
        assert "Home" in driver.title

    def test_new_simple(self):
        test_delivery_name = "Test Delivery"
        test_consignee = "Test Test McTest Sr"
        test_cod = "3.80"
        test_instructions = "Automated Test"
        driver = self.driver
        driver.get("secret")
        self.assertIn("Home", driver.title)
        driver.implicitly_wait(10)
        elem = driver.find_element(By.XPATH,
                                   '//*[@id="CustomerPortalTemplate"]/div[2]/div/div[2]/div[1]/div/div/div/div[1]/div[1]/div[2]/ul/li/a/div')
        elem.click()
        # Located Delivery Name field via XPATH
        elem = driver.find_element(By.XPATH, "//label/span[. = 'Delivery Name']/../../input")
        elem.send_keys(test_delivery_name)
        # Located Consignee field via XPATH
        elem = driver.find_element(By.XPATH, "//input[@title = 'Search Contacts']")
        elem.send_keys(test_consignee)
        # Select Test Consignee
        elem = driver.find_element(By.XPATH, "// div[ @ title = '" + test_consignee + "']")
        elem.click()
        # Set COD
        elem = driver.find_element(By.XPATH, "//label/span[. = 'Cash on Delivery']/../../input")
        elem.send_keys(test_cod)
        # Set Special Instructions
        elem = driver.find_element(By.XPATH, "//label/span[. = 'Special Instructions']/../../textarea")
        elem.send_keys(test_instructions)
        # Save
        elem = driver.find_element(By.XPATH, "//button[@title = 'Save']")
        elem.click()
        # Verify that Delivery name was set correctly
        elem = driver.find_element(By.XPATH,
                                   "//span[@title = 'Delivery Name']/../div/div/span[@class = 'uiOutputText']")
        assert test_delivery_name == elem.text, "Delivery Name mismatch"
        # Verify that Consignee was set correctly
        elem = driver.find_element(By.XPATH, "//span[. = 'Delivery Consignee']/../div/div/div/a")
        assert test_consignee == elem.text, "Consignee mismatch"
        # Verify that Consignee was set correctly
        elem = driver.find_element(By.XPATH,
                                   "//span[. = 'Cash on Delivery']/../../div[@class = 'slds-form-element__control slds-grid itemBody']/span")
        assert test_cod == elem.text[1:], "COD mismatch"
        # Verify that Special Instructions was set correctly
        elem = driver.find_element(By.XPATH,
                                   "//span[. = 'Special Instructions']/../../div[@class = 'slds-form-element__control slds-grid itemBody']/span")
        assert test_instructions == elem.text, "Special Instructions mismatch"

    def tearDown(self):
        self.driver.close()
        # pass


class TestDelivery(unittest.TestCase):

    def setUp(self):
        self.driver = webdriver.Firefox()
        driver = self.driver
        driver.get("secret")
        self.assertIn("Login | Pharmacy", driver.title)
        elem = driver.find_element_by_id("username")
        elem.send_keys(username)
        elem = driver.find_element_by_id("password")
        elem.send_keys(password)
        elem = driver.find_element_by_id("Login")
        elem.send_keys(Keys.RETURN)
        WebDriverWait(driver, 10).until(EC.title_is("Home"))  # give 10 seconds for login to process
        assert "Home" in driver.title

    def test_new_simple(self):
        test_delivery_name = "Test Delivery"
        test_consignee = "Test Test McTest Sr"
        test_cod = "3.80"
        test_instructions = "Automated Test"
        driver = self.driver
        driver.get("secret")
        self.assertIn("Home", driver.title)
        driver.implicitly_wait(10)
        elem = driver.find_element(By.XPATH,
                                   '//*[@id="CustomerPortalTemplate"]/div[2]/div/div[2]/div[1]/div/div/div/div[1]/div[1]/div[2]/ul/li/a/div')
        elem.click()
        # Located Delivery Name field via XPATH
        elem = driver.find_element(By.XPATH, "//label/span[. = 'Delivery Name']/../../input")
        elem.send_keys(test_delivery_name)
        # Located Consignee field via XPATH
        elem = driver.find_element(By.XPATH, "//input[@title = 'Search Contacts']")
        elem.send_keys(test_consignee)
        # Select Test Consignee
        elem = driver.find_element(By.XPATH, "// div[ @ title = '" + test_consignee + "']")
        elem.click()
        # Set COD
        elem = driver.find_element(By.XPATH, "//label/span[. = 'Cash on Delivery']/../../input")
        elem.send_keys(test_cod)
        # Set Special Instructions
        elem = driver.find_element(By.XPATH, "//label/span[. = 'Special Instructions']/../../textarea")
        elem.send_keys(test_instructions)
        # Save
        elem = driver.find_element(By.XPATH, "//button[@title = 'Save']")
        elem.click()
        # Verify that Delivery name was set correctly
        elem = driver.find_element(By.XPATH,
                                   "//span[@title = 'Delivery Name']/../div/div/span[@class = 'uiOutputText']")
        assert test_delivery_name == elem.text, "Delivery Name mismatch"
        # Verify that Consignee was set correctly
        elem = driver.find_element(By.XPATH, "//span[. = 'Delivery Consignee']/../div/div/div/a")
        assert test_consignee == elem.text, "Consignee mismatch"
        # Verify that Consignee was set correctly
        elem = driver.find_element(By.XPATH,
                                   "//span[. = 'Cash on Delivery']/../../div[@class = 'slds-form-element__control slds-grid itemBody']/span")
        assert test_cod == elem.text[1:], "COD mismatch"
        # Verify that Special Instructions was set correctly
        elem = driver.find_element(By.XPATH,
                                   "//span[. = 'Special Instructions']/../../div[@class = 'slds-form-element__control slds-grid itemBody']/span")
        assert test_instructions == elem.text, "Special Instructions mismatch"

    def tearDown(self):
        self.driver.close()
        # pass


if __name__ == "__main__":
    suite = unittest.TestLoader().loadTestsFromModule(sys.modules[__name__])
    unittest.TextTestRunner(verbosity=3).run(suite)