Selenium 系列篇（三）：窗口篇

1. 窗口操作

Selenium 自动化过程中，对窗口本身的操作包含：打开一个窗口、关闭一个窗口、刷新一个窗口、窗口的回退和前进。

其中，打开一个窗口只需要使用 driver.get( url) 方法，传入一个网页地址，即可以驱动浏览器打开目标网址。

# 打开一个窗口网站
driver.get("http://www.baidu.com")

在多窗口页面切换时，WebDriver 提供了 API ，包含：back()、forward()、refresh() 方便我们对浏览器执行回退、前进、刷新页面的操作。

# 打开一个窗口网站
driver.get("http://www.baidu.com")

# 打开第二个网站
driver.get("http://www.google.com")

# 回退到第一个窗口网站
driver.back()

# 前进到第二个窗口网站
driver.forward()

# 刷新当前页面
driver.refresh()

默认窗体是全屏显示，也可以显式指定窗口展示的宽和高。

# 设定窗口的宽为1000，高为800
driver.set_window_size(1000, 800)

为了测试报告更直观，往往需要截图附件，Selenium 可以随时对某一个窗口页面进行截图。

# 保存当前页面为图片
driver.save_screenshot('./temp.png')

自动化完成之后，我们一般需要关闭窗口，使用 close() 可以关闭当前窗口，quit() 退出驱动程序，关闭所有页面。

# 关闭当前窗口页面
driver.close()

# 关闭所有窗口页面
driver.quit()

2. 内部元素操作

窗口内部元素使用最多的操作包含：点击按钮、输入框的清空和设置内容。

# 点击某个按钮
driver.find_element_by_id("element_btn").click()

# 清空输入框
driver.find_element_by_id("element_edit").clear()

# 输入框内设置内容
driver.find_element_by_id("element_edit").send_keys("Python自动化测试社区")

获取到一个元素之后，就能拿到元素的内容和全部属性。

# 首先查找到网页元素
norm_element = driver.find_element_by_id("element_norm")

# 获取元素的尺寸大小
print(norm_element.size)

# 获取元素的所有属性，比如：type属性
print(norm_element.get_attribute("type"))

3. 等待操作

自动化打开一个网页的时候，内部网页元素加载完全有一点的延迟性，因此在做 Web 端的自动化测试的时候，一般都需要在测试 case 时加入一些等待操作。

常用等待操作有 3 种，分别是：sleep、隐式等待、显示等待

其中，sleep(timeout) 是设定一个固定的等待时长，强行进行等待，使用方便的同时，效率最低，不建议使用。

# 强行等待 10s
sleep(10)

隐式等待也是设定一个固定的等待时间，对整个生命周期的元素都起作用，每一个元素都会等待加载完全，直到超过设定的等待时间。

# 隐式等待设定时长为5s
driver.implicitly_wait(5)

driver.get('http://www.google.com')

# 隐式等待所有元素加载完成,直到超过设定的最长时间
driver.find_element_by_id("element_id").click()

显示等待相比隐式等待更灵活，是先设定一个条件函数和一个最长等待时长，轮询判断条件函数的返回值，如果返回 True，则开始执行后面的操作，否则会一直等待，直到超时报元素未找到异常。

Selenium 中使用 expected_conditions 指定了很多条件函数（也可以自定义条件函数），具体可以参考官网。

# 使用方式
# 指定超时时间和条件函数
WebDriverWait(driver,timeout).until(method)

比如：下面就是轮询等待 10s，直到页面元素可见。

# 显示等待 10s，直到元素出现
element = WebDriverWait(driver, 10).until(
    EC.visibility_of_element_located((By.ID, "element_id"))
)

上面的 3 种等待，显式等待和隐式等待使用更常见；隐式等待针对全局，可以动态的设置等待时长；显式等待最灵活，可以最大程度地提高测试用例的执行效率。

4. 内嵌页面

有一些复杂的网页会包含 iframe HTML 内联框架元素，WebDriver API 是没法直接找到 iframe 表单内嵌页面的元素。

如果想定位到内嵌页面的元素，可以使用 switch_to_frame( frame_id/frame) 切换到对应的 iframe ，然后再去查找内部元素。

# 使用 iframe 的 id 或者 iframe 对象切换到 iframe
driver.switch_to.frame("frame_id")
driver.switch_to.frame(target_frame)

# 然后再查找 iframe 内部元素
driver.find_element_by_name("pwd").send_keys("******")

操作完 iframe 之后，使用 switch_to.default_content() 切换到外层页面，继续后面的操作。

# 切换到外层页面
driver.switch_to.default_content()

需要注意的是，针对多层 iframe 嵌套，需要一层一层地切换 iframe，操作完对应 iframe 内部操作之后，返回到最外层页面。

比如：iframeA- iframeB - iframeC，要查找 iframeC 内部元素控件

# 切换到第一层
driver.switch_to.frame("iframeA")

# 切换到第二层
driver.switch_to.frame("iframeB")

切换到第三层
edriver.switch_to.frame("iframeC")

# 操作第三层元素
....

# 切换到最外层元素
driver.switch_to.default_content()

5. 窗口切换

有时候，点击网页内部某个链接，会打开一个新的窗口，实际上，每一个窗口都有特定的窗口句柄：window_handle。

获取当前窗口页面的窗口句柄：current_window_handle。

获取所有窗口页面的窗口句柄：window_handles，返回值是一个窗口句柄列表。

要操作某一个窗口内的元素，首先要使用 switch_to_window() 切换到对应的窗口，然后才能操作内部元素。

# 获取当前窗口的句柄
window_handle_main = driver.current_window_handle

# 跳到其他窗口界面
...

# 获取所有的窗口句柄
window_handles = driver.window_handles

# 其他窗口的句柄
window_handle_other = None

# 遍历找到注册窗口句柄
for handle in window_handles:
    if handle != window_handle_main:
        # 其他窗口句柄
        window_handle_other = handle

        # 切换到第二个窗口内
        driver.switch_to.window(window_handle_other)

        # 第二个窗口内部操作
        ......

        # 关闭当前窗口，即第二个窗口页面
        driver.close()

# 切换到主窗口
driver.switch_to.window(window_handle_main)

6. 特殊元素

针对网页中的一些特殊网页元素，比如：JS 警告对话框、下拉选项，Selenium 都提供了对应的 API，方便我们快速地进行处理。

警告对话框会在窗口上浮动弹出一个警告对话框，WebDriver 使用 switch_to.alert() 切换到对话框本身，然后就能操作对话框了。

常见操作包含：获取对话框文本内容、同意对话框、关闭对话框，如果对话框内存在输入框的话，还可以使用 send_keys() 输入内容。

# 切换到对话框
dialog_alert = driver.switch_to.alert

# 获取警告框的提示信息
acontent = dialog_alert.text

# 对话框的操作
# 操作包含 accept()、dismiss()、send_keys()
dialog_alert.accept()

下拉选项 Select 也是比较常用的网页元素，要模拟选择某一项值，需要借助 WebDriver 提供的 Select 类。

Select 类可以通过索引、文本内容、value 属性值来模拟选择下拉选项的某一项。

from selenium.webdriver.support.select import Select

# 下拉选项元素
element_select = driver.find_element_by_xpath("//select[@id='select_id']")

# 利用 Select 类进行选择
# 1、通过value来选择
Select(element_select).select_by_value(element_value)

# 2、通过索引来选择
# Select(element_select).select_by_index(element_index)

# 3、通过内容来选择
Select(element_select).select_by_visible_text(element_text)