selenium+unittest实现web自动化的示例代码

前面我们学习unittest 的目的是用它编写 Web 自动化测试用例,所以接下来会将 unittest 与Selenium 结合起来写一个 Web 自动化测试的例子。

我们仍以百度搜索为例,创建 test_baidu.py 文件

# _*_ coding:utf-8 _*_
"""
name:zhangxingzai
date:2023/2/25
"""

import unittest
from time import sleep
from selenium import webdriver
from selenium.webdriver.common.by import By


class TestBaidu(unittest.TestCase):

  def setUp(self):
      self.driver = webdriver.Firefox()
      self.baidu_url = 'https://www.baidu.com'

  def test_search_key_unittest(self):
      self.driver.get(self.baidu_url)
      self.driver.find_element(By.ID, 'kw').send_keys('unittest')
      self.driver.find_element(By.ID, 'su').click()
      sleep(2)
      title = self.driver.title
      self.assertEqual(title, 'unittest_百度搜索')

  def test_search_key_selenium(self):
      self.driver.get(self.baidu_url)
      self.driver.find_element(By.ID, 'kw').send_keys('selenium')
      self.driver.find_element(By.ID, 'su').click()
      sleep(2)
      title = self.driver.title
      self.assertEqual(title, 'selenium_百度搜索')

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


if __name__ == '__main__':
  unittest.main

对上面的代码不做过多介绍,都是以 unittest 创建测试类和方法的。方法中的代码是Selenium 脚本。不过,这里的代码存在一些问题,我们来一一改进。

首先,观察代码可以发现,两个测试用例中的步骤是一样的,唯一的区别是搜索的关键字和断言的结果不同。在学习过python模块化,所以这里把操作步骤封装成一个方法。

代码如下:

# _*_ coding:utf-8 _*_
"""
name:zhangxingzai
date:2023/2/25
"""

import unittest
from time import sleep
from selenium import webdriver
from selenium.webdriver.common.by import By


class TestBaidu(unittest.TestCase):

  def setUp(self):
      self.driver = webdriver.Firefox()
      self.baidu_url = 'https://www.baidu.com'
  
  # 将百度首页的访问和搜索过程封装成一个baidu_search()方法
  def baidu_search(self, search_key):
      self.driver.get(self.baidu_url)
      self.driver.find_element(By.ID, 'kw').send_keys(search_key)
      self.driver.find_element(By.ID, 'su').click()
      sleep(2)

  def test_search_key_unittest(self):
      search_key = 'unittest'
      self.baidu_search(search_key)
      self.assertEqual(self.driver.title, search_key + '百度搜索')

  def test_search_key_selenium(self):
      search_key = "selenium"
      self.baidu_search(search_key)
      self.assertEqual(self.driver.title, search_key + "百度搜索")
  
  def tearDown(self):
      self.driver.close()

if __name__ == '__main__':
  unittest.main

这里将百度首页的访问和搜索过程封装成一个 baidu_search()方法,并定义 search_key参数为搜索关键字,根据接收的关键字执行不同内容的搜索。这里的 baidu_search()方法不会被当作测试用例执行,因为根据 unittest 查找和执行测试用例的规则,它只会把以“test”开头的方法当作测试用例。

另一个值得讨论的问题是,测试用例的断言要不要写在封装的方法中?从前面的代码可以看出,测试的断言点是一样的。这里更倾向于把断言写在每一条测试用例里面。因为很多时候就算操作步骤是一样的,断言点也不完全一样。从设计的角度来看,断言写在每一个测试用例中也会更加清晰。

另外我们还发现每一条测试用例都要启动和关闭一次浏览器,这是非常耗时的,那么如何减少浏览器的启动和关闭次数呢?利用前面学过的setUpClass/tearDownClass可以解决这个问题。

# _*_ coding:utf-8 _*_
"""
name:zhangxingzai
date:2023/2/25
"""

import unittest
from time import sleep
from selenium import webdriver
from selenium.webdriver.common.by import By


class TestBaidu(unittest.TestCase):

  @classmethod
  def setUpClass(cls):
      cls.driver = webdriver.Firefox()
      cls.baidu_url = 'https://www.baidu.com'

  def baidu_search(self, search_key):
      self.driver.get(self.baidu_url)
      self.driver.find_element(By.ID, 'kw').send_keys(search_key)
      self.driver.find_element(By.ID, 'su').click()
      sleep(2)

  def test_search_key_unittest(self):
      search_key = 'unittest'
      self.baidu_search(search_key)
      self.assertEqual(self.driver.title, search_key + '_百度搜索')

  def test_search_key_selenium(self):
      search_key = 'selenium'
      self.baidu_search(search_key)
      self.assertEqual(self.driver.title, search_key + '_百度搜索')

  @classmethod
  def tearDownClass(cls):
      cls.driver.close()


if __name__ == '__main__':
  unittest.main

修改前:

修改后:

可以看到修改后省去16秒之多,这里虽然我们将 driver 驱动定义为 cls.driver,但是在每个测试用例中使用时依然为self.driver。

当整个测试类中的所有测试用例都运行完成后,会调用 cls.driver.quit()关闭浏览器。

当一个测试类中有多条测试用例时,这种方式将会大大缩短测试用例的执行时间。

这样我们就得到了一个简单的百度搜索的自动化测试脚本,读者可以根据此例试试别的场景。

关于selenium+unittest实现web自动化的示例代码的文章就介绍至此,更多相关selenium unittest web自动化内容请搜索编程宝库以前的文章,希望以后支持编程宝库

 前言格斗游戏,曾经是街机厅里最火爆的游戏之一,甚至可以把“之一”去掉,那个年代的格斗游戏就是街机游戏的王。如今的影响力已经大不如前,已经基本上消亡了。今天让小编重 ...