#-*- coding:utf-8 -*-

from selenium import webdriver
import unittest
import HTMLTestRunner  
import sys
from time import sleep
import xlrd

reload(sys)
sys.setdefaultencoding("utf-8")

class LoadBaiduSearchTestData:
    def __init__(self, path):
        self.path = path
    
    def load_data(self):
        # 打開excel文件
        excel = xlrd.open_workbook(self.path)
        
        # 獲取第一個工作表
        table = excel.sheets()[0]
        
        # 獲取行數
        nrows = table.nrows
        
        # 從第二行開始遍歷數據
        # 存入一個list中
        test_data = []
        for i in range(1, nrows):
            test_data.append(table.row_values(i))
        
        # 返回讀取的數據列表    
        return test_data

class BaiduTest(unittest.TestCase):
    """百度首頁搜索測試用例"""
    def setUp(self):
        self.driver = webdriver.Chrome()
        self.driver.implicitly_wait(30)
        self.base_url = u"http://www.baidu.com"
        self.path = u"baidu_search.xlsx"
        
    def test_baidu_search(self):
        driver = self.driver
        print u"開始[case_0001]百度搜索"
        
        
        # 加載測試數據
        test_excel = LoadBaiduSearchTestData(self.path)
        data = test_excel.load_data()
        print data
        
        # 循環參數化
        for d in data:
            # 打開百度首頁
            driver.get(self.base_url)
            
            # 驗證標題
            self.assertEqual(driver.title, u"百度一下，你就知道")
            sleep(1)
            
            driver.find_element_by_id("kw").clear()
            # 參數化 搜索詞
            driver.find_element_by_id("kw").send_keys(d[1])
            sleep(1)
            driver.find_element_by_id("su").click()
            sleep(1)
            
            # 驗證搜索結果標題
            self.assertEqual(driver.title, d[2])
            sleep(2)
            
        
    def tearDown(self):
        self.driver.quit()
        

if __name__ == '__main__':
    testunit = unittest.TestSuite()
    testunit.addTest(BaiduTest('test_baidu_search'))