带unittest的脚本分析.py

import unittest
from selenium import webdriver
class 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()