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()