百度登录、退出示例

# -*- coding:utf-8 -*-
"""
百度登录，退出
"""
from selenium import webdriver
import unittest
from selenium.webdriver import ActionChains
from selenium.webdriver.support.ui import WebDriverWait

class Baidu_Demo(unittest.TestCase):
    
    def setUp(self):
        self.driver = webdriver.Firefox()
        self.driver.get('http://www.baidu.com')
    
    def test_login(self):
        driver = self.driver
        driver.find_element_by_xpath("//*[@id='lb']").click()
        driver.find_element_by_name('userName').send_keys('你的帐号')
        driver.find_element_by_name('password').send_keys('你的密码')
        driver.find_element_by_name('memberPass').click()                   #去掉记住密码
        driver.find_element_by_id('TANGRAM__PSP_10__submit').click()
        
        try:
            WebDriverWait(driver, 5).until(lambda x: x.find_element_by_id('s_username_top'))        
            myele = driver.find_element_by_id('s_username_top')   
            myele_text = myele.text
            
            action_myele = ActionChains(driver)
            action_myele.move_to_element(myele).perform()               #移动鼠标到元素我的帐号上
            
            WebDriverWait(driver, 5).until(lambda x: x.find_element_by_link_text('退出'))
            driver.find_element_by_link_text('退出').click()            #退出登录     
        except Exception as e:
            print e
        
        self.assertEqual("你的帐号", myele_text, '登录失败')
    
    def tearDown(self):
        self.driver.quit()
        
if __name__ == '__main__':
    unittest.main()

# -*- coding:utf-8 -*-
"""
百度登录，退出
"""
from selenium import webdriver
import unittest
from selenium.webdriver import ActionChains
from selenium.webdriver.support.ui import WebDriverWait
driver = webdriver.Firefox()
class Baidu_Demo(unittest.TestCase):
    
    def test_login(self):      
        driver.get('http://www.baidu.com')  
        driver.find_element_by_xpath("//*[@id='lb']").click()
        driver.find_element_by_name('userName').send_keys('expect88')
        driver.find_element_by_name('password').send_keys('luoqing1227')
        driver.find_element_by_name('memberPass').click()                   #去掉记住密码
        driver.find_element_by_id('TANGRAM__PSP_10__submit').click()
        
        self.assertTrue('expect88' in driver.page_source) 
   
    def test_qiut(self):
        WebDriverWait(driver, 5).until(lambda x: x.find_element_by_id('s_username_top'))        
        myele = driver.find_element_by_id('s_username_top')
            
        action_myele = ActionChains(driver)
        action_myele.move_to_element(myele).perform()               #移动鼠标到元素我的帐号上
            
        WebDriverWait(driver, 5).until(lambda x: x.find_element_by_link_text('退出'))
        driver.find_element_by_link_text('退出').click()
        
if __name__ == '__main__':
    unittest.main()