webdriver高级应用- 高亮显示正在操作的页面元素

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

def highLightElement(driver,element):
    # 封装好的高亮显示页面元素的方法
    # 使用JavaScript代码将传入的页面元素对象的背景颜色和边框颜色分别设置为
    # 绿色和红色
    driver.execute_script("arguments[0].setAttribute('style',\
    arguments[1]);", element,"background:green; border:2px solid red;")

class TestDemo(unittest.TestCase):
    def setUp(self):
        # 获取浏览器驱动实例
        self.driver = webdriver.Ie(executable_path = "e:\\IEDriverServer")

    def test_HighLightWebElement(self):
        url = "http://sogou.com"
        # 访问搜狗首页
        self.driver.get(url)
        searchBox = self.driver.find_element_by_id("query")
        # 调用高亮显示元素的封装函数，将搜索输入框进行高亮显示
        highLightElement(self.driver, searchBox)
        # 等待3秒，以便查看高亮效果
        time.sleep(3)
        searchBox.send_keys(u"光荣之路自动化测试")
        submitButton = self.driver.find_element_by_id("stb")
        # 调用高亮显示元素的封装函数，将搜索按钮进行高亮显示
        highLightElement(self.driver, submitButton)
        time.sleep(3)
        submitButton.click()
        time.sleep(3)

    def tearDown(self):
        # 退出浏览器
        self.driver.quit()

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