Selenium WebDriver- 指定页面加载时间

#encoding=utf-8import unittestimport timefrom selenium import webdriverfrom selenium.webdriver import ActionChainsfrom selenium.common.exceptions import TimeoutExceptionfrom selenium.webdriver.common.keys import Keysclass setPageLoadTime(unittest.TestCase): ???def setUp(self): ???????# 启动火狐浏览器 ???????self.driver = webdriver.Ie(executable_path = "e:\\IEDriverServer") ???def test_PageLoadTime(self): ???????# 设定页面加载限制时间为4秒 ???????self.driver.set_page_load_timeout(4) ???????self.driver.maximize_window() ???????try: ???????????startTime = time.time() ???????????self.driver.get("http://mail.126.com") ???????except TimeoutException: ???????????print u‘页面加载超过设定时间，超时‘ ???????????# 当页面加载时间超过设定时间， ???????????# 通过执行Javascript来stop加载，然后继续执行后续动作 ???????????self.driver.execute_script(‘window.stop()‘) ???????end = time.time() - startTime ???????print end ???????# 切换进frame控件 ???????self.driver.switch_to.frame("x-URS-iframe") ???????# 获取用户名输入框 ???????userName = self.driver.find_element_by_xpath(‘//input[@name="email"]‘) ???????# 输入用户名 ???????userName.send_keys("xxx") ???????# 获取密码输入框 ???????pwd = self.driver.find_element_by_xpath("//input[@name=‘password‘]") ???????# 输入密码 ???????pwd.send_keys("xxx") ???????# 发送一个回车键 ???????pwd.send_keys(Keys.RETURN) ???????time.sleep(5) ???assert u"退出" in self.driver.page_source ???def tearDown(self): ???????self.driver.quit()if __name__ == ‘__main__‘: ???unittest.main()