import unittestfrom selenium import webdriverclass BaiduTest(unittest.TestCase): def setUp(self): self.driver = webdriver.Firefox() self.driver.implicitly_wait(30) self.base_url = "http://www.baidu.com" """ 接下来定义空的verificationErrors数组,脚本运行时的错误信息将被记录到这个数组中。 """ self.verificationErrors = [] """ 定义accept_next alert变量,表示是否继续接受下一个警告,初始状态为True。 """ self.accept_next_alert = True """ setUp用于设置初始化工作,在执行每一个测试用例前先被执行,它与tearDown方法相呼应, 后者在每一个测试用例执行后被执行。这里的初始化工作定义了浏览器启动和基础URL地址。 """ def test_baidu(self): driver = self.driver driver.get(self.base_url + "/") driver.find_element_by_id("kw").clear() driver.find_element_by_id("kw").send_keys("selenium ide") driver.find_element_by_id("su").click() def is_element_present(self,how,what): try: self.driver.find_element(by = how,value=what) except NoSuchElementException as e: return False return True """ is_element_present方法用于查找页面元素是否存在,通过find_element()来接收元素的定位方法(how) 和定位值(what)。如果定位到元素则返回True,否则抛出异常并返回False。 try..except..…为Python 语言的异常处理。 """ def is_alert_present(self): try: self.driver.switch_to_alert() except NoAlertPresentException as e: return False return True """ is_alert_present()方法用于判断当前页面是否存在警告框, 利用WebDriver提供的switch_to_alert()方法来捕捉页面上警告框。如果捕捉到警告框则返回True, 否则将抛出NoAlertPresentException 类型异常,并返回False。 """ """ 不管页面是否出现警告框,返回结果都为True。所以,可以调整该方法为driver.switch_to_alertO.text, 用于获取当前页面上的警告提示信息。可以获取到就返回True,获取不到则返回False。 """ def close_alert_and_get_its_text(self): try: alert = self.driver.switch_to_alert() alert_text = alert.text if self.accept_next_alert: alert.accept() else: alert.dismiss() return alert_text finally: self.accept_next_alert = True """ close_alert_and_get_its_text()关闭警告并获得警告信息。首先通过switch_to_alert()获得警告, 通过text 获得警告框信息。接着通过if 语句判断accept_next_alert的状态,在setUp()中已经初始化状态为True, 如果为True,则通过accept()接受警告,否则dismiss()忽略此警告。 """ def tearDown(self): self.driver.quit() self.assertEqual([],self.verificationErrors) """ tearDown()方法在每个测试方法执行后调用,这个方法用于测试用例执行后的清理工作, 如退出浏览器、关闭驱动,恢复用例执行状态等。 """ """ 在setUp()方法中定义了verificationErors为空数组,这里通过assertEqual()比较其是否为空, 如果为空则说明用例执行的过程中没有出现异常,否则将抛出AssertionError异常。 """if __name__ == '__main__': unittest.main()
来源:https://www.cnblogs.com/zhang-da/p/12210761.html