WebDriver高阶API（1）

WebDriver高阶API（1）

1、使用JavaScript操作页面元素

#encoding=utf-8
import time
import unittest
import traceback
from selenium import webdriver
from selenium.common.exceptions import WebDriverException

class TestDemo(unittest.TestCase):
    def setUp(self):
        #启动浏览器
        self.driver = webdriver.Ie(executable_path = "D:\IEDriverServer")
    def test_executeScript(self):
        url = "http://www.sogou.com"
        #访问搜狗首页
        self.driver.get(url)
        #构造JavaScript查找搜狗首页的搜索输入框并输入“魔兽世界”的代码字符串
        searchInputBoxJs = "document.getElementById('query').value='魔兽世界'"
        #构造JavaScript查找搜狗首页的搜索按钮并点击的代码字符串
        searchButtonJs = "document.getElementById('stb').click()"
        try:
            #通过JavaScript代码在搜索首页输入框中输入“魔兽世界”
            self.driver.execute_script(searchInputBoxJs)
            time.sleep(2)
            #通过JavaScript代码点击搜狗首页上的搜索按钮
            self.driver.execute_script(searchButtonJs)
            time.sleep(2)
            #断言
            self.assertTrue(u"魔兽世界" in self.driver.page_source)
        except WebDriverException,e:
            #当定位失败时，会抛出WebDriverException异常
            print u"在页面中没有找到要操作的页面元素",traceback.print_exc()
        except AssertionError,e:
            print u"页面不存在断言的关键字"
        except Exception,e:
            #发生其他异常时，打印异常堆栈信息
            print traceback.print_exc()

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

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

2、操作滚动条 JS

#encoding=utf-8
import time
import unittest
import traceback
from selenium import webdriver

class TestDemo(unittest.TestCase):
    def setUp(self):
        #启动浏览器
        self.driver = webdriver.Ie(executable_path = "D:\IEdriverServer")
    def test_scroll(self):
        url = "http://www.sohu.com"
        #访问搜狗首页
        try:
            self.driver.get(url)
            #使用JavaScript的scrollTo函数和document.body.scrollHeight参数
            # 将页面滚动条滑动到页面的最下方
            self.driver.execute_script("window.scrollTo(0,document.body.scrollHeight);")
            #停顿3秒，用于人工验证滚动条是否滑动到指定位置
            #根据测试需要，可注释下面的停顿代码
            time.sleep(3)

            #使用JavaScript 的scrollIntoView函数将被遮挡的元素滚动到可见屏幕上
            #scrollIntoView(true)表示将元素滚动到屏幕中间
            #scrollIntoView(false)表示将元素滚动到屏幕底部
            self.driver.execute_script("document.getElementsByTagName('a')[500].scrollIntoView(true);")
            time.sleep(3)

            #使用javascript的scrollBy方法，使用0和400横纵坐参数
            #将页面纵向向下滚动400像素
            self.driver.execute_script("window.scrollBy(0,400)")
            time.sleep(3)
        except Exception,e:
            print traceback.print_exc()

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

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

3、操作Ajax产生的浮动框

方法一：通过模拟键盘下箭头进行选择悬浮框选项

#encoding=utf-8
import time
import unittest
from selenium import webdriver
from selenium.webdriver.common.keys import Keys

class TestDemo(unittest.TestCase):
    def setUp(self):
        #启动浏览器
        self.driver = webdriver.Ie(executable_path = "D:\IEDriverServer")
    def test_AjaxDivOptionByKeys(self):
        url = "http://www.sogou.com"
        #访问sogou首页
        self.driver.get(url)
        #找到搜狗首页中的搜索输入框页面元素
        searchBox = self.driver.find_element_by_id("query")
        #在输入框输入 “魔兽世界”
        searchBox.send_keys(u"魔兽世界")
        #等待2秒，以便悬浮框加载完成
        time.sleep(2)
        for i in range(3):
            #选择悬浮框中的第几个联想关键词选项就循环几次
            #模拟键盘单击下箭头
            searchBox.send_keys(Keys.DOWN)
            time.sleep(0.5)

        #当按下箭头到想要选择的选项后，瑞模拟键盘按回车键，选中该选项
        searchBox.send_keys(Keys.ENTER)
        time.sleep(2)

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

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

方法二：通过匹配模糊内容选择悬浮框中选项

#encoding=utf-8
import time
import unittest
import traceback
from selenium import webdriver
from selenium.common.exceptions import NoSuchElementException

class TestDemo(unittest.TestCase):
    def setUp(self):
        #启动浏览器
        self.driver = webdriver.Ie(executable_path = "D:\IEDriverServer")
    def test_AjaxDivOptionByWords(self):
        url = "http://www.sogou.com/"
        #访问搜狗首页
        self.driver.get(url)
        try:
            #找到搜狗首页中的搜索输入框页面元素
            searchBox = self.driver.find_element_by_id("query")
            #在搜索输入框中输入“魔兽世界”
            searchBox.send_keys(u"魔兽世界")
            #等待2秒，以便悬浮框完成林加载
            time.sleep(2)
            #查找内容包含“怀旧服”的悬浮框选项
            suggetion_option = self.driver.find_element_by_xpath("//ul/li[contains(.,'怀旧服')]")
            #单击找到的悬浮选项
            suggetion_option.click()
            time.sleep(2)
        except NoSuchElementException,e:
            #打印异常堆栈信息
            print traceback.print_exc()

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

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

方法三：固定选择某一项，使用索引

#encoding=utf-8
import time
import unittest
import traceback
from selenium import webdriver
from selenium.common.exceptions import NoSuchElementException

class TestDemo(unittest.TestCase):
    def setUp(self):
        #启动浏览器
        self.driver = webdriver.Ie(executable_path = "D:\IEDriverServer")
    def test_AjaxDivOptionByIndex(self):
        url = "http://www.sogou.com"
        #访问搜狗首页
        self.driver.get(url)
        try:
            #找到搜狗首页中的搜索输入框页面元素
            searchBox = self.driver.find_element_by_id("query")
            #在输入框中输入 “魔兽世界”
            searchBox.send_keys(u"魔兽世界")
            #等待2秒以便悬浮杠加载完成
            time.sleep(2)
            #查找悬浮框中的第三项，只需要更改li[3]中索引数字，就可以实现任意单击悬浮框中的选项。注意，索引从1开始
            suggetion_option = self.driver.find_element_by_xpath("//*[@id='vl']/div[1]/ul/li[3]")
            #单击找到的选项
            suggetion_option.click()
            time.sleep(3)
        except NoSuchElementException,e:
            #打印异常堆栈信息
            print traceback.print_exc()

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

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

4、结束Windows中浏览器进程

#encoding=utf-8
import unittest
from selenium import webdriver

class TestDemo(unittest.TestCase):
    def test_killWindowsProcess(self):
        chromeDriver = webdriver.Chrome(executable_path="D:\chromedriver")
        ieDriver = webdriver.Ie(executable_path="D:\IEDriverServer")
        #导入os包
        import os
        #结束ie浏览器进程
        returnCode = os.system("taskkill /F /iM iexplore.exe")
        if returnCode == 0:
            print u"成功结束Ie浏览器进程"
        else:
            print u"结束IE浏览器进程失败"

        #结束chrome浏览器
        returnCode = os.system("taskkill /F /iM chrome.exe")
        if returnCode == 0:
            print u"成功结束chrome浏览器进程"
        else:
            print u"结束chrome浏览器进程失败"

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