数据驱动应用

import csv
import codecs
import unittest
from itertools import islice
from selenium import webdriver
import time
from HTMLTestRunner import HTMLTestRunner

class TestBaidu(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls.driver = webdriver.Chrome()
        cls.base_url = "https://www.baidu.com"
        cls.test_data = []
        with codecs.open(‘baidu_data.csv‘, ‘r‘, ‘utf_8_sig‘) as f:
            data = csv.reader(f)
            for line in islice(data,1,None):
                cls.test_data.append(line)

    @classmethod
    def tearDownClass(cls):
        cls.driver.quit()


    def baidu_search(self, search_key):
        self.driver.get(self.base_url)
        self.driver.find_element_by_id("kw").send_keys(search_key)
        self.driver.find_element_by_id("su").click()
        time.sleep(2)

    def test_search_selenium(self):
        self.baidu_search(self.test_data[0][1])

    def test_search_unittest(self):
        self.baidu_search(self.test_data[1][1])

    def test_search_parameterized(self):
        self.baidu_search(self.test_data[2][1])

if __name__=="__main__":
    suit = unittest.TestSuite()
    suit.addTest(TestBaidu("test_search_selenium"))
    suit.addTest(TestBaidu("test_search_unittest"))
    suit.addTest(TestBaidu("test_search_parameterized"))
    
    now_time = time.strftime("%Y-%m-%d  %H_%M_%S")
    with open(now_time + " Baidu_search.html", "wb") as fp:
        runner = HTMLTestRunner(stream=fp,title="百度搜索测试报告",                                description="运行环境：win10， Chrome浏览器")
        runner.run(suit)
        fp.close()

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

class TestBaidu(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls.driver = webdriver.Chrome()
        cls.base_url = "https://www.baidu.com"
      
   
    def baidu_search(self, search_key):
        self.driver.get(self.base_url)
        self.driver.find_element_by_id("kw").send_keys(search_key)
        self.driver.find_element_by_id("su").click()
        time.sleep(2)

    #通过parameterized实现参数化
    @parameterized.expand([("case1","selenium"),("case2","unittest"),                           ("case3","parameterized"),])

    def test_search(self,name,search_key):
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title,search_key + "_百度搜索")

    @classmethod
    def tearDownClass(cls):
        cls.driver.quit()
                          

if __name__=="__main__":
    suit = unittest.TestSuite()
    #添加整个测试类
    suit.addTests(unittest.TestLoader().loadTestsFromTestCase(TestBaidu))
    now_time = time.strftime("%Y-%m-%d  %H_%M_%S")
    with open(now_time + " Baidu_search.html", "wb") as fp:
        runner = HTMLTestRunner(stream=fp,title="百度搜索测试报告",                                description="运行环境：win10， Chrome浏览器",verbosity=2)
        runner.run(suit)
        fp.close()