简介：
数据驱动数据库驱动方式,就是数据配置在数据库里面，主程序调用的时候每次用从数据库里取出的数据作为参数，进行操作，
需要掌握的地方是对数据库的操作，要灵活的找到目标数据
测试数据：
创建数据库、表
Sql.py
#encoding=utf-8
# 创建gloryroad数据库sql语句
create_database = 'CREATE DATABASE IF NOT EXISTS gloryroad DEFAULT CHARSET utf8 COLLATE utf8_general_ci;'
# 创建testdata表
drop_table_if_exist_sql="drop table if exists testdata;"
create_table = """
create table testdata(
id int not null auto_increment comment '主键',
bookname varchar(40) unique not null comment '书名',
author varchar(30) not null comment '作者',
test_result varchar(30) default null,
primary key(id)
)engine=innodb character set utf8 comment '测试数据表';
"""
数据库初始化，创建测试数据
DatabaseInit.py
#encoding=utf-8
import pymysql
from Sql import *
class DataBaseInit(object):
# 本类用于完成初始化数据操作
# 创建数据库，创建数据表，向表中插入测试数据
def __init__(self, host, port, dbName, username, password, charset):
self.host = host
self.port = port
self.db = dbName
self.user = username
self.passwd = password
self.charset = charset
def 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(drop_table_if_exist_sql)
cur.execute(create_table)
except pymysql.Error as e:
raise e
else:
# 关闭游标
cur.close()
# 提交操作
conn.commit()
# 关闭连接
conn.close()
print (u"创建数据库及表成功")
def insertDatas(self):
try:
# 连接mysql数据库中具体某个库
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 WebDriver实战宝典', '吴晓华'),
('HTTP权威指南', '古尔利'),
('探索式软件测试', '惠特克'),
('暗时间', '刘未鹏')])
except pymysql.Error as e:
raise e
else:
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="test",
charset="utf8"
)
db.create()
db.insertDatas()
print("数据库初始化结束")
单独执行结果：
创建数据库及表成功
初始数据插入成功
Selenium WebDriver实战宝典 吴晓华
HTTP权威指南 古尔利
探索式软件测试 惠特克
暗时间 刘未鹏
数据库初始化结束
MysqlUtil.py
#encoding=utf-8
import pymysql
from DatabaseInit import DataBaseInit
class MyMySQL(object):
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 datasTuple
def 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="test",
charset="utf8"
)
print (db.getDataFromDataBases())
单独执行的结果：
创建数据库及表成功
初始数据插入成功
Selenium WebDriver实战宝典 吴晓华
HTTP权威指南 古尔利
探索式软件测试 惠特克
暗时间 刘未鹏
(('Selenium WebDriver实战宝典', '吴晓华'), ('HTTP权威指南', '古尔利'), ('探索式软件测试', '惠特克'), ('暗时间', '刘未鹏'))
最后执行的脚本
data_driven_by_mysql_1.py
#encoding=utf-8
from selenium import webdriver
import time
import datetime
from openpyxl import *
import pymysql
def get_test_data():
conn = pymysql.connect(
host = "localhost",
port = 3306,
user = "root",
passwd = "test" ,
db = "gloryroad",
charset = "utf8"
)
# 使用cursor()方法获取数据库的操作游标
cursor = conn.cursor()
cursor.execute("select * from testdata;")
resSet = cursor.fetchall()
print("共%s条数据。" %len(resSet))
print( resSet)
# 关闭游标
cursor.close()
# 提交事务
conn.commit()
# 关闭数据库连接
conn.close()
return resSet
def update_test_result(data,result):
conn = pymysql.connect(
host = "localhost",
port = 3306,
user = "root",
passwd = "test" ,
db = "gloryroad",
charset = "utf8"
)
# 使用cursor()方法获取数据库的操作游标
cursor = conn.cursor()
print( 'update testdata set test_result="'+result+'" where bookname="'+data+'";')
update=cursor.execute('update testdata set test_result="'+result+'" where bookname="'+data+'";')
print( u"修改语句受影响的行数：", update)
# 关闭游标
cursor.close()
# 提交事务
conn.commit()
# 关闭数据库连接
conn.close()
driver = webdriver.Firefox(executable_path="D:\python\geckodriver")
test_result=[]
for data in get_test_data():
try:
driver.get("http://www.baidu.com")
driver.find_element_by_id("kw").send_keys(data[1])
driver.find_element_by_id("su").click()
time.sleep(3)
assert data[2] in driver.page_source
update_test_result(data[1], "成功")
except AssertionError as e:
print( data[2] + "断言失败")
update_test_result(data[1], "断言失败")
except Exception as e:
print(e)
print(data[1] + "测试执行出现异常")
update_test_result(data[1], "执行出现异常")
driver.quit()
执行结果：
D:pythonpython.exe D:/db_data/data_driven_by_mysql_1.py
共4条数据。
((1, 'Selenium WebDriver实战宝典', '吴晓华', None), (2, 'HTTP权威指南', '古尔利', None), (3, '探索式软件测试', '惠特克', None), (4, '暗时间', '刘未鹏', None))
update testdata set test_result="成功" where bookname="Selenium WebDriver实战宝典";
修改语句受影响的行数： 1
update testdata set test_result="成功" where bookname="HTTP权威指南";
修改语句受影响的行数： 1
update testdata set test_result="成功" where bookname="探索式软件测试";
修改语句受影响的行数： 1
update testdata set test_result="成功" where bookname="暗时间";
修改语句受影响的行数： 1
Process finished with exit code 0
总结：
数据驱动DB驱动、Excel驱动方式和txt驱动方式原理大同小异，都是把数据从文件中取出来，传进主程序，脚本运行完毕后再把结果写入文件中；
难点在于对不同的文件类型进行读取可能需要专门的程序包来处理；都是对基础的运用进行整合，真正项目上用到肯定比这复杂的多。。。