selenium小白学习笔记（6） - 使用parameterized参数化

参数化

假设对百度搜索功能进行测试，需要搜索很多个关键字。若对每一次搜索都写一个测试方法，那么需要很多行代码才能完成，而这些测试方法中只是搜索的关键字不同，其他操作步骤都是相同的。这样的测试代码看上去不够整洁，更加不利于维护。
使用parameterized可以解决这个问题，由于是第三方模块，所以需要自己手动安装一下，安装语句为pip install parameterized
（p.s.还有其他的参数化方法，如ddt，如果需要可以自己研究）

脚本例子

上脚本：

import unittest
import time
from selenium import webdriver
from parameterized import parameterized

class testBaiduSearch(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls.driver=webdriver.Chrome()
        cls.baseURL=r"https://www.baidu.com/"

    @parameterized.expand([["case1","one"],["case2","two"],["case3","three"]])
    def test_search(self,case,keys):
        self.driver.get(self.baseURL)
        time.sleep(3)
        self.driver.find_element_by_id("kw").send_keys(keys)
        self.driver.find_element_by_id("su").click()
        time.sleep(3)
        self.assertEqual(self.driver.title,keys+"_百度搜索")
        
    @classmethod
    def tearDownClass(cls):
        cls.driver.quit()

if __name__=="__main__":
    unittest.main(verbosity=2)

脚本说明

1、首先需要导入parameterized模块
2、在百度搜索的测试方法test_search前，加上装饰器@parameterized.expand()，expand中的参数是测试用例名字和百度搜索关键字。例子中是三个测试用例，用例case1的关键字是one，用例case2的关键字为two，用例case3的关键字为three。
3、在百度搜索的测试方法test_search的参数中加上测试用例名字和百度搜索关键字。
4、为了测试结果显示的更加详细，在unittest.main()函数中，加上了参数verbosity=2
5、执行结果

使用csv保存测试参数

若要使用的参数很多，可使用csv来维护参数列表，在执行测试时从csv文件中读取内容再进行操作即可。
1、本地新建一个csv文件，如图：

2、脚本修改一下，增加从csv文件中读取的部分：

import unittest
import time
import csv
from selenium import webdriver
from parameterized import parameterized

with open(r"I:\python\testBaidu.csv","r") as rf:
    myreader=csv.reader(rf)
    keylist=list(myreader)

class testBaiduSearch(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls.driver=webdriver.Chrome()
        cls.baseURL=r"https://www.baidu.com/"
    
    @parameterized.expand(keylist)
    def test_search(self,case,keys):
        self.driver.get(self.baseURL)
        time.sleep(3)
        self.driver.find_element_by_id("kw").send_keys(keys)
        self.driver.find_element_by_id("su").click()
        time.sleep(3)
        self.assertEqual(self.driver.title,keys+"_百度搜索")
        
    @classmethod
    def tearDownClass(cls):
        cls.driver.quit()

if __name__=="__main__":
    unittest.main(verbosity=2)

脚本说明：
1、先导入csv模块
2、以下语句为从本地csv文件中读取内容到keylist中
with open(r"I:\python\testBaidu.csv",“r”) as rf:
myreader=csv.reader(rf)
keylist=list(myreader)
3、将装饰器@parameterized.expand()的参数设置为keylist

综上来看，当测试的数据很多很多时，使用csv文件来维护会更加方便。