selenium python自动化测试 ddt数据驱动

安装ddt  pip install ddt

直接上代码：

# coding:utf-8
import ddt
import time
import excelunit
import unittest
from selenium import webdriver
 
# 测试数据
data = excelunit.ExcelUtil("test.xlsx", "sheet1")
testData = data.dict_data()
print testData
 
 
@ddt.ddt
class Bolg(unittest.TestCase):
    """登录博客"""
    def setUp(self):
        self.driver = webdriver.Firefox()
        url = "https://passport.cnblogs.com/user/signin"
        self.driver.get(url)
        self.driver.implicitly_wait(10)
 
    def login(self, username, psw):
        """登录方法,账号和密码参数化"""
        self.driver.find_element_by_id("input1").send_keys(username)
        self.driver.find_element_by_id("input2").send_keys(psw)
        self.driver.find_element_by_id("signin").click()
        time.sleep(3)
 
    def is_login_sucess(self):
        """判断是否获取到登录账户名称"""
        try:
            text = self.driver.find_element_by_id("lnk_current_user").text
            print text
            return True
        except:
            return False
 
    @ddt.data(*testData)
    def test_login(self, data):
        """登录案例"""
        print ("当前测试数据%s" % data)
        # 调用登录方法
        self.login(data["username"], data["password"])
        # 判断结果
        result = self.is_login_sucess()
        self.assertTrue(result)
 
    def tearDown(self):
        self.driver.quit()
 
 
if __name__ == "__main__":
    unittest.main()