环境搭建
(1)安装python,配置环境变量,安装pip
(2)安装selenium:pip install selenium
(3)浏览器准备
①安装浏览器驱动
- Chrome驱动文件下载https://chromedriver.storage.googleapis.com/index.html?path=2.35/
- Firefox驱动文件下载Releases · mozilla/geckodriver · GitHub
②下载后把驱动文件放到目录下
③配置环境变量:我的电脑–>属性–>系统设置–>高级–>环境变量–>系统变量–>Path,将“驱动器地址(比如F:\GeckoDriver)”目录添加到Path的值中
(4)验证:使用selenium打开浏览器
from selenium import webdriver
browser = webdriver.Firefox() # Firefox浏览器
browser = webdriver.Chrome() # Chrome浏览器
browser = webdriver.Ie() # Internet
Explorer浏览器 browser = webdriver.Edge() # Edge浏览器
browser = webdriver.Opera() # Opera浏览器
browser = webdriver.PhantomJS() # PhantomJS
browser.quit() #退出并关闭窗口的每一个相关的驱动程序
browser.close() #关闭当前窗口
browser.back() #返回上个界面
如果报错FileNotFoundError: [WinError 2] 系统找不到指定的文件,一般是驱动器设置有问题,可参考报错:“FileNotFoundError: [WinError 2] 系统找不到指定的文件和selenium.common.exceptions.WebDriverException: Messa”_ui自动化driver filenotfounderror系统找不到指定文件-CSDN博客
(5)webdriver简介
webdriver与浏览器的交互方式:利用浏览器原生的API,封装成一套更加面向对象的SeleniumWebDriver API,直接操作浏览器页面里的元素,甚至操作浏览器本身(截屏,窗口大小,启动,关闭,安装插件,配置证书之类的)。由于使用的是浏览器原生的API,速度大大提高,而且调用的稳定性交给了浏览器厂商本身,显然是更加科学。然而带来的一些副作用就是,不同的浏览器厂商,对Web元素的操作和呈现多少会有一些差异,这就直接导致了Selenium WebDriver要分浏览器厂商不同,而提供不同的实现。例如Firefox就有专门的FirefoxDriver,Chrome就有专门的ChromeDriver等等。
现在支持的WebDriver:AndroidDriver,ChromeDriver, FirefoxDriver, HtmlUnitDriver, EventFiringWebDriver,InternetExplorerDriver, IPhoneDriver, IPhoneSimulatorDriver, RemoteWebDriver。
代码:
import unittest
from time import sleep
from selenium import webdriver
class TestBaiDu(unittest.TestCase):
@classmethod
#用setUp,每执行一条测试用例就会启动和关闭一次浏览器,所以此处用setUpClass
def setUpClass(cls):
cls.driver = webdriver.Chrome()
cls.base_url = "https://www.baidu.com/"
# 模块化,将搜索封装成一个方法
def baidu_search(self,search_key):
self.driver.get(self.base_url)
self.driver.find_element_by_id("kw").send_keys(search_key)
self.driver.find_element_by_id("su").click()
sleep(2)
def test_search_key_selenium(self):
search_key = "selenium"
self.baidu_search(search_key)
self.assertEqual(self.driver.title,search_key+"_百度搜索")
def test_search_key_unittest(self):
search_key = "unittest"
self.baidu_search(search_key)
self.assertEqual(self.driver.title,search_key+"_百度搜索")
@classmethod
def tearDownClass(cls):
cls.driver.quit()
if __name__ == '__main__':
unittest.main()
结果:
Ran 2 tests in 17.891s
OK