selenium常用的API（三）獲取網頁title、html源碼

獲取網頁title

獲取頁面title的方法可以直接用driver.title獲取到，然后可以把獲取到的結果用做斷言。

#encoding=utf-8
from selenium import webdriver
import unittest
import time

class VisitUrl(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Ie(executable_path = "e:\\IEDriverServer")

    def test_getTitle(self):
        self.driver.get("http://www.baidu.com")
        # 調用driver的title屬性獲取頁面的title屬性值
        title = self.driver.title
        print u"當前網頁的title屬性值為：", title
        # 斷言頁面的title屬性值是否是“百度一下，你就知道”,如果斷言失敗則打印"頁面title屬性值錯誤！"
        self.assertEqual(title, u"百度一下，你就知道", "頁面title屬性值錯誤！")

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

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

獲取HTML源碼

driver.get("https://www.taobao.com")
# 調用driver的page_source屬性獲取頁面源碼
pageSource = driver.page_source
# 打印頁面源碼
print pageSource.encode("gbk", "ignore")
# 斷言頁面源碼中是否包含“購物”兩個關鍵字，以此判斷頁面內容是否正確
self.assertTrue(u"購物" in pageSource)