python爬蟲之selenium--獲取HTML源碼斷言和URL地址

python爬蟲之selenium--獲取HTML源碼斷言和URL地址前言

在做測試的時候經常會使用selenium獲取網頁url與原始碼，判斷當前執行是否正確，接下來我們一起看下如果通過selenium獲取url及網頁原始碼

獲取HTML源碼

from selenium import webdriverimport unittest

class Test_source(unittest.TestCase):    def Test_getPageSource(self):        url = 'http://www.sogou.com'        self.driver = webdriver.Chrome()        self.driver.get(url)
        # 調取driver的page_source屬性獲取頁面源碼        pageSource = self.driver.page_source
        print('當前網頁的源碼：', pageSource)
        # 斷言頁面源碼中是否包含"新聞"兩個關鍵字，以此判斷頁面內容是否正確        self.assertTrue(u"新聞" in pageSource, "頁面源碼中未找到新聞關鍵字")

test1 = Test_source()test1.Test_getPageSource()
獲取URL地址
from selenium import webdriverimport unittest

class Test_Pageurl(unittest.TestCase):    def test_getCurrentPageUrl(self):        url = 'http://www.sogou.com'        self.driver = webdriver.Chrome()        self.driver.get(url)
        # 獲取當前頁面的url        currentPageUrl = self.driver.current_url        print('當前網頁的url：', currentPageUrl)
        # 斷言當前網頁的網址是否為 https://www.sogou.com/        self.assertEqual(currentPageUrl, 'https://www.sogou.com/', "當前網頁網址非預期")

test1 = Test_Pageurl()test1.test_getCurrentPageUrl()
斷言源碼中的關鍵字
from selenium import webdriverimport unittestimport time

class Test_KeyWord(unittest.TestCase):    def test_assertKeyWord(self):        url = 'http://www.baidu.com'        self.driver = webdriver.Chrome()        self.driver.get(url)        self.driver.find_element_by_id('kw').send_keys('自動化測試')        self.driver.find_element_by_id('su').click()        time.sleep(4)        assert '自動化' in self.driver.page_source, '頁面中不存在該關鍵字'

test1 = Test_KeyWord()test1.test_assertKeyWord()