#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()
原文地址：http://blog.51cto.com/13496943/2173349