selenium-unittest框架介绍（一）

unittest是python自带的单元测试框架，下面通过代码介绍unittest框架

#coding=utf-8
from selenium import webdriver
import unittest
from time import sleep

class TestBaiDu(unittest.TestCase):

    def setUp(self):
        """
        用于测试用例执行前的初始化工作
        :return: 
        """
        self.driver = webdriver.Chrome()
        driver = self.driver

        #浏览器最大化
        driver.maximize_window()
        driver.get("https://www.baidu.com")
        driver.implicitly_wait(10)

    def test_baidu_search(self):
        """
        测试用例必须以test开头
        :return: 
        """
        driver = self.driver
        #进入百度首页后输入“selenium”，点击搜索
        driver.find_element_by_id("kw").send_keys("selenium")
        driver.find_element_by_id("su").click()

        sleep(2)
        #获取输入框的信息，并与之前的信息进行比较
        mc = driver.find_element_by_id("kw").get_attribute('value')
        #判断mc与selenium是否相等
        self.assertEqual(mc, "selenium")

    def tearDown(self):
        """
        执行测试用例的收尾工作，一般为关闭浏览器
        :return: 
        """
        driver =self.driver
        #关闭浏览器
        driver.quit()

if __name__ == '__main__':
    """
    main()方法可以将一个单元测试模块变为一个可独立执行的脚本。
    该方法会执行以test开头的方法
    """
    unittest.main()