import unittest

from time import sleep

from ddt import ddt, data

from pymysql import connect

from selenium import webdriver

def getMySQLTestData():

# 查詢數據庫的方法

db = connect(host="localhost",

user="root",

password="123456",

db="world",

port=3306,

charset="utf8")

# 打開數據庫連接

cur = db.cursor()

# 使用cursor()方法獲取操作游標

sql = "SELECT `search_word`, `search_result` FROM testdata;"

# sql語句

cur.execute(sql)

# 執行sql語句

results = cur.fetchall()

# 獲取查詢的結果

db.commit()

# 提交

cur.close()

# 關閉游標

db.close()

# 斷開數據庫連接

return results

# 返回一個list

@ddt

# ddt驅動

class MyTestCase(unittest.TestCase):

def setUp(self):

self.driver = webdriver.Firefox()

self.base_url = "https://www.baidu.com/"

self.driver.get(self.base_url)

self.driver.maximize_window()

sleep(2)

@data(*getMySQLTestData())

# 傳參

def test_something(self, sqlTestData):

searchTerm, searchResult = sqlTestData

driver = self.driver

driver.find_element_by_xpath(".//*[@id=‘kw‘]").send_keys(searchTerm)

driver.find_element_by_xpath(".//*[@id=‘su‘]").click()

sleep(2)

responseText = driver.find_element_by_xpath(".//*[@id=‘1‘]/h3/a").text

self.assertEqual(responseText, searchResult)

def tearDown(self):

self.driver.close()

self.driver.quit()

if __name__ == ‘__main__‘:

unittest.main()

總結

以上是生活随笔為你收集整理的python读取mysql数据_Selenium（Python） ddt读取MySQL数据驱动的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。