selenium--获取HTML源码断言和URL地址

from selenium import webdriver
import 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()

from selenium import webdriver
import 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 webdriver
import unittest
import 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()