Selenium

获取HTML源码

1from selenium import webdriver
2import unittest
3
4
5class Test_source(unittest.TestCase):
6    def Test_getPageSource(self):
7        url = 'http://www.sogou.com'
8        self.driver = webdriver.Chrome()
9        self.driver.get(url)
10
11        # 调取driver的page_source属性获取页面源码
12        pageSource = self.driver.page_source
13
14        print('当前网页的源码：', pageSource)
15
16        # 断言页面源码中是否包含"新闻"两个关键字，以此判断页面内容是否正确
17        self.assertTrue(u"新闻" in pageSource, "页面源码中未找到新闻关键字")
18
19
20test1 = Test_source()
21test1.Test_getPageSource()

获取URL地址

1from selenium import webdriver
2import unittest
3
4
5class Test_Pageurl(unittest.TestCase):
6    def test_getCurrentPageUrl(self):
7        url = 'http://www.sogou.com'
8        self.driver = webdriver.Chrome()
9        self.driver.get(url)
10        
11        # 获取当前页面的url
12        currentPageUrl = self.driver.current_url
13        print('当前网页的url：', currentPageUrl)
14
15        # 断言当前网页的网址是否为 https://www.sogou.com/
16        self.assertEqual(currentPageUrl, 'https://www.sogou.com/', "当前网页网址非预期")
17
18
19test1 = Test_Pageurl()
20test1.test_getCurrentPageUrl()

 断言源码中的关键字

1from selenium import webdriver
2import unittest
3import time
4
5
6class Test_KeyWord(unittest.TestCase):
7    def test_assertKeyWord(self):
8        url = 'http://www.baidu.com'
9        self.driver = webdriver.Chrome()
10        self.driver.get(url)
11        self.driver.find_element_by_id('kw').send_keys('自动化测试')
12        self.driver.find_element_by_id('su').click()
13        time.sleep(4)
14        assert '自动化' in self.driver.page_source, '页面中不存在该关键字'
15
16
17test1 = Test_KeyWord()
18test1.test_assertKeyWord()