#encoding=utf-8
from selenium import webdriver
import time
import traceback
#獲取測試數(shù)據(jù)
with open("data.txt") as file_obj:
    datas = file_obj.readlines()

result = []
#遍歷測試數(shù)據(jù)
for data in datas:

    try:
        driver = webdriver.Chrome(executable_path = "e:\\chromedriver")
        url = "https://www.baidu.com"

        driver.get(url)

        driver.find_element_by_id("kw").send_keys(data.strip().split("||")[0].decode("gbk"))

        driver.find_element_by_id("su").click()

        time.sleep(3)

        assert data.strip().split("||")[1].decode("gbk")  in driver.page_source

    except Exception as e:
        traceback.print_exc()
        #因為結(jié)果是ANSI格式保存的，所以編碼為gbk，不至于亂碼
        result.append(data.strip().split("||")[0] +u"失敗\n".encode("gbk"))
    else:
        result.append(data.strip().split("||")[0]+u"成功\n".encode("gbk"))

    time.sleep(2)

#測試結(jié)果寫入文件，result列表的內(nèi)容需要時字符串   
with open("result.txt","a") as file_obj:
    file_obj.writelines(result)

driver.quit()