#encoding=utf-8
from selenium import webdriver
import time
import traceback
#获取测试数据
with open("data.txt") as file_obj:
    datas = file_obj.readlines()

result = []
#遍历测试数据
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()
        #因为结果是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)

#测试结果写入文件，result列表的内容需要时字符串   
with open("result.txt","a") as file_obj:
    file_obj.writelines(result)

driver.quit()