Gausst松鼠会

这个屌丝很懒，什么也没留下！

热门标签

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

作者：Gausst松鼠会 | 2024-03-15 15:47:15

踩

前面我们学习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()关闭浏览器。

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

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


现在我也找了很多测试的朋友，做了一个分享技术的交流群，共享了很多我们收集的技术文档和视频教程。
如果你不想再体验自学时找不到资源，没人解答问题，坚持几天便放弃的感受
可以加入我们一起交流。而且还有很多在自动化，性能，安全，测试开发等等方面有一定建树的技术大牛
分享他们的经验，还会分享很多直播讲座和技术沙龙
可以免费学习！划重点！开源的！！！
qq群号：485187702【暗号：csdn11】

最后感谢每一个认真阅读我文章的人，看着粉丝一路的上涨和关注，礼尚往来总是要有的，虽然不是什么很值钱的东西，如果你用得到的话可以直接拿走！ 希望能帮助到你！【100%无套路免费领取】

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/242600