#encoding=utf-8

import unittest

import time

from selenium import webdriver

from selenium.webdriver import ActionChains

class VisitSogouByIE(unittest.TestCase):

    def setUp(self):

        #启动IE浏览器

        #self.driver = webdriver.Firefox(executable_path = "e:\\geckodriver")

        self.driver = webdriver.Ie(executable_path = "e:\\IEDriverServer")

    def test_HandleFrame(self):

        from selenium.webdriver.support import expected_conditions as EC

        from selenium.webdriver.support.ui import WebDriverWait

        from selenium.common.exceptions import TimeoutException

        url = "http://127.0.0.1/frameset.html"

        # 访问自动以测试网页

        self.driver.get(url)

        # 使用索引方式进入指定的frame页面，索引号从0开始。

        # 所以想进入中间的frame，需要使用索引号1

        # 如果没有使用此行代码，则无法找到页面中左侧frame中的任何页面元素

        self.driver.switch_to.frame(0)

        # 找到左侧frame中的p标签元素

        leftFrameText = self.driver.find_element_by_xpath("//p")

        # 断言左侧frame中的文字是否和“这是左侧 frame 页面上的文字”几个关键字相一致

        self.assertAlmostEqual(leftFrameText.text, u"这是左侧 frame 页面上的文字")

        # 找到左侧frame中的按钮元素，并点击该元素

        self.driver.find_element_by_tag_name("input").click()

        try:

            # 动态等待alert窗体出现

            alertWindow = WebDriverWait(self.driver, 10).until(EC.alert_is_present())

            # 打印alert消息

            print alertWindow.text

            alertWindow.accept()

        except TimeoutException, e:

            print e

        # 使用driver.switchTo.default_content方法，从左侧frame中返回到frameset页面

        # 如果不调用此行代码，则无法从左侧frame页面中直接进入其他frame页面

        self.driver.switch_to.default_content()

        # 通过标签名找到页面中所有的frame元素，然后通过索引进入该frame

        self.driver.switch_to.frame(self.driver.find_elements_by_tag_name("frame")[1])

        # 断言页面源码中是否存在“这是中间 frame 页面上的文字”关键子串

        assert u"这是中间 frame 页面上的文字" in self.driver.page_source

        # 再输入框中输入“我在中间frame”

        self.driver.find_element_by_tag_name("input").send_keys(u"我在中间frame")

        self.driver.switch_to.default_content()

        self.driver.switch_to.frame(self.driver.find_element_by_id("rightframe"))

        assert u"这是右侧 frame 页面上的文字" in self.driver.page_source

        self.driver.switch_to.default_content()

        self.driver.switch_to.frame("rightframe")

        assert u"这是右侧 frame 页面上的文字" in self.driver.page_source

        self.driver.switch_to.default_content()

    def tearDown(self):

        # 退出IE浏览器

        self.driver.quit()

if __name__ == '__main__':

    unittest.main()