# 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,杨过