使用javaScript操作页面元素

from selenium import webdriver
import time
import unittest
from selenium.common.exceptions import WebDriverException
import traceback

class javaSciptWebdriver(unittest.TestCase):
    # 使用javaScript操作页面元素
    def setUp(self):
        self.driver = webdriver.Chrome()
        self.url = "http://www.baidu.com"

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

    def test_executesScript(self):
        self.driver.get(self.url)
        # 构造javaScript查找百度首页的搜索输入框的代码字符串
        searchInputBoxJs = "document.getElementById('kw').value='光荣之路';"
        # 构造javaScript查找百度首页的搜索按钮的代码字符串
        searchButtonJs = "document.getElementById('su').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:
            print(u'在页面中没有找到要操作的页面元素',traceback.print_exc())
        except AssertionError:
            print(u'页面上不存在断言的关键字符串')
        except Exception:
            print(traceback.print_exc())

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