1 #-*- coding:utf-8 -*- 2 __author__ = "carry" 3 4 from selenium import webdriver 5 from selenium.webdriver.common.by import By 6 from selenium.webdriver.common.keys import Keys 7 from selenium.webdriver.support.ui import Select 8 from selenium.common.exceptions import NoSuchAttributeException 9 import unittest,time,re 10 import HTMLTestRunner #引入HTMLTestRunner 11 class Baidu(unittest.TestCase): 12 def setUp(self): 13 self.driver = webdriver.Firefox() 14 self.driver.implicitly_wait(30) 15 self.base_url = "http://www.baidu.com" 16 self.verificationErrors = [] 17 self.accept_next_alert =True 18 #百度搜索用例 19 def test_baidu_search(self): 20 driver = self.driver 21 driver.get(self.base_url+"/") 22 driver.find_element_by_id("kw").send_keys("selenium") 23 driver.find_element_by_id("su").click() 24 time.sleep(2) 25 driver.close() 26 #百度设置用例 27 def test_baidu_set(self): 28 driver = self.driver 29 #进入搜素设置页 30 driver.get(self.base_url+"/gaoji/preferences.html") 31 #设置每页搜索结果为50页 32 m = driver.find_element_by_name("NR") 33 m.find_element_by_xpath("//option[@value='50']").click() 34 time.sleep(2) 35 #保存设置的信息 36 driver.find_element_by_xpath("/html/body/form/div/input").click() 37 time.sleep(2) 38 driver.switch_to_alert().accept() 39 def tearDown(self): 40 self.driver.quit() 41 self.assertEqual([],self.verificationErrors) 42 if __name__ == "__main__": 43 #定义一个单元测试容器 44 testunit = unittest.TestSuite() 45 #将测试用例加入到测试容器中 46 testunit.addTest(Baidu("test_baidu_search")) 47 testunit.addTest(Baidu("test_baidu_set")) 48 #定义一个报告存放路径，支持相对路径 49 filename = r'D:selenium_file esult.html' 50 fp = file(filename,'wb') 51 #定义测试报告 52 runner = HTMLTestRunner.HTMLTestRunner( 53 stream=fp, 54 title=u'百度搜索测试报告', 55 description=u'用例执行情况：' 56 ) 57 #运行测试用例 58 runner.run(testunit)