selenium 編程入門

一個簡單的例子

from selenium import webdriver
from selenium.webdriver.common.keys import Keys
import time

driver = webdriver.Chrome()	# 創建瀏覽器對象
driver.get("http://www.baidu.com")	# 向百度發送請求
assert "百度一下，你就知道" in driver.title	
kw = driver.find_element_by_id("kw")	# 獲取搜索框
time.sleep(1)	# 等待1秒鐘
kw.send_keys("自動化測試")	# 向搜索框中輸入字符串
assert "No results found." not in driver.page_source
driver.close()	# 關閉瀏覽器

使用selenium與unittest編寫單元測試

import unittest
from selenium import webdriver
from selenium.webdriver.common.keys import Keys

class SeleniumTestCaseDemo(unittest.TestCase):
    def setUp(self):
    	"""起始方法"""
        self.driver = webdriver.Firefox()

    def test_search_in_python_org(self):
    	"""測試用例方法, 必須以test開頭才能被識別爲測試用例方法"""
        driver = self.driver
        driver.get("http://www.python.org")
        self.assertIn("Python", driver.title)
        elem = driver.find_element_by_name("q")
        elem.send_keys("pygame")
        elem.send_keys(Keys.RETURN)
        assert "results found." not in driver.page_source

    def tearDown(self):
    	"""結束方法"""
        self.driver.close()

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