selenium--高亮显示正在操作的元素

from selenium import webdriver
import unittest, time
def highLightElement(driver, element):
    '''
    封装好的高亮显示页面元素的方法
    使用JS代码将传入的页面元素对象的背景颜色和边框
   颜色分别设置为绿色和红色
    '''
    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.Chrome()
    def test_HighLightWebElement(self):
        url = 'http://www.sogou.com'
        self.driver.get(url)
        searchBox = self.driver.find_element_by_id('query')
        # 调用高亮显示元素的封装函数，将搜索框高亮显示
        highLightElement(self.driver, searchBox)
        time.sleep(3)
        searchBox.send_keys('测试开发')
        sumitbutton = self.driver.find_element_by_id('stb')
        # 调用高亮显示元素的封装函数，将搜索按钮高亮显示
        highLightElement(self.driver, sumitbutton)
        time.sleep(3)
        sumitbutton.click()
        time.sleep(3)
    def tearDown(self):
        self.driver.quit()
if __name__ == '__main__':
    unittest.main()