Selenium 模块化（一）

#!/usr/bin/python
# -*- coding:utf-8

from selenium import webdriver
import unittest,time

class baidu(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Chrome()
        self.driver.implicitly_wait(30)
        self.url = "http://www.baidu.com"
        self.driver.get(self.url)

    # 百度搜索
    def test_Baidu(self):
        try:
            driver = self.driver
            driver.find_element_by_id("kw").clear()
            driver.find_element_by_id("kw").send_keys("Selenium")
            driver.find_element_by_id("su").click()
            time.sleep(5) # 强制等待，查看界面是否正确
        except:
            driver.get_screenshot_as_file(u"D:\\Bug\\错误截图.jpg") # 错误截图
    def tearDown(self):
        self.driver.close()
        self.driver.quit()

# 执行程序
if __name__ == '__main__':
    unittest.main

上述为百度搜索模块实例化。

主要的实现方式为导入 unittest 测试框架。

setUp：程序执行时，自动会运行的其中的代码。所有在这个方法当中，通常定义一些公共的参数变量。

tearDown：程序执行结束时，自动会运行其中的代码。在该方法中通常是关闭程序当中所打开的一些资源、进行等。

body：主体内容。主要以 test_命名开头。因为 unittest 无法识别其他命名开头的方法。主要用来编写脚本的执行内容。