selenium实战2 登陆博客园

#-*- coding:utf-8 -*-
__author__ = "carry"


from selenium import  webdriver
import time
import unittest

class Blog(unittest.TestCase):
    def setUp(self):
        self.driver=webdriver.Firefox()
        url="http://www.cnblogs.com "
        self.driver.get(url)
    def login(self,username,pwd):
        self.driver.find_element_by_link_text("登录").click()
        time.sleep(2)
        self.driver.find_element_by_xpath(".//*[@id='input1']").send_keys(username)
        self.driver.find_element_by_xpath(".//*[@id='input2']").send_keys(pwd)
        self.driver.find_element_by_id("signin").click()
        self.driver.save_screenshot('xxx.jpg')
        time.sleep(2)
        print self.driver.title
    def is_login_sucess(self):
        try:
            text=self.driver.find_element_by_id("user_nav_blog_link").text
            print text
            return True
        except:
            return False
    def test_01(self):
        self.login("xxxx","xxxx")
        result=self.is_login_sucess()
        self.assertTrue(result)
    def tearDown(self):
        self.driver.quit()

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