1. 程式人生 > >七:python-unittest資料驅動(ddt)

七:python-unittest資料驅動(ddt)

前言

在設計用例的時候,有些用例只是引數資料的輸入不一樣,比如登入這個功能,操作過程但是一樣的。如果用例重複去寫操作過程會增加程式碼量,對應這種多組資料的測試用例,可以用資料驅動設計模式,一組資料對應一個測試用例,用例自動載入生成。

一、環境準備

1.安裝ddt模組,開啟cmd輸入pip install ddt線上安裝

>>pip install ddt

二、資料驅動原理

1.測試資料為多個字典的list型別

2.測試類前加修飾@ddt.ddt

3.case前加修飾@ddt.data()

4.執行後用例會自動載入成三個單獨的用例

5.測試結果:

Testing started at 21:51 ...
start!
{'username': 'selenium\xe7\xbe\xa4', 'psw': '232607095'}
end!
start!
{'username': 'python\xe7\xbe\xa4', 'psw': '226296743'}
end!
start!
{'username': 'appium\xe7\xbe\xa4', 'psw': '512200893'}
end!

三、selenium案例

3.程式碼參考如下

# 測試資料
testData = data.dict_data()
print testData

@ddt.ddt
class Bolg(unittest.TestCase):
    u'''登入部落格'''
    def setUp(self):
        self.driver = webdriver.Firefox()
        url = "https://passport.cnblogs.com/user/signin"
        self.driver.get(url)
        self.driver.implicitly_wait(30)

    def login(self, username, psw):
        u'''這裡寫了一個登入的方法,賬號和密碼引數化'''
        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):
        u'''判斷是否獲取到登入賬戶名稱'''
        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):
        u'''登入案例參考'''
        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()