import csv<br />import unittest<br />from time import sleep<br /><br />from ddt import ddt, data, unpack<br />from selenium import webdriver<br /><br /><br />def getCsvData():<br /> # 读取CSV文件<br /> value_rows = []<br /> with open('./TextData.csv', encoding='UTF-8') as f:<br /> f_csv = csv.reader(f)<br /> next(f_csv)<br /> for r in f_csv:<br /> value_rows.append(r)<br /> return value_rows<br /><br />@ddt<br /># 数据驱动<br />class MyTestCase(unittest.TestCase):<br /><br /> def setUp(self):<br /> self.driver = webdriver.Firefox()<br /> self.base_url = "https://www.baidu.com/"<br /> self.driver.get(self.base_url)<br /> self.driver.maximize_window()<br /> sleep(2)<br /><br /> @data(*getCsvData())<br /> # 测试数据<br /><br /> @unpack<br /> # 解包<br /> def test_something(self, searchTerm, searchResult):<br /> driver = self.driver<br /> driver.find_element_by_xpath(".//*[@id='kw']").send_keys(searchTerm)<br /> driver.find_element_by_xpath(".//*[@id='su']").click()<br /> sleep(2)<br /> responseText = driver.find_element_by_xpath(".//*[@id='1']/h3/a").text<br /> self.assertEqual(responseText, searchResult)<br /><br /> def tearDown(self):<br /> self.driver.close()<br /> self.driver.quit()<br /><br />if __name__ == '__main__':<br /> unittest.main()<br /><br />搜索词,响应<br />中国,中国_百度百科<br />美国,美国_百度百科<br />英国,英国_百度百科
