1.4　WebDriver脚本示例

WebDriver是Selenium的核心组成部分，也是本书重点讲解的内容之一。本节先带大家浏览一下测试脚本的样式。

from selenium import webdriver
import unittest

class VisitPTPress(unittest.TestCase):
  def setUp(self):
           self.driver = webdriver.Chrome()

      def test_open_ptpress(self):
           self.driver.get('https://www.ptpress.com.cn/') # 打开人民邮电出版社官网
           self.assertIn('图书', self.driver.page_source) # 断言：网页中有“图书”字样

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

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

上述脚本采用Python unittest框架编写，整个操作流程如下。

●导入WebDriver的包。

●打开Chrome浏览器。

●访问人民邮电出版社官网。

●判断打开的网页中是否有“图书”字样。

●退出浏览器。