selenium简单代码入门

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

import os,sys,string
import time
import unittest
from selenium import webdriver
from selenium.webdriver.common.by import By

#unittest是python下的单元测试模块
class WeiboDate(unittest.TestCase):

    #初始化,并使用Firefox浏览器驱动
    def setUp(self):
        self.driver = webdriver.Firefox()

    def test_WeiboSearch(self):
        driver = self.driver
        driver.get("http://data.weibo.com/index/")
        #wait
        driver.implicitly_wait(10)        
        driver.find_element(By.XPATH,"//div[@id='pl_index_searchMain']//input").send_keys("apple watch")
        driver.find_element_by_css_selector("a.index-search").click()
        time.sleep(3)
        #截图
        driver.get_screenshot_as_file("/opt/selenium/script/jietu.png")
        #最大化浏览器
        driver.maximize_window()    
        driver.forward()
        #cookie
        cookie = driver.get_cookies()
        file=open('cookie','w+').writelines("cookie")

    def tearDown(self):
        self.driver.close()

if __name__ == "__main__":
    unittest.main()