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

猜你喜欢

转载自www.cnblogs.com/lbfang/p/12296366.html