測試數據
create_database = 'CREATE DATABASE IF NOT EXISTS davieyang DEFAULT CHARSET utf8 COLLATE utf8_general_ci;'
drop_table = 'DROP TABLE testdata;'
create_table = """
CREATE TABLE testdata(
ID int primary key not null auto_increment comment '主鍵',
BOOKNAME varchar(40) unique not null comment '書名',
AUTHOR varchar(30) not null comment '作者'
)engine = innodb character set utf8 comment '測試數據表';
"""
獲取數據庫測試數據方法
"""
__title__ = ''
__author__ = 'davieyang'
__mtime__ = '2018/4/21'
"""
import pymysql
from TestData.SqlScripts import create_table
from TestData.SqlScripts import create_database
from TestData.SqlScripts import drop_table
class MySQL(object):
def __init__(self, host, port, dbName, username, password, charset):
self.conn = pymysql.connect(
host=host,
port=port,
db=dbName,
user=username,
password=password,
charset=charset
)
self.cur = self.conn.cursor()
def create(self):
try:
self.cur.execute(create_database)
self.conn.select_db("davieyang")
self.cur.execute(drop_table)
self.cur.execute(create_table)
'''
cur.execute("drop database if exists davieyang") #如果davieyang數據庫存在則刪除
cur.execute("create database davieyang") #新創建一個數據庫davieyang
cur.execute("use davieyang") #選擇davieyang這個數據庫
# sql 中的內容爲創建一個名爲testdata的表
sql = """create table testdata(id BIGINT,name VARCHAR(20),age INT DEFAULT 1)""" #()中的參數可以自行設置
conn.execute("drop table if exists testdata") # 如果表存在則刪除
conn.execute(sql)# 創建表
# 刪除
# conn.execute("drop table testdata")
conn.close()# 關閉遊標連接
connect.close()# 關閉數據庫服務器連接 釋放內存
'''
except pymysql.Error as e:
raise e
else:
self.cur.close()
self.conn.commit()
self.conn.close()
print(u"創建數據庫和表成功")
def insertDatas(self):
try:
sql = "insert into testdata(bookname, author) values(%s, %s);"
self.cur.executemany(sql, [('selenium xml DataDriven', 'davieyang'),
('selenium excel DataDriven', 'davieyang'),
('selenium ddt data list', 'davieyang')])
except pymysql.Error as e:
raise e
else:
self.conn.commit()
print(u"初始數據插入成功")
self.cur.execute("select * from testData;")
for i in self.cur.fetchall():
print(i[1], i[2])
self.cur.close()
self.conn.close()
def getDataFromDataBase(self):
self.cur.execute("select bookname, author from testdata;")
dataTuple = self.cur.fetchall()
return dataTuple
def closeDataBase(self):
self.cur.close()
self.conn.commit()
self.conn.close()
if __name__ == "__main__":
db = MySQL(
host="localhost",
port=3306,
dbName="davieyang",
username="root",
password="root",
charset="utf8"
)
print(db.getDataFromDataBase())
db.closeDataBase()
測試腳本
"""
__title__ = ''
__author__ = 'davieyang'
__mtime__ = '2018/4/21'
"""
from selenium import webdriver
import unittest
import time
import logging
import traceback
import ddt
from Util.MysqlDBUtil import MySQL
from selenium.common.exceptions import NoSuchElementException
logging.basicConfig(
level=logging.INFO,
format='%(asctime)s %(filename)s[line: %(lineno)d] %(levelname)s %(message)s',
datefmt='%a, %d %b %Y %H:%M:%S',
filename='F:\\DataDriven\\TestResults\TestResults.TestResults',
filemode='w'
)
def getTestDatas():
db = MySQL(
host="localhost",
port=3306,
dbName="davieyang",
username="root",
password="root",
charset="utf8"
)
testData = db.getDataFromDataBase()
db.closeDataBase()
return testData
@ddt.ddt
class DataDrivenByMySQL(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Chrome(executable_path=r"F:\automation\webdriver\chromedriver.exe")
@ddt.data(* getTestDatas())
def test_dataDrivenByMySQL(self, data):
testData, expectData =data
url = "http://www.baidu.com"
self.driver.get(url)
self.driver.maximize_window()
print(testData, expectData)
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 as e:
logging.error(u"查找的頁面元素不存在,異常堆棧信息爲:" + str(traceback.format_exc()))
except AssertionError as e:
logging.info(u"搜索 ‘%s’,期望 ‘%s’ ,失敗" % (testData, expectData))
except Exception as e:
logging.error(u"未知錯誤,錯誤信息:" + str(traceback.format_exc()))
else:
logging.info(u"搜索 ‘%s’,期望 ‘%s’ ,通過" % (testData, expectData))
def tearDown(self):
self.driver.quit()
if __name__ == "__main__":
unittest.main()