前面我们学习unittest 的目的是用它编写 Web 自动化测试用例,所以接下来会将 unittest 与Selenium 结合起来写一个 Web 自动化测试的例子。
我们仍以百度搜索为例,创建 test_baidu.py 文件
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 | # _*_ 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模块化,所以这里把操作步骤封装成一个方法。
代码如下:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 | # _*_ 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可以解决这个问题。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 | # _*_ 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 |
修改前:
![](https://img-blog.csdnimg.cn/img_convert/ca1d02b2b87b70471da2cf412dd47dc8.png)
修改后:
![](https://img-blog.csdnimg.cn/img_convert/7701e22210663e1d5860053b440d94eb.png)
可以看到修改后省去16秒之多,这里虽然我们将 driver 驱动定义为 cls.driver,但是在每个测试用例中使用时依然为self.driver。
当整个测试类中的所有测试用例都运行完成后,会调用 cls.driver.quit()关闭浏览器。
当一个测试类中有多条测试用例时,这种方式将会大大缩短测试用例的执行时间。
这样我们就得到了一个简单的百度搜索的自动化测试脚本,读者可以根据此例试试别的场景。
最后感谢每一个认真阅读我文章的人,看着粉丝一路的上涨和关注,礼尚往来总是要有的,虽然不是什么很值钱的东西,如果你用得到的话可以直接拿走! 希望能帮助到你!【100%无套路免费领取】
![](https://img-blog.csdnimg.cn/direct/db7e3aa9e1bc4369bb1eeb59370a323b.png)
![](https://img-blog.csdnimg.cn/0edaf198db0444d19309d1de13f0b209.png)