#创建gloryroad数据库SQL语句
create_database = 'create database if not exists gloryroad default charset utf8 collate utf8_general_ci;'
#创建testdata表
create_table = """create table if not exists testdata(id int not null auto_increment comment '主键',bookname varchar(40) unique not null comment '书名',author varchar(30) not null comment '作者',primary key(id))engine = innodb character set utf8 comment '测试数据表';
"""

#初始化数据库脚本
import pymysql
from Sql import *class DataBaseInit():#本类用于完成初始化数据库操作#创建数据库，创建数据表，向表中插入测试数据def __init__(self, host, port, dbName, username, password, charset):self.host = hostself.port = portself.db = dbNameself.user = usernameself.passwd = passwordself.charset = charsetdef create(self):try:#连接MySQL数据库conn = pymysql.connect(host = self.host,port = self.port,user = self.user,passwd = self.passwd,charset = self.charset)#获取数据库游标cur = conn.cursor()#创建数据库
            cur.execute(create_database)#选择创建好的gloryroad数据库conn.select_db("gloryroad")#创建测试表
            cur.execute(create_table)except pymysql.Error as e:raise eelse:#关闭游标
            cur.close()#提交操作
            conn.commit()#关闭连接
            conn.close()print("创建数据库及表成功")def insertDatas(self):try:#连接数据库中具体某个库conn = pymysql.connect(host=self.host,port=self.port,db = self.db,user=self.user,passwd=self.passwd,charset=self.charset)cur = conn.cursor()#向测试表中插入测试数据sql = "insert into testdata(bookname, author) values(%s, %s);"res = cur.executemany(sql, [('Selenium Data Driven', 'Flutter'),('Python 基础教程', 'Magnus Lie Hetland'),('算法设计与分析', '张德富'),('计算机网络', '谢希仁')])except pymysql.Error as e:raise eelse:conn.commit()print("初始数据插入成功")#通过打印，确认数据是否插入成功cur.execute("select * from testdata;")for i in cur.fetchall():print(i[1], i[2])cur.close()conn.close()if __name__ == '__main__':db = DataBaseInit(host="localhost",port=3306,dbName="gloryroad",username="root",password="root",charset="utf8")db.create()db.insertDatas()print("数据库初始化结束")

#从数据库中获取测试数据
import pymysql
from DatabaseInit import DataBaseInitclass MyMySQL():def __init__(self, host, port, dbName, username, password, charset):#进行数据库初始化dbInit = DataBaseInit(host, port, dbName, username, password, charset)dbInit.create()dbInit.insertDatas()self.conn = pymysql.connect(host = host,port = port,db = dbName,user = username,passwd = password,charset = charset)self.cur = self.conn.cursor()def getDataFromDataBases(self):#从testdata表中获取需要的测试数据#bookname作为搜索关键词，author作为预期关键词self.cur.execute("select bookname, author from testdata")#从查询区域取回所有的查询结果datasTuple = self.cur.fetchall()return datasTupledef closeDatabase(self):#数据库后期清理工作
        self.cur.close()self.conn.commit()self.conn.close()if __name__ == '__main__':#实例化对象db = MyMySQL(host="localhost",port=3306,dbName="gloryroad",username="root",password="root",charset="utf8")#函数调用print(db.getDataFromDataBases())db.closeDatabase()

#执行数据驱动测试脚本
from selenium import webdriver
import unittest, time
import logging, traceback
import ddt
from MysqlUtil import MyMySQL
from selenium.common.exceptions import NoSuchElementException
from BSTestRunner import BSTestRunner#初始化日志对象
logging.basicConfig(#日志级别level=logging.INFO,#日志格式#时间、代码所在的文件名、代码行号、日志级别名字、日志信息format='%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s',#打印日志的时间datefmt='%a, %Y-%m-%d %H:%M:%S',#日志文件存放的目录(目录必须存在)及日志文件名filename='./dataDrivenReport.log',#打开日志文件的方式filemode='w'
)def getTestDatas():#对象实例化db = MyMySQL(host="localhost",port=3306,dbName="gloryroad",username="root",password="root",charset="utf8")#从数据库测试表中获取测试数据testData = db.getDataFromDataBases()#关闭数据库连接
    db.closeDatabase()#返回数据return testData@ddt.ddt
class TestDemo(unittest.TestCase):def setUp(self):self.driver = webdriver.Chrome()@ddt.data(*getTestDatas())def test_dataDrivenByDatabase(self, data):#对获得的数据进行解包testData, expectData = dataurl = "http://www.baidu.com"#访问百度首页
        self.driver.get(url)#将浏览器窗口最大化
        self.driver.maximize_window()print(testData, expectData)#设置隐式等待时间为10秒self.driver.implicitly_wait(10)try:# 找到搜索输入框,并输入测试数据self.driver.find_element_by_id("kw").send_keys(testData)# 找到搜索按钮，并单击self.driver.find_element_by_id("su").click()time.sleep(3)#断言期望结果是否出现在页面源代码中self.assertTrue(expectData in self.driver.page_source)except NoSuchElementException:logging.error("查找的页面元素不存在,异常堆栈信息:" + str(traceback.format_exc()))except AssertionError:logging.info("搜索%s,期望%s,失败" % (testData, expectData))except Exception:logging.error("未知错误,错误信息:" + str(traceback.format_exc()))else:logging.info("搜索%s,期望%s,通过" % (testData, expectData))def tearDown(self):self.driver.quit()if __name__ == '__main__':# unittest.main()testCase = unittest.TestLoader().loadTestsFromTestCase(TestDemo)# 将多个测试类加载到测试套件中suite = unittest.TestSuite(testCase)filename = "./test.html"fp = open(filename, 'wb')runner = BSTestRunner(stream=fp,title='数据库驱动测试',description='测试用例执行情况：')runner.run(suite)fp.close()