执行JS

# -*- coding:utf-8 -*-
"""
执行 js 脚本
"""

from selenium import webdriver
import unittest

class login(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Firefox()
        self.driver.implicitly_wait(30)
        self.base_url = 'http://www.51testing.com'
        self.verificationErrors = []
        
    def test_loginpass(self):
        driver = self.driver
        driver.get(self.base_url)
        
        js='var q=document.getElementById("username");q.style.border="1px solid red";'
        driver.execute_script(js)
        driver.find_element_by_id('username').send_keys('username')
        driver.find_element_by_id('userpass').send_keys('password')
        driver.find_element_by_id('dologin').click()
        verify = raw_input('输入你看到的验证码,按 Enter 键: ')
        driver.find_element_by_id('xspace-seccode').send_keys(verify)
        driver.find_element_by_id('securitysubmit').click()
        
    def tearDown(self):
        self.driver.quit()
        self.assertEqual([], self.verificationErrors)
        
if __name__ == '__main__':
    unittest.main()

 

# -*- coding:utf-8 -*-
"""
调用JS代码返回网页的title,并统计超链接的数量
"""

from selenium import webdriver

driver = webdriver.Firefox()
driver.get("http://www.baidu.com")
title = driver.execute_script("return document.title")
print title

links = driver.execute_script("var links = document.getElementsByTagName('A'); return links.length")
print links