unittest实战（三）：用例编写

# coding:utf-8
import unittest
from selenium import webdriver
import time
from ddt import ddt, data, unpack, file_data
import yaml

chrome_driver = r"C:\Program Files (x86)\Python38-32\Lib\site-packages\selenium\webdriver\chrome\chromedriver.exe"

def readfile():
    params = []
    file = open('param.txt', 'r', encoding='utf-8')
    for line in file.readlines():
        params.append(line.strip('\n').split(','))
    print(params)
    return params

@ddt
class forTesttest(unittest.TestCase):
    def setUp(self) -> None:
        self.driver = webdriver.Chrome(executable_path=chrome_driver)
        self.driver.get('https://www.baidu.com')

    def tearDown(self) -> None:
        time.sleep(5)
        self.driver.quit()

    @data(*readfile())
    @unpack
    def test_a(self, url, txt):
        self.driver.get(url)
        self.driver.find_element_by_id('kw').send_keys(txt)
        self.driver.find_element_by_id('su').click()

if __name__ == '__main__':
    unittest.main()

param.txt

https://www.baidu.com,房龙波
https://www.baidu.com,陈亚茸
https://www.baidu.com,小龙女
https://www.baidu.com,杨过