Python Selenium 之数据驱动测试的实现

数据驱动模式的测试好处相比普通模式的测试就显而易见了吧！使用数据驱动的模式，可以根据业务分解测试数据，只需定义变量，使用外部或者自定义的数据使其参数化，从而避免了使用之前测试脚本中固定的数据。可以将测试脚本与测试数据分离，使得测试脚本在不同数据集合下高度复用。不仅可以增加复杂条件场景的测试覆盖，还可以极大减少测试脚本的编写与维护工作。

下面将使用Python下的数据驱动模式(ddt)库，结合unittest库以数据驱动模式创建百度搜索的测试。

ddt库包含一组类和方法用于实现数据驱动测试。可以将测试中的变量进行参数化。

可以通过python自带的pip命令进行下载并安装：pip install ddt .

一个简单的数据驱动测试

为了创建数据驱动测试，需要在测试类上使用@ddt装饰符，在测试方法上使用@data装饰符。@data装饰符把参数当作测试数据，参数可以是单个值、列表、元组、字典。对于列表，需要用@unpack装饰符把元组和列表解析成多个参数。

下面实现百度搜索测试，传入搜索关键词和期望结果，代码如下：

1. import unittest

2. from selenium import webdriver

3. from ddt import ddt, data, unpack

5. @ddt

6. class SearchDDT(unittest.TestCase):

7. '''docstring for SearchDDT'''

8. def setUp(self):

9. self.driver = webdriver.Chrome()

10. self.driver.implicitly_wait(30)

11. self.driver.maximize_window()

12. self.driver.get("https://www.baidu.com")

14. # specify test data using @data decorator

15. @data(('python', 'PyPI'))

16. @unpack

17. def test_search(self, search_value, expected_result):

18. search_text = self.driver.find_element_by_id('kw')

19. search_text.clear()

20. search_text.send_keys(search_value)

22. search_button = self.driver.find_element_by_id('su')

23. search_button.click()

25. tag = self.driver.find_element_by_link_text("PyPI").text

26. self.assertEqual(expected_result, tag)

28. def tearDown(self):

29. self.driver.quit()

31. if __name__ == '__main__':

32. unittest.main(verbosity=2)

在test_search()方法中，search_value与expected_result两个参数用来接收元组解析的数据。当运行脚本时，ddt把测试数据转换为有效的python标识符，生成名称为更有意义的测试方法。结果如下：

使用外部数据的数据驱动测试

如果外部已经存在了需要的测试数据，如一个文本文件、电子表格或者数据库，那也可以用ddt来直接获取数据并传入测试方法进行测试。

下面将借助外部的CSV(逗号分隔值)文件和EXCLE表格数据来实现ddt。

通过CSV获取数据

同上在@data装饰符使用解析外部的CSV（testdata.csv）来作为测试数据（代替之前的测试数据）。其中数据如下：

接下来，先要创建一个get_data()方法，其中包括路径（这里默认使用当前路径）、CSV文件名。调用CSV库去读取文件并返回一行数据。再使用@ddt及@data实现外部数据驱动测试百度搜索，代码如下：

1. import csv, unittest

2. from selenium import webdriver

3. from ddt import ddt, data, unpack

5. def get_data(file_name):

6. # create an empty list to store rows

7. rows = []

8. # open the CSV file

9. data_file = open(file_name, "r")

10. # create a CSV Reader from CSV file

11. reader = csv.reader(data_file)

12. # skip the headers

13. next(reader, None)

14. # add rows from reader to list

15. for row in reader:

16. rows.append(row)

17. return rows

19. @ddt

20. class SearchCSVDDT(unittest.TestCase):

21. def setUp(self):

22. self.driver = webdriver.Chrome()

23. self.driver.implicitly_wait(30)

24. self.driver.maximize_window()

25. self.driver.get("https://www.baidu.com")

27. # get test data from specified csv file by using the get_data funcion

28. @data(*get_data('testdata.csv'))

29. @unpack

30. def test_search(self, search_value, expected_result):

31. search_text = self.driver.find_element_by_id('kw')

32. search_text.clear()

33. search_text.send_keys(search_value)

35. search_button = self.driver.find_element_by_id('su')

36. search_button.click()

38. tag = self.driver.find_element_by_link_text("PyPI").text

39. self.assertEqual(expected_result, tag)

41. def tearDown(self):

42. self.driver.quit()

44. if __name__ == '__main__':

45. unittest.main(verbosity=2)

测试执行时，@data将调用get_data()方法读取外部数据文件，并将数据逐行返回给@data。执行的结果也同上~
如果对软件测试、接口测试、自动化测试、面试经验交流。感兴趣可以加软件测试交流：1085991341，还会有同行一起技术交流。

通过Excel获取数据

测试中经常用Excle存放测试数据，同上在也可以使用@data装饰符来解析外部的CSV（testdata.csv）来作为测试数据（代替之前的测试数据）。其中数据如下：

接下来，先要创建一个get_data()方法，其中包括路径（这里默认使用当前路径）、EXCEL文件名。调用xlrd库去读取文件并返回数据。再使用@ddt及@data实现外部数据驱动测试百度搜索，代码如下：

1. import xlrd, unittest

2. from selenium import webdriver

3. from ddt import ddt, data, unpack

5. def get_data(file_name):

6. # create an empty list to store rows

7. rows = []

8. # open the CSV file

9. book = xlrd.open_workbook(file_name)

10. # get the frist sheet

11. sheet = book.sheet_by_index(0)

12. # iterate through the sheet and get data from rows in list

13. for row_idx in range(1, sheet.nrows): #iterate 1 to maxrows

14. rows.append(list(sheet.row_values(row_idx, 0, sheet.ncols)))

15. return rows

17. @ddt

18. class SearchEXCLEDDT(unittest.TestCase):

19. def setUp(self):

20. self.driver = webdriver.Chrome()

21. self.driver.implicitly_wait(30)

22. self.driver.maximize_window()

23. self.driver.get("https://www.baidu.com")

25. # get test data from specified excle spreadsheet by using the get_data funcion

26. @data(*get_data('TestData.xlsx'))

27. @unpack

28. def test_search(self, search_value, expected_result):

29. search_text = self.driver.find_element_by_id('kw')

30. search_text.clear()

31. search_text.send_keys(search_value)

33. search_button = self.driver.find_element_by_id('su')

34. search_button.click()

36. tag = self.driver.find_element_by_link_text("PyPI").text

37. self.assertEqual(expected_result, tag)

39. def tearDown(self):

40. self.driver.quit()

42. if __name__ == '__main__':

43. unittest.main(verbosity=2)

与上面读取CVS文件一样，测试执行时，@data将调用get_data()方法读取外部数据文件，并将数据逐行返回给@data。执行的结果也同上~

如果想从数据库的库表中获取数据，同样也需要一个get_data()方法，并且通过DB相关的库来连接数据库、SQL查询来获取测试数据。

以上就是本文的全部内容，希望对大家的学习有所帮助。有被帮助到的朋友欢迎点赞，评论。

 

总结：

感谢每一个认真阅读我文章的人！！！

作为一位过来人也是希望大家少走一些弯路，如果你不想再体验一次学习时找不到资料，没人解答问题，坚持几天便放弃的感受的话，在这里我给大家分享一些自动化测试的学习资源，希望能给你前进的路上带来帮助。

软件测试面试文档

我们学习必然是为了找到高薪的工作，下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料，并且有字节大佬给出了权威的解答，刷完这一套面试资料相信大家都能找到满意的工作。

 

          视频文档获取方式：
这份文档和视频资料，对于想从事【软件测试】的朋友来说应该是最全面最完整的备战仓库，这个仓库也陪伴我走过了最艰难的路程，希望也能帮助到你！以上均可以分享，点下方小卡片即可自行领取。