第 4 章 WebDriver API

4.15 操作 Cookie

有时我们需要验证浏览器中的 Cookie 是否正确，因为基于真实的 Cookie 是无法通过白盒测试和集成测试的。WebDriver 提供了操作 Cookie 的相关方法，可以读取、添加和删除 Cookie。

WebDriver 操作 Cookie 的方法如下。
● get_cookies（）：获得所有 Cookie。
● get_cookie（name）：返回字典中 key 为「name」的 Cookie。
● add_cookie（cookie_dict）：添加 Cookie。
● delete_cookie（name，optionsString）：删除名为 OpenString 的 Cookie。
● delete_all_cookies（）：删除所有 Cookie。

下面通过 get_cookies()获取当前浏览器的所有 Cookie。

"""
* get_cookies() 获得所有 cookie 信息。
* get_cookie(name) 返回字典的 key 为“ name”的 cookie 信息。
* add_cookie(cookie_dict) 添加 cookie。“ cookie_dict”指字典对象，必须有name和value 值。
* delete_cookie(name,optionsString)  删除 cookie 信息。“name”是要删除的 cookie 的名称，
                                   “optionsString”是该 cookie 的选项，目前支持的选项包括“路径”，“域”。
* delete_all_cookies() 删除所有 cookie 信息。
"""
from selenium import webdriver

driver = webdriver.Chrome()
driver.get("http://www.baidu.com")

# 获得所有cookie信息并打印
cookie = driver.get_cookies()
print(cookie)

# 添加cookie信息
driver.add_cookie({'name': 'key-aaaaaaa', 'value': 'value-bbbbbb'})

# 遍历指定的cookies
for cookie in driver.get_cookies():
    print("%s -> %s" % (cookie['name'], cookie['value']))

driver.quit()

执行结果如下。

从执行结果可以看出，Cookie 中的数据是以字典形式存放的。知道了 Cookie 中数据的存放形式后，即可按照这种形式向浏览器中添加 Cookie。

# 添加cookie信息
driver.add_cookie({'name': 'key-aaaaaaa', 'value': 'value-bbbbbb'})

# 遍历指定的cookies
for cookie in driver.get_cookies():
    print("%s -> %s" % (cookie['name'], cookie['value']))

执行结果如下：

从执行结果可以看出，最后一条 Cookie 是在脚本执行过程中通过 add_cookie()方法添加的。通过遍历得到所有的 Cookie，从而找到字典中 key 为「name」和「value」的 Cookie 值。
delete_cookie() 和 delete_all_cookies() 方法的使用也很简单，前者通过 name 删除一个指定的 Cookie，后者直接删除浏览器中的所有 Cookies。

4.16 调用 JavaScript

有些页面操作不能依靠 WebDriver 提供的 API 来实现，如浏览器滚动条的拖动。这时就需要借助 JavaScript 脚本。WebDriver 提供了 execute_script()方法来执行 JavaScript 代码。
用于调整浏览器滚动条位置的 JavaScript 代码如下。

 

window.scrollTo()方法用于设置浏览器窗口滚动条的水平位置和垂直位置。第一个参数表示水平的左边距，第二个参数表示垂直的上边距，代码如下。

16_js_window_scroll.py

"""
* execute_script() 调用JavaScript实现滚动条操作。
"""
from selenium import webdriver
from time import sleep

driver = webdriver.Chrome()
driver.get("http://www.baidu.com")

driver.set_window_size(800, 600)

driver.find_element_by_id("kw").send_keys("selenium")
driver.find_element_by_id("su").click()
sleep(2)

# 通过javascript设置浏览器窗口的滚动条位置
js = "window.scrollTo(100,450);"
driver.execute_script(js)
sleep(3)

driver.quit()

首先，在浏览器中打开百度，搜索「selenium」，通过 set_window_size()方法将浏览器窗口设置为固定宽、高显示，目的是让窗口出现水平和垂直滚动条。然后，通过 execute_script()方法执行 JavaScript 代码来控制浏览器滚动条的位置，如图 4-8 所示。

当然，JavaScript 的作用不仅仅体现在浏览器滚动条的操作上，它还可以在页面中的 textarea 文本框中输入内容，如图 4-9 所示。

文本框的 HTML 代码如下。

虽然可以通过 id 定位到元素，但是不能通过 send_keys()在文本框中输入文本信息。在这种情况下，可以借助 JavaScript 代码输入文本信息。

首先，定义要输入的内容 text。然后，将 text 与 JavaScript 代码通过「+」进行拼接，这样做的目的是为了方便自定义输入内容。最后，通过 execute_script()执行 JavaScript 代码。

4.17 处理 HTML5 视频播放

HTML5 技术非常流行，主流的浏览器都支持 HTML5，使得越来越多的开发者开始使用 HTML5。
WebDriver 支持在指定的浏览器上测试 HTML5，另外，还可以使用 JavaScript 测试这些功能，这样就可以在任意浏览器上测试 HTML5 了。

大多数浏览器使用插件（如 Flash）播放视频，但是，不同的浏览器需要使用不同的插件。HTML5 定义了一个新的元素 <video>，指定了一个标准的方式嵌入电影片段。
IE9+、Firefox、Opera、Chrome 都支持元素 <video>。

下面介绍如何自动化测试 <video>，<video> 提供了 JavaScript 接口和多种方法及属性。

"""
测试HTML5页面
"""
from time import sleep
from selenium import webdriver

driver = webdriver.Chrome()
driver.get("http://videojs.com/")

video = driver.find_element_by_id("preview-player_html5_api")

# 返回播放文件地址
url = driver.execute_script("return arguments[0].currentSrc;", video)
print(url)

# 播放视频
print("start")
driver.execute_script("arguments[0].play()", video)

# 播放15秒钟
sleep(15)

# 暂停视频
print("stop")
driver.execute_script("arguments[0].pause()", video)

driver.quit()

JavaScript 有个内置的对象叫作 arguments。arguments 包含了函数调用的参数数组，[0]表示取对象的第 1 个值。
currentSrc 返回当前音频/视频的 URL。如果未设置音频/视频，则返回空字符串。
load()、play()和 pause() 控制视频的加载、播放和暂停。

4.18 滑动解锁

滑动解锁是目前比较流行的解锁方式，如图 4-11 所示。

图 4-11 滑动解锁
当我们单击滑块时，改变的只是 CSS 样式，HTML 代码段如下。

slide-to-unlock-handle 表示滑块。在滑动过程中，滑块的左边距会逐渐变大，因为它在向右移动。
slide-to-unlock-progress 表示滑过之后的背景色，背景色的区域会逐渐增加，因为滑块在向右移动。

18_slide_unlock.py

"""
滑动解锁
"""
from time import sleep
from selenium import webdriver
from selenium.webdriver import ActionChains
from selenium.common.exceptions import UnexpectedAlertPresentException

driver = webdriver.Chrome()
driver.get("https://www.helloweba.com/demo/2017/unlock/")

# 定位滑动块
slider = driver.find_elements_by_class_name("slide-to-unlock-handle")[0]
action = ActionChains(driver)
action.click_and_hold(slider).perform()

for index in range(200):
    try:
        action.move_by_offset(2, 0).perform()
    except UnexpectedAlertPresentException:
        break
    action.reset_actions()
    sleep(0.1)  # 等待停顿时间


# 打印警告框提示
success_text = driver.switch_to.alert.text
print(success_text)

在这个脚本中，用到下面几个方法。
● click_and_hold（）：单击并按下鼠标左键，在鼠标事件中介绍过。
● move_by_offset（）：移动鼠标，第一个参数为x坐标距离，第二个参数为y坐标距离。
● reset_action（）：重置 action。
执行完成，滑动效果如图 4-12 所示。

接下来，再看另外一种应用，上下滑动选择日期，如图 4-13 所示。

参考前面的操作，通过 ActionChains 类可以实现上下滑动选择日期，但是这里要介绍另外一种方法，即通过 TouchActions 类实现上下滑动选择日期。

18_slide_data.py

"""
滑动选择日历
"""
from time import sleep
from selenium import webdriver


driver = webdriver.Chrome()

driver.get("http://www.jq22.com/yanshi4976")
sleep(2)
driver.switch_to.frame("iframe")
driver.find_element_by_id("appDate").click()

# 定位要滑动的年、月、日
dwwos = driver.find_elements_by_class_name("dwwo")
year = dwwos[0]
month = dwwos[1]
day = dwwos[2]

action = webdriver.TouchActions(driver)
action.scroll_from_element(year, 0, 5).perform()
action.scroll_from_element(month, 0, 30).perform()
action.scroll_from_element(day, 0, 30).perform()
# ……

这里使用 TouchActions 类中的 scroll_from_element()方法滑动元素，参数如下。

● on_element：滑动的元素。
● xoffset：x坐标距离。
● yoffset：y坐标距离。

4.19 窗口截图

自动化测试用例是由程序执行的，因此有时候打印的错误信息不够直观。如果在脚本执行出错时能够对当前窗口进行截图并保存，那么通过截图就可以非常直观地看到脚本出错的原因。WebDriver 提供了截图函数 save_screenshot ()，可用来截取当前窗口。

19_screenshot.py

"""
* save_screenshot() 截取窗口图片。
"""
from selenium import webdriver

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

# 对当前窗口截图，并指定图片的保存位置
driver.save_screenshot("./files/baidu_img.png")

WebDriver 建议使用 png 作为图片的后缀名。脚本运行完成后，会在当前 files/目录中生成 baidu_img.png 图片。

4.20 关闭窗口

在前面的例子中一直使用 quit()方法，其含义为退出相关的驱动程序和关闭所有窗口。
除此之外，WebDriver 还提供了 close()方法，用来关闭当前窗口。例如，在 4.10 节中会打开多个窗口，当准备关闭其中某个窗口时，可以用 close()方法。

driver.quit()