学习了一个月的selenium+python，终于学有所成，下面以一个简单的项目来总结学习所得。
1. 项目结构
在项目结构中，大家要注意到：每一个源文件夹中都要有一个__init__.py文件，确记，这是规定，不解析。
先简单解说一下项目结构：
1) config源文件夹，这个主要存放一些配置文件，如里面的golobalparameter.py就是存放一些公共参数的，我这个文件里面主要把一些文件存放的路径写成了公共参数，这样子，要修改路径的时候直接打开这个文件就可以修改，而不用一个个文件地找参数。
2) data文件夹，是个普通的文件夹，里面存放了测试数据excel文件；
3) error_img文件夹，一样是普通的文件夹，测试过程中如果发生异常，通过代码可以自动截图然后把图片存放到error_img文件夹中。
4) log文件夹，也是普通文件夹，用于存放测试过程中输出的日志文件，这个日志文件是一次生成然后不断写入的，一旦生成了log.log文件，那么不管你执行多少次测试，输出了多少日志，都会写入一个log.log文件中。当然，如果把log.log删除之后，执行测试的时候也会再次生成的。
5) report文件夹，也是普通的文件夹，用于存放最后生成的.html测试报告；
6) src源文件夹，这个源文件夹下包含了两子源文件夹，一个是common源文件夹，用于存放一些公共类；另一个是test_case源文件夹，用于存放各测试模块的代码。
7) runtest.py文件，用于执行测试。
8) runner.bat文件，是用于设置windows定时执行测试任务所要用到的的cmd命令，因为我后来用了jenkins来做定时任务，所以可以不管。
2. 项目所有代码文件
1) globalparameter.py
# coding:utf-8
__author__ = 'helen'
import time
'''
配置全局参数
'''
# 项目的绝对路径（因为 windows执行时需要绝对路径才能执行通过）
project_path = "D:\for2017\SPframework-Helen\"
# 测试用例代码存放路径（用于构建suite,注意该文件夹下的文件都应该以test开头命名）
test_case_path = project_path+"src\test_case"
# 日志文件存储路径
log_path = project_path+"log\mylog.log"
# 测试报告存储路径，并以当前时间作为报告名称前缀
report_path = project_path+"report\"
report_name = report_path+time.strftime('%Y%m%d%H%S', time.localtime())
# 异常截图存储路径,并以当前时间作为图片名称前缀
img_path = project_path+"error_img\"+time.strftime('%Y%m%d%H%S', time.localtime())
# 设置发送测试报告的公共邮箱、用户名和密码
smtp_sever = 'smtp.xxxx.com' # 邮箱SMTP服务，各大运营商的smtp服务可以在网上找，然后可以在foxmail这些工具中验正
email_name = "li***@gpdi.com" # 发件人名称
email_password = "*****" # 发件人登录密码
email_To = '50****@qq.com;5*****016@qq.com;h****@163.com' # 收件人
2) excel_data.py
# coding:utf-8
__author__ = 'helen'
import xlrd
from src.common import log
'''
读取excel文件
'''
class excel:
def __init__(self):
self.mylog = log.log()
def open_excel(self,file):
try:
data = xlrd.open_workbook(file)
return data
except Exception, e:
self.mylog.error(u"打开excel文件失败")
def excel_table(self,file, sheetName):
data = self.open_excel(file)
# 通过工作表名称，获取到一个工作表
table = data.sheet_by_name(sheetName)
# 获取行数
Trows = table.nrows
# 获取 第一行数据
Tcolnames = table.row_values(0)
lister = []
for rownumber in range(1,Trows):
row = table.row_values(rownumber)
if row:
app = {}
for i in range(len(Tcolnames)):
app[Tcolnames[i]] = row[i]
lister.append(app)
return lister
3) log.py
# coding:utf-8
__author__ = 'helen'
import logging
from config import globalparameter
'''
配置日志文件，输出INFO级别以上的日志
'''
class log:
def __init__(self):
self.logname = "mylog"
def setMSG(self, level, msg):
# 之前把下面定义log的一大段代码写在了__init__里面，造成了日志重复输出
# 此大坑，谨记谨记！！！!
logger = logging.getLogger()
# 定义Handler输出到文件和控制台
fh = logging.FileHandler(globalparameter.log_path)
ch = logging.StreamHandler()
# 定义日志输出格式
formater = logging.Formatter("%(asctime)s %(levelname)s %(message)s' ")
fh.setFormatter(formater)
ch.setFormatter(formater)
# 添加Handler
logger.addHandler(fh)
logger.addHandler(ch)
# 添加日志信息，输出INFO级别的信息
logger.setLevel(logging.INFO)
if level=='debug':
logger.debug(msg)
elif level=='info':
logger.info(msg)
elif level=='warning':
logger.warning(msg)
elif level=='error':
logger.error(msg)
# 移除句柄，否则日志会重复输出
logger.removeHandler(fh)
logger.removeHandler(ch)
fh.close()
def debug(self, msg):
self.setMSG('debug', msg)
def info(self, msg):
self.setMSG('info', msg)
def warning(self, msg):
self.setMSG('warning', msg)
def error(self, msg):
self.setMSG('error', msg)
4) send_email.py
# coding:utf-8
__author__ = 'helen'
import os,smtplib
from config import globalparameter
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
from src.common import log
'''
邮件发送最新的测试报告
'''
class send_email:
def __init__(self):
self.mylog = log.log()
# 定义邮件内容
def email_init(self,report,reportName):
with open(report,'rb')as f:
mail_body = f.read()
# 创建一个带附件的邮件实例
msg = MIMEMultipart()
# 以测试报告作为邮件正文
report_file = MIMEText(mail_body,'html','utf-8')
# 定义附件名称（附件的名称可以随便定义，你写的是什么邮件里面显示的就是什么）
report_file["Content-Disposition"] = 'attachment;filename='+reportName
msg.attach(report_file) # 添加附件
msg['Subject'] = '自动化测试报告' # 邮件标题
msg['From'] = globalparameter.email_name #发件人
msg['To'] = globalparameter.email_To #收件人列表
try:
server = smtplib.SMTP(globalparameter.smtp_sever)
server.login(globalparameter.email_name,globalparameter.email_password)
server.sendmail(msg['From'],msg['To'].split(';'),msg.as_string())
server.quit()
except smtplib.SMTPException:
self.mylog.error(u'邮件发送测试报告失败 at'+__file__)
def sendReport(self):
# 找到最新的测试报告
lists = os.listdir(globalparameter.report_path)
report_list = sorted(lists)
new_report = os.path.join(globalparameter.report_path,report_list[-1])
# 发送邮件
self.email_init(new_report,report_list[-1])
5) runtest.py
# coding:utf-8
__author__ = 'helen'
import unittest,time,HTMLTestRunner
from config import globalparameter
from src.common import send_email
'''
构建测试套件，并执行测试
'''
# 构建测试集,包含src/test_case目录下的所有以test开头的.py文件
suite = unittest.defaultTestLoader.discover(start_dir=globalparameter.test_case_path,pattern='test*.py')
# 执行测试
if __name__=="__main__":
report_name = globalparameter.report_name+"Report.html"
fb = open(report_name,'wb')
runner = HTMLTestRunner.HTMLTestRunner(
stream=fb,
title=u'自动化测试报告',
description=u'项目描述。………'
)
runner.run(suite)
fb.close()
# 发送邮件
time.sleep(10) # 设置睡眠时间，等待测试报告生成完毕（这里被坑了＝＝）
email = send_email.send_email()
email.sendReport()
6) test_baidu.py
# -*- coding: utf-8 -*-
from selenium import webdriver
from selenium.common.exceptions import NoAlertPresentException
from selenium.webdriver.common.action_chains import ActionChains
import unittest
from config import globalparameter
from src.common import excel_data, log
'''
百度页面相关测试用例
'''
class TestBaidu(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Firefox()
self.driver.implicitly_wait(30)
self.base_url = "https://www.baidu.com/"
self.verificationErrors = []
self.accept_next_alert = True
self.mylog = log.log()
self.excel = excel_data.excel()
def test_baidu_search(self):
u'''百度搜索功能测试'''
driver = self.driver
listdata = self.excel.excel_table("D:\for2017\SPframework-Helen\data\baidu.xlsx","search")
print(len(listdata))
if (len(listdata)<=0):
self.mylog.error(u"读取excel参数异常，参数列表小于等于0")
assert 0,u"读取excel参数异常，参数列表小于等于0"
for i in range(0,len(listdata)):
try:
driver.get(self.base_url + "/")
driver.find_element_by_id("kw").clear()
driver.find_element_by_id("kw").send_keys(listdata[i]["keyword"])
driver.find_element_by_id("su").click()
URL_title = driver.title
self.assertEqual(URL_title, listdata[i]['keyword']+u"_百度搜索", u"搜索失败")
self.mylog.info(u'百度搜索成功 '+__file__)
except:
driver.get_screenshot_as_file(globalparameter.img_path+"test_baidu.png")
self.mylog.error(u"百度搜索失败"+__file__)
def test_baidu_changeTo_hao123(self):
u'''测试从百度首页跳转到hao123网站'''
driver = self.driver
try:
driver.get(self.base_url+"/")
driver.find_element_by_link_text("hao123").click()
self.assertEqual(driver.current_url,"https://www.hao123.com/")
self.mylog.info(u'从百度成功跳转到hao123网站')
except:
driver.get_screenshot_as_file(globalparameter.img_path+"test_baidu_changeTo_hao123.png")
self.mylog.error(u'从百度跳转到hao123网站失败')
def test_baidu_more(self):
u'''打开百度产品大全页面'''
driver = self.driver
try:
driver.get(self.base_url+"/")
mouse = driver.find_element_by_link_text(u"更多产品")
ActionChains(driver).move_to_element(mouse).perform()
driver.find_element_by_link_text(u"全部产品>>").click()
self.assertEqual(driver.current_url,"https://www.baidu.com/more/")
self.mylog.info(u'打开百度产品页面成功')
except:
driver.get_screenshot_as_file(globalparameter.img_path+"test_baidu_more.png")
self.mylog.error(u'打开百度产品页面失败')
def is_alert_present(self):
try: self.driver.switch_to_alert()
except NoAlertPresentException as e: return False
return True
def close_alert_and_get_its_text(self):
try:
alert = self.driver.switch_to_alert()
alert_text = alert.text
if self.accept_next_alert:
alert.accept()
else:
alert.dismiss()
return alert_text
finally: self.accept_next_alert = True
def tearDown(self):
self.driver.close()
self.assertEqual([], self.verificationErrors)
if __name__ == "__main__":
unittest.main()
7) test_sogo.py
# -*- coding: utf-8 -*-
from selenium import webdriver
from selenium.common.exceptions import NoAlertPresentException
import unittest,time
from src.common import log
from config import globalparameter
'''
搜狗页面相关测试
'''
class TestSogo(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Firefox()
self.driver.implicitly_wait(30)
self.base_url = "https://www.sogou.com/"
self.verificationErrors = []
self.accept_next_alert = True
self.mylog = log.log()
def test_sogo(self):
u'''搜狗搜索功能测试'''
try:
driver = self.driver
driver.get(self.base_url + "/")
driver.find_element_by_id("query").clear()
driver.find_element_by_id("query").send_keys("selenium")
driver.find_element_by_id("query").submit()
time.sleep(5)
self.assertEqual(driver.title, u"selenium - 搜狗搜索", u"sogo搜索模块出错")
self.mylog.info(u"sogo搜索模块成功 "+__file__)
except:
driver.get_screenshot_as_file(globalparameter.img_path+"test_sogo.png")
self.mylog.error(u"sogo搜索模块失败 "+__file__)
def is_alert_present(self):
try: self.driver.switch_to_alert()
except NoAlertPresentException as e: return False
return True
def close_alert_and_get_its_text(self):
try:
alert = self.driver.switch_to_alert()
alert_text = alert.text
if self.accept_next_alert:
alert.accept()
else:
alert.dismiss()
return alert_text
finally: self.accept_next_alert = True
def tearDown(self):
self.driver.close()
self.assertEqual([], self.verificationErrors)
if __name__ == "__main__":
unittest.main()
3. 使用jenkins设置定时执行测试任务
这个傻瓜式配置即可，不写了。