Python+Selenium unittest单元测试框架

1. 新建一个testbaidu.py的文件
2. 导入unittest模块
3. 当前测试类继承unittest.TestCase，相当于当前利用unittest创建了一个test case，这个test case是能够被unittest直接识别。
4. 写setUP(),主要是打开浏览器和打开站点
5. 写一个test_search（）用例写搜索的代码
6. 写tearDown(),主要是浏览器退出操作

# coding=utf-8
import time
import unittest
from selenium import webdriver
 
 
class BaiduSearch(unittest.TestCase):
 
    def setUp(self):
        """
        测试固件的setUp()的代码，主要是测试的前提准备工作
        :return:
        """
        self.driver = webdriver.Chrome("C:\\Users\suxiahua\AppData\Local\Programs\Python\Python36\Lib\site-packages\chromedriver_win32\chromedriver.exe")
        self.driver.maximize_window()
        time.sleep(3)
        self.driver.get("https://www.baidu.com")
 
    def tearDown(self):
        """
        测试结束后的操作，这里基本上都是关闭浏览器
        :return:
        """
        self.driver.quit()
 
    def test_baidu_search(self):
        """
        这里一定要test开头，把测试逻辑代码封装到一个test开头的方法里。
        :return:
        """
        self.driver.find_element_by_id('kw').send_keys('selenium')
        time.sleep(1)
        try:
            assert 'selenium' in self.driver.title
            print ('Test Pass.')
        except Exception as e:
            print ('Test Fail.', format(e))
 
if __name__ == '__main__':
    unittest.main()