案例1-自動化測試-資料驅動測試-unittest和ddt

#encoding =utf-8
# -*- coding:utf-8 -*-
from selenium import webdriver
import time,unittest
import logging,traceback
import ddt
from selenium.common.exceptions import NoSuchElementException
logger = logging.getLogger('mylogger1')
logger.setLevel(logging.DEBUG)
# 建立一個handler，用于寫入日志檔案
fh = logging.FileHandler('C:/test.log')
fh.setLevel(logging.DEBUG)
# 再建立一個handler，用于輸出到控制台
ch = logging.StreamHandler()
ch.setLevel(logging.DEBUG)
# 定義handler的輸出格式
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
fh.setFormatter(formatter)
ch.setFormatter(formatter)
# 給logger添加handler
logger.addHandler(fh)
logger.addHandler(ch)

logger.info(u'開始列印日記')
@ddt.ddt
class Testdemo(unittest.TestCase):
    def setUp(self):
        self.driver=webdriver.Firefox(executable_path="C:\\webdriver\geckodriver")
    @ddt.data([u"神奇動物在哪裡",u"葉茨"],[u"瘋狂動物城",u"古德溫"],[u"大話西遊之月光寶盒",u"hahaha"])
    @ddt.unpack
    def test_date(self,testdata,expectdata):
        url="http://www.baidu.com"
        self.driver.get(url)
        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,e:
            logger.error(u"查找的頁面元素不存在，異常資訊："+str(traceback.format_exc()))
        except AssertionError,e:
            logger.info(u"搜尋失敗")
        else:
            logger.info(u"搜尋通過")

    def tearDown(self):
            self.driver.quit()

if __name__=='__main__':
    unittest.main()