Python+Selenium_UI自动化操作(6)——获取当前页面的源码

UI自动化——获取页面源码

#练习：打开浏览器，获取页面源码
#语法：page_source

#encoding=utf-8
from selenium import webdriver
import time,unittest,sys,re
"""
**unittest框架执行顺序：**
1、先执行setUp，做测试数据准备；
2、执行测试用例方法testXXXX，测试用例都是以test开头的方法，可以有多个测试用例；
3、所有的测试用例执行完成后，执行tearDown，进行资源回收，防止资源泄露，比如窗口句柄。
"""
class  TestGetPageSource(unittest.TestCase):

    def setUp(self):
        # setUp是一个初始化方法，为test案例做数据准备
        # 当前方法的数据准备动作是：启动chrome浏览器
        self.b=webdriver.Chrome()

    def test_get_page_source(self):
        url="http://www.baidu.com"
        # 访问百度调用page_source属性获取页面源码
        self.b.get(url)
        pagesource=self.b.page_source
        print(type(pagesource))
        # print(pagesource)
        # 好像获取到的源码就是gbk格式，不需要下面转成gbk格式，因此注释掉了下面那句
        #print(pagesource.encode("gbk",ignore))
        # 获取某些关键字在源码中的数量
        print(pagesource.count("百度"))
        # 断言某些关键字是否在源码中
        self.assertTrue(u"百度" in pagesource)
          
   
    def tearDown(self):
        self.b.quit()

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