脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → selenium unittest web自动化

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

 更新时间：2023年03月02日 08:33:05   作者：小胖虎*  
本文主要介绍了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自动化内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

相关文章

  • Python函数参数的4种方式

    Python函数参数的4种方式

    本文主要介绍了Python函数参数的4种方式,主要包括必选参数,默认参数,可选参数,关键字参数,具有一定的参考价值,感兴趣的可以了解一下
    2024-01-01
  • Python 机器学习之线性回归详解分析

    Python 机器学习之线性回归详解分析

    回归是监督学习的一个重要问题，回归用于预测输入变量和输出变量之间的关系，特别是当输入变量的值发生变化时，输出变量的值也随之发生变化。回归模型正是表示从输入变量到输出变量之间映射的函数
    2021-11-11
  • Python中字符串的格式化方法小结

    Python中字符串的格式化方法小结

    这篇文章主要介绍了Python中字符串的格式化方法小结,提到了针对Python2.x与3.x版本相异情况下的不同技巧,需要的朋友可以参考下
    2016-05-05
  • Django日志及中间件模块应用案例

    Django日志及中间件模块应用案例

    这篇文章主要介绍了Django日志及中间件模块应用案例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-09-09
  • Python第三方库qrcode或MyQr生成博客地址二维码

    Python第三方库qrcode或MyQr生成博客地址二维码

    使用第三方库qrcode或者MyQr给自己的博客网址快速生成二维码，一键分享，文中含有详细示例代码，有需要的朋友可以借鉴参考下，希望能够有所帮助
    2021-10-10
  • Python实现SSH隧道功能的示例代码

    Python实现SSH隧道功能的示例代码

    SSH隧道是利用SSH协议建立一个加密通道,以保护通过不安全网络传输的数据,本文将介绍如何使用Python来实现SSH隧道功能,感兴趣的可以了解下
    2025-02-02
  • Matplotlib学习笔记之plt.xticks()用法

    Matplotlib学习笔记之plt.xticks()用法

    在matplotlib中ticks表示的是刻度,而刻度有两层意思,一个是刻标(locs),一个是刻度标签(tick labels),下面这篇文章主要给大家介绍了关于Matplotlib学习笔记之plt.xticks()用法的相关资料,需要的朋友可以参考下
    2022-09-09
  • Python实现将MySQL数据库查询结果导出到Excel

    Python实现将MySQL数据库查询结果导出到Excel

    在实际工作中,我们经常需要将数据库中的数据导出到Excel表格中进行进一步的分析和处理,Python中的pymysql和xlsxwriter库提供了很好的解决方案,下面我们就来看看具体操作方法吧
    2023-11-11
  • Python+OpenCV实现黑白老照片上色功能

    Python+OpenCV实现黑白老照片上色功能

    我们都知道，有很多经典的老照片，受限于那个时代的技术，只能以黑白的形式传世。尽管黑白照片别有一番风味，但是彩色照片有时候能给人更强的代入感。本文就来用Python和OpenCV实现老照片上色功能，需要的可以参考一下
    2023-02-02
  • 使用Ray集群简单创建Python分布式应用程序

    使用Ray集群简单创建Python分布式应用程序

    面对计算密集型的任务，除了多进程，就是分布式计算，如何用 Python 实现分布式计算呢？今天分享一个很简单的方法，那就是借助于 Ray
    2021-09-09

最新评论