selenium 知识汇总

[TOC]

Install

pip install selenium
selenium 是自动测试工具，支持各种界面式浏览器，实际上就是在这些浏览器里面安装一个 selenium 的插件，方便地实现 Web 界面测试，即 selenium 支持这些浏览器驱动。PhantomJS 也是一个浏览器，selenium 也能够支持。

Demo

from selenium import webdriver

browser = webdriver.Chrome()
browser.get('http://www.baidu.com/')

驱动文件
放在 path 路径下

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

driver = webdriver.Chrome()
driver.get("http://www.python.org")
assert "Python" in driver.title
elem = driver.find_element_by_name("q")
elem.send_keys("pycon")
elem.send_keys(Keys.RETURN)
print driver.page_source

测试用例：

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

class PythonOrgSearch(unittest.TestCase):

    def setUp(self):
        self.driver = webdriver.Chrome()

    def test_search_in_python_org(self):
        driver = self.driver
        driver.get("http://www.python.org")
        self.assertIn("Python", driver.title)
        elem = driver.find_element_by_name("q")
        elem.send_keys("pycon")
        elem.send_keys(Keys.RETURN)
        assert "No results found." not in driver.page_source

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

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

测试用例是继承了 unittest.TestCase 类，继承这个类表明这是一个测试类。setUp方法是初始化的方法，这个方法会在每个测试类中自动调用。每一个测试方法命名都有规范，必须以 test 开头，会自动执行。最后的 tearDown 方法会在每一个测试方法结束之后调用。这相当于最后的析构方法。在这个方法里写的是 close 方法，你还可以写 quit 方法。不过 close 方法相当于关闭了这个 TAB 选项卡，然而 quit 是退出了整个浏览器。当你只开启了一个 TAB 选项卡的时候，关闭的时候也会将整个浏览器关闭。

页面交互

假如有一个表单输入框：
<input type="text" name="passwd" id="passwd-id" />

元素选取

单个元素选取：

  1. find_element_by_id
  2. find_element_by_name
  3. find_element_by_xpath
  4. find_element_by_link_text
  5. find_element_by_partial_link_text
  6. find_element_by_tag_name
  7. find_element_by_class_name
  8. find_element_by_css_selector

多个元素选取：

  1. find_elements_by_name
  2. find_elements_by_xpath
  3. find_elements_by_link_text
  4. find_elements_by_partial_link_text
  5. find_elements_by_tag_name
  6. find_elements_by_class_name
  7. find_elements_by_css_selector

还可以用 By 类确定哪种选择方式：

from selenium.webdriver.common.by import By

driver.find_element(By.XPATH, '//button[text()="Some text"]')
driver.find_elements(By.XPATH, '//button')

By 类的一些属性如下：

ID = "id"
XPATH = "xpath"
LINK_TEXT = "link text"
PARTIAL_LINK_TEXT = "partial link text"
NAME = "name"
TAG_NAME = "tag name"
CLASS_NAME = "class name"
CSS_SELECTOR = "css selector"

可以获取这个元素：

element = driver.find_element_by_id("passwd-id")
element = driver.find_element_by_name("passwd")
element = driver.find_elements_by_tag_name("input")
element = driver.find_element_by_xpath("//input[@id='passwd-id']")

element = driver.find_element_by_id("q")
element.size() //大小 dict
element.get_attribute("type") //获取元素属性
element.is_displayed() //是否可见
element.text //获取文本信息

driver.current_url //当前url

用 xpath 时需要注意，如果有多个元素匹配了 xpath，只会返回第一个匹配的元素。如果没有找到，会抛出 NoSuchElementException 的异常。

填充表单

向文本输入内容

element.send_keys("some text")
还可以利用 Keys 这个类模拟点击某个按键
element.send_keys("and some", Keys.ARROW_DOWN)

element.clear() //输入的文本会被清除

Select 方法

下拉选项卡处理：

element = driver.find_element_by_xpath("//select[@name='name']")
all_options = element.find_elements_by_tag_name("option")
for option in all_options:
    print("Value is: %s" % option.get_attribute("value"))
    option.click()

WebDriver 提供更简单的方法 Select：

from selenium.webdriver.support.ui import Select
select = Select(driver.find_element_by_name('name'))
select.select_by_index(index)
select.select_by_visible_text("text")
select.select_by_value(value)

可以根据索引来选择，根据值来选择，根据文字来选择。

全部取消选择：

select = Select(driver.find_element_by_id('id'))
select.deselect_all()

获取所有已选选项：

select = Select(driver.find_element_by_xpath("xpath"))
all_selected_options = select.all_selected_options

获取所有可选选项：

options = select.options

提交表单

driver.find_element_by_id("submit").click()
相当于模拟点击 submit 按钮提交表单
也可以单独提交某个元素
element.submit() WebDriver 会在表单中寻找该元素所在表单，如果发现该元素没有被表单包围，程序会抛出 NoSuchElementException

元素拖拽

指定被拖动的元素和拖动目标元素，然后利用 ActionChains 类实现

element = driver.find_element_by_name("source")
target = driver.find_element_by_name("target")

from selenium.webdriver import ActionChains
action_chains = ActionChains(driver)
action_chains.drag_and_drop(element, target).perform()

元素从 source 拖动到 target

页面窗口切换

一个浏览器肯定有很多窗口，所以肯定有方法实现窗口的切换：
driver.switch_to_window("windowName")
另外可以使用 window_handles 方法获取每个窗口的操作对象：

for handle in driver.window_handles:
    driver.switch_to_window(handle)

切换 frame：
driver.switch_to_frame("frameName.0.child")
之后焦点会切换到 name 为 child 的 frame 上

弹窗处理

当触发了某个事件之后，页面出现了弹框提示，如何处理呢？
alert = driver.switch_to_alert() 可以获取弹窗对象
.get_text()
.accept() //点击确定
.dismiss() //点击取消
send_keys() //输入值

历史纪录

前进和后台：

driver.forward()
driver.back()

Cookies处理

给页面添加 Cookies：

# Go to the correct domain
driver.get("http://www.example.com")

# Now set the cookie. This one's valid for the entire domain
cookie = {'name' : 'foo', 'value' : 'bar'}
driver.add_cookie(cookie)

获取页面 Cookies：

# Go to the correct domain
driver.get("http://www.example.com")

# And now output all the available cookies for the current URL
driver.get_cookies()

页面等待

这是非常重要的一部分，现代网页越来越多采用 Ajax 技术，这样程序就不能确定某个元素何时加载完毕，会让元素定位困难并且报出 ElementNotVisibleException 的概率提高
隐式等待：等待特定的时间
显式等待：指定某一条件，直到这个条件成立时继续执行

from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
try:
    element = WebDriverWait(driver, 10).until(
        EC.presence_of_element_located((By.ID, "myDynamicElement"))
    )
finally:
    driver.quit()

显式等待

显式等待指定某个条件，设置最长等待时间，如果这个时间内还没有找到元素，便会抛出异常

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

driver = webdriver.Chrome()
driver.get("http://somedomain/url_that_delays_loading")
try:
    element = WebDriverWait(driver, 10).until(
        EC.presence_of_element_located((By.ID, "myDynamicElement"))
    )
finally:
    driver.quit()

程序默认会 500ms 调用一次来查看元素是否已经生成，如果本来元素就是存在的，那么会立即返回
下面是一些内置的等待条件，可以直接调用这些条件，而不用自己写某些等待条件了：

title_is
title_contains
presence_of_element_located
visibility_of_element_located
visibility_of
presence_of_all_elements_located
text_to_be_present_in_element
text_to_be_present_in_element_value
frame_to_be_available_and_switch_to_it
invisibility_of_element_located
element_to_be_clickable – it is Displayed and Enabled.
staleness_of
element_to_be_selected
element_located_to_be_selected
element_selection_state_to_be
element_located_selection_state_to_be
alert_is_present

from selenium.webdriver.support import expected_conditions as EC

wait = WebDriverWait(driver, 10)
element = wait.until(EC.element_to_be_clickable((By.ID,'someid')))

隐式等待

隐式等待比较简单，就是简单地设置一个等待时间，单位为秒。

from selenium import webdriver

driver = webdriver.Chrome()
driver.implicitly_wait(10) # seconds
driver.get("http://somedomain/url_that_delays_loading")
myDynamicElement = driver.find_element_by_id("myDynamicElement")

Demo

from selenium import webdriver
from selenium.webdriver.common.keys import Keys
driver = wendriver.Firefox()
driver.get("http://www.python.org")
assert "Python" in
elem = driver.find_element_by_name("q")
elem.clear()
elem.send_keys("pycon")
elem.send_keys(Keys.RETURN)
assert "No results found." not in driver.page_source
driver.close()

Http 协议分析

GET / HTTP/1.1\r\n
Host: rk-zyk.zx.ga\r\n
Connection: Keep-Alive\r\n

例子

from selenium import webdriver
from selenium.webdriver.common.action_chains import ActionChains

driver = webdriver.Firefox()
driver.get('http://www.baidu.com')
driver.maximize_window()

setting = driver.find_element_by_link_text(u'设置')
ActionChains(driver).move_to_element(setting).perform()
driver.find_element_by_link_text(u'高级搜索'),click()
最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容