importunittestfromddtimportddt, data, unpack#读取txt文件中的内容，strip()方法去掉首位的指定字符。defread_txt():list = []
    file = open('param.txt','r',encoding='utf-8')forlineinfile.readlines():
        list.append(line.strip('\n').split(','))returnlist@ddtclassaaa(unittest.TestCase):defsetUp(self):pass@data(*read_txt())deftest1_data(self,args1):print(args1)deftearDown(self):passif__name__ =='__main__':
    unittest.main()

诸葛,司马
马超,关羽

importtimeimportunittestimportddtfromseleniumimportwebdriverfromselenium.webdriver.common.byimportByfroma.excel_utilimportExcelUtil@ddt.ddtclassTestDemo(unittest.TestCase):eu = ExcelUtil()defsetUp(self)->None:self.driver = webdriver.Chrome()
        self.driver.get("https://www.baidu.com")
        self.driver.maximize_window()
        self.driver.implicitly_wait(10)@ddt.data(*eu.read_excel("D:\\testdata.xlsx","Sheet1"))deftest_data_driver_by_jsonfile(self,value):try:
            testData,expectData = tuple(value)
            self.driver.find_element(By.ID,"kw").send_keys(testData)
            time.sleep(3)
            self.driver.find_element(By.ID,"su").click()
            time.sleep(5)
            self.assertTrue(expectDatainself.driver.page_source)except:
            print("搜索：%s,期望：%s，失败"%(testData,expectData))passelse:
            print("搜索：%s,期望：%s，通过"%(testData,expectData))passdeftearDown(self)->None:self.driver.quit()if__name__ =='__main__':
    unittest.main()

importxlrdclassExcelUtil:#读取excel数据defread_excel(self,excel_path,sheet_name):xls = xlrd.open_workbook(excel_path)
        sheet = xls.sheet_by_name(sheet_name)
        dataList=[]forlineinrange(1,sheet.nrows):
            tempList = []
            tempList.append(sheet.cell_value(line,1))
            tempList.append(sheet.cell_value(line,2))
            dataList.append(tempList)returndataListif__name__ =='__main__':
    data = ExcelUtil().read_excel("d:\\testdata.xlsx","Sheet1")
    print(data)

importtimeimportpymysqlaspymysqlclassDataBase:#删除datadriver数据库的SQL语句drop_database =" drop database if exists datadriver;"#创建datadriver的SQL语句：create_database ="create database datadriver default charset utf8 COLLATE utf8_general_ci;"#创建testdata表的SQL语句：create_table ="""
                   create table testdata(
                        id int not null auto_increment comment '主键',
                        searchContent varchar(40) unique not null comment '百度搜索的内容',
                        assertContent varchar(30) not null comment '断言的内容',
                        primary key(id)
                   );
                   """def__init__(self,host,port,username,password,db):self.host = host
        self.port = port
        self.username = username
        self.password = password
        self.db = db#连接数据库创建数据库和表defcreate_databse_and_table(self):try:#连接Mysql数据库。conn = pymysql.connect(host=self.host,port=self.port,user=self.username,password=self.password)#获取数据库游标cur = conn.cursor()#删除数据库cur.execute(self.drop_database)
            time.sleep(3)#执行创建数据库的SQL语句cur.execute(self.create_database)
            time.sleep(3)#选中创建好的datadriver数据库。conn.select_db("datadriver")#执行创建表的SQL语句cur.execute(self.create_table)# 提交操作conn.commit()exceptException:raiseelse:
            cur.close()#关闭游标conn.close()#关闭连接print("创建数据库和表成功！")#插入数据definsert_data(self):try:# 连接Mysql数据库。conn = pymysql.connect(host=self.host, port=self.port, user=self.username, password=self.password,db=self.db)# 获得数据库游标cur = conn.cursor()# 执行插入数据的SQL语句sql ="insert into testdata(searchContent,assertContent) values(%s,%s)"cur.executemany(sql,[('神奇动物在哪里','叶茨'),('疯狂动物城','霍华德'),('杜兰特','凯文')])# 提交操作conn.commit()exceptException:raiseelse:
            cur.close()# 关闭游标conn.close()# 关闭连接print("数据插入成功！")#查询数据defselect_data(self):result =""try:
            conn = pymysql.connect(host=self.host,port=self.port,user=self.username,password=self.password,db=self.db)
            cur = conn.cursor()
            cur.execute("select searchContent,assertContent from testdata;")
            result=cur.fetchall()exceptException:raiseelse:
            cur.close()# 关闭游标conn.close()# 关闭连接print("数据查询成功！")returnresultif__name__ =='__main__':
    db = DataBase("localhost",3306,"root","admin","datadriver")
    db.create_databse_and_table()
    db.insert_data()
    print(db.select_data())

importtimeimportunittestimportwarningsimportddtfromseleniumimportwebdriverfromselenium.webdriver.common.byimportByfroma.databaseimportDataBasedefget_data():db = DataBase("localhost",3306,"root","admin","datadriver")
    data = db.select_data()returndata@ddt.ddtclassTestDemo(unittest.TestCase):defsetUp(self)->None:warnings.simplefilter('ignore',ResourceWarning)#处理资源警告。self.driver = webdriver.Chrome()
        self.driver.get("https://www.baidu.com")
        self.driver.maximize_window()
        self.driver.implicitly_wait(10)@ddt.data(*get_data())deftest_data_driver_by_jsonfile(self,data):try:
            testData,expectData = data
            self.driver.find_element(By.ID,"kw").send_keys(testData)
            time.sleep(3)
            self.driver.find_element(By.ID,"su").click()
            time.sleep(5)
            self.assertTrue(expectDatainself.driver.page_source)except:
            print("搜索：%s,期望：%s，失败"%(testData,expectData))passelse:
            print("搜索：%s,期望：%s，通过"%(testData,expectData))passdeftearDown(self)->None:self.driver.quit()if__name__ =='__main__':
    unittest.main()