Selenium（Python） ddt讀取MySQL資料驅動

標籤：eth   資料   sleep   term   sqlt   cut   name   讀取   body   

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()

 

Selenium（Python） ddt讀取MySQL資料驅動