﻿import os
import sys
import random
import unittest

sys.path.insert(1, os.path.abspath(os.path.join(__file__, "../..")))
import base_test

repo_root = os.path.abspath(os.path.join(__file__, "../../.."))
sys.path.insert(1, os.path.join(repo_root, "tools", "webdriver"))
from webdriver import exceptions


class OpenAndCloseWindowTest(base_test.WebDriverBaseTest):
    def setUp(self):
        self.driver.get(self.webserver.where_is("command_contexts/res/first-page.html"))

    def tearDown(self):
        handles = self.driver.get_window_handles()

        for i in range(len(handles) - 1):
            self.driver.switch_to_window(handles[i])
            self.driver.close()

        self.driver.switch_to_window(self.driver.get_window_handles()[0])

    def test_open_new_window(self):
        handles = self.driver.get_window_handles()
        self.driver.find_element_by_id("open_new_window").click()
        self.assertEquals(len(handles) + 1, len(self.driver.get_window_handles()))

    def test_get_window_handles_returns_the_windows_that_have_been_opened(self):
        self.driver.find_element_by_id("open_new_window").click()
        handles = self.driver.get_window_handles()
        self.driver.switch_to_window(handles[0])
        url1 = self.driver.get_current_url()
        self.driver.switch_to_window(handles[1])
        url2 = self.driver.get_current_url()

        if url1 == self.webserver.where_is("controlling_windows/res/other-page.html"):
            self.assertEquals(url2, self.webserver.where_is("controlling_windows/res/first-page.html"))
        elif url1 == self.webserver.where_is("controlling_windows/res/first-page.html"):
            self.assertEquals(url2, self.webserver.where_is("controlling_windows/res/other-page.html"))
        else:
            self.fail("The wrong set of URLs were returned")

    def test_close_window(self):
        open_windows = len(self.driver.get_window_handles())

        self.driver.find_element_by_id("open_new_window").click()
        self.assertEquals(1 + open_windows, len(self.driver.get_window_handles()))

        self.driver.close()
        self.assertEquals(open_windows, len(self.driver.get_window_handles()))

    def test_command_sent_to_closed_window_returns_no_such_window_exception(self):
        self.driver.find_element_by_id("open_new_window").click()
        self.driver.close()

        with self.assertRaises(exceptions.NoSuchWindowException):
            self.driver.get_window_handle()

if __name__ == "__main__":
    unittest.main()