Selenium-基本使用

Selenium是一个用于Web应用程序测试的工具。Selenium测试直接运行在浏览器中，就像真正的用户在操作一样。支持的浏览器包括IE（7, 8, 9, 10, 11），Mozilla Firefox，Safari，Google Chrome，Opera等。这个工具的主要功能包括：测试与浏览器的兼容性——测试你的应用程序看是否能够很好得工作在不同浏览器和操作系统之上。测试系统功能——创建回归测试检验软件功能和用户需求。支持自动录制动作和自动生成 .Net、Java、Perl等不同语言的测试脚本。

安装

pip install selenium

安装驱动程序

下面介绍几种主流浏览器的驱动程序下载地址：

浏览器	下载地址
Chrome	https://sites.google.com/a/chromium.org/chromedriver/downloads
Edge	https://developer.microsoft.com/en-us/microsoft-edge/tools/webdriver/
Firefox	https://github.com/mozilla/geckodriver/releases
Safari	https://webkit.org/blog/6900/webdriver-support-in-safari-10/

下面以windows环境，Chrome浏览器为例，我下载的是2.36版本的驱动chromedriver_win32.zip，解压得到一个chromedriver.exe文件，将其放置在浏览器的安装目录之下，如图所示：

selenium的简单用法

如果你已经按照上面安装了selenium，并且安装了驱动程序，下面来简单的玩一玩selenium：

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

# 创建Chrome WebDriver实例，此路径为驱动程序的路径
driver = Chrome("C:\Program Files (x86)\Google\Chrome\Application\chromedriver.exe")
# 加载URL网页
driver.get("http://www.baidu.com/")
# 判断标题中是否包含该字符串
assert "百度一下" in driver.title
# 查找id=kw的节点
element = driver.find_element_by_id("kw")
# 清空原有的输入，输入python进行搜索
element.clear()
element.send_keys("python")
element.send_keys(Keys.RETURN)
# 判断是否有结果返回
assert "No results found." not in driver.page_source
# 关闭当前标签，也可以使用quit()关闭浏览器
driver.close()

上面的栗子，简单介绍了一下selenium的用法，接下来将结合selenium与unittest来编写测试：

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

# 继承至TestCase，表示这是一个测试用例类
class BaiduCase(unittest.TestCase):

    # 初始化的一部分
    def setUp(self):
        self.driver = Chrome("C:\Program Files (x86)\Google\Chrome\Application\chromedriver.exe")

    # 测试用例方法，名称可以自定义，方法名称始终以test开头
    def test_serch(self):
        self.driver.get("https://www.baidu.com/")
        assert "百度一下" in self.driver.title
        element = self.driver.find_element_by_id("kw")
        element.clear()
        element.send_keys("python")
        element.send_keys(Keys.RETURN)
        assert "No results found." not in self.driver.page_source

    # 在执行完各种测试用例方法之后会执行，为一个清理操作
    def tearDown(self):
        self.driver.close()


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

执行上面的代码，将会返回如下信息：

.
----------------------------------------------------------------------
Ran 1 test in 2.914s

OK

值得注意的是，如果没有配置驱动，或者路径写错了，将会出现WebDriverException错误

与网页互动

与一个网页进行的真正的交互，具体的说，与网页的HTML元素进行交互。如果需要与之交互，那么久必须要查找到HTML的元素。WebDriver提供了多种查找HTML元素的方法。例如，给定一个元素为：

<input type="text" name="passwd" id="passwd-id"/>

我们可以使用下面任意方法查找到它：

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

值的注意的是，使用xpath进行匹配时，如果有多个元素与查询匹配，只会返回第一个元素。如果什么都找不到，将会返回NoSuchElementException错误。当你查找到一个元素时，你可以用它做什么？首先你可能会想到在文本标签中输入一些文本：

element.send_keys("python")

也可以使用Keys来模拟一些箭头键：

from selenium.webdriver.common.keys import Keys
element.send_keys(Keys.ARROW_DOWN)

你也可以使用clear()方法来清除文本标签里的内容：

element.clear()

切换下拉菜单的状态，我们可以依次循环选择每个选项并依次选择每个选项：

element = driver.find_element_by_xpath("//select[@name='name']")
all_options = element.find_elements_by_tag_name("option")
for option in all_options:
    option.click()

正如你所看到的，这并不是处理select元素的最有效的方式。WebDriver类包还包括一个名为Select的类，它提供了与这些类交互的有用的方法：

from selenium.webdriver.support.ui import Select

element = driver.find_element_by_xpath("//select[@name='name']")
select = Select(element)

select.select_by_index(index)

select.select_by_visible_text("text")

select.select_by_value("value")

WebDriver还提供取消选择所有选项的功能：

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

Select还具有获取所有已经选中的标签的方法：

element = driver.find_element_by_id('id')
select = Select(element)
all_options = select.all_selected_options

获取所有可用的选项:

options = select.options

完成后表格填写后，您可以需要提交，一种方法是找到submit按钮并点击它：

driver.find_element_by_id('submit').click()

WebDriver在每个元素上都有一个submit的便利方法，如果你在一个表单的元素上调用它，Webdriver会遍历DOM直到包含表单，然后调用它：

element.submit()

WebDriver还提供了前进与后退的操作：

driver.forward()
driver.back()

WebDriver还可以进行设置Cookies和获取Cookies：

driver.get('https://www.baidu.com/')

driver.add_cookie({"name":"foo","value":"laozhang"})

print(driver.get_cookies()