直接贴代码:
import unittest
from selenium import webdriver
from time import sleep
import os
import time # 定义打开浏览器的方法,这里用的是Chrome,火狐为Firfox,IE为Ie,必须在根目录下对应的driver才能调用
dr = webdriver.Chrome()
# 浏览器最大化
dr.maximize_window() # 创建测试类LoginCase,用unittest的测试框架的格式
class LoginCase(unittest.TestCase):
global path
path = 'F:/Python_test/'
# 定义登录方法,被测试用例调用
def login(self, username, password):
# 需要测试的网页
dr.get('https://passport.cnblogs.com/user/signin') # 需要输入的用户名,变量名和方法中的一致,find_element_by_id('input1')为抓取到的用户名的输入框,用谷歌或者火狐F12可以抓取到
dr.find_element_by_id('input1').send_keys(username) # 需要输入的密码,变量名和方法中的一致,find_element_by_id('input2')为抓取到的密码的输入框
dr.find_element_by_id('input2').send_keys(password) # 点击登陆按钮,find_element_by_id('signin')为抓取到的登陆按钮,click()为点击事件
dr.find_element_by_id('signin').click() # 点击登陆按钮 # 定义测试方法,框架中测试方法以test_开头,底下引号中的中文会在报告中显示,利于清楚的知道测试目的 def test_login_success(self):
'''用户名、密码正确'''
self.login('anyXXX', 'XXXXXX') # 调用定义的login方法,传入正确用户名和密码
sleep(1) # 可以用get_screenshot_as_file方法用来截图,可自定义截图后的保存位置和图片命名
dr.get_screenshot_as_file(path + "login_success.jpg") # 定义了一个实际值,用谷歌或者火狐F12可以抓取到登陆后显示的用户名,.text是获取地址的文本值
# assert先判断需要的实际值是否正确,正确,继续运行用例;如果不正确,不继续运行该用例并返回错误
assert dr.find_element_by_id('lnk_current_user').text, '判断的元素错误,请确认!' # 将实际值赋值给一个变量link,方便比较,可自定义
link = dr.find_element_by_id('lnk_current_user').text # 用assertTrue(x)方法来断言,登录成功后预期的值是否和定义的实际值一致
self.assertTrue('anyXXX' in link) def test_login_pwd_error(self):
'''用户名正确、密码不正确'''
self.login('1', '1') # 正确用户名,错误密码
sleep(1)
dr.get_screenshot_as_file(path + "login_password_error.jpg")
assert dr.find_element_by_id("tip_btn").text, '提示信息类型错误!'
error_message = dr.find_element_by_id('tip_btn').text
#self.assertIn('用户名或密码错误', error_message) # 用assertIn(a,b)方法来断言
assert error_message[0:7],'用户名或密码错误' def test_login_pwd_null(self):
'''用户名正确、密码为空'''
self.login('1', '') # 密码为空
sleep(1)
dr.get_screenshot_as_file(path + "login_password_null.jpg")
assert dr.find_element_by_id("tip_input2").text, '提示信息类型错误!'
error_message = dr.find_element_by_id('tip_input2').text
#self.assertEqual(error_message, '请输入密码') # 用assertEqual(a,b)方法来断言
assert error_message, '请输入密码' def test_login_user_error(self):
'''用户名错误、密码正确'''
self.login('1', '1') # 密码正确,用户名错误
sleep(1)
dr.get_screenshot_as_file(path + "login_username_error.jpg")
assert dr.find_element_by_id("tip_btn").text, '提示信息类型错误!'
error_message = dr.find_element_by_id('tip_btn').text
#self.assertIn('该用户不存在', error_message) # 用assertIn(a,b)方法来断言 a in b
assert error_message[0:5], '该用户不存在' def test_login_user_null(self):
'''用户名为空、密码正确'''
self.login('', '1') # 用户名为空,密码正确
sleep(1)
dr.get_screenshot_as_file(path + "login_username_null.jpg")
assert dr.find_element_by_id("tip_input1").text, '提示信息类型错误!'
error_message = dr.find_element_by_id('tip_input1').text
#self.assertEqual(error_message, '请输入登录用户名') # 用assertEqual(a,b)方法来断言
assert error_message,'请输入登录用户名' # 每个test_执行完执行一次tearDown()方法
def tearDown(self):
sleep(1)
# refresh()方法为刷新浏览器
dr.refresh()
# 关闭浏览器
dr.quit() if __name__ == '__main__':
# 导入HTMLTestRunner库,这句也可以放在脚本开头
import HTMLTestRunner # 定义脚本标题,加u为了防止中文乱码 report_title = u'登陆模块测试报告' # 定义脚本内容,加u为了防止中文乱码
desc = u'博客园登陆模块测试报告详情:' # 定义date为日期,time为时间
date = time.strftime("%Y%m%d")
time = time.strftime("%Y%m%d%H%M%S") # 定义path为文件路径,目录级别,可根据实际情况自定义修改
path = 'F:/Python_test/' + date + "/login/" + time + "/" # 定义报告文件路径和名字,路径为前面定义的path,名字为report(可自定义),格式为.html
report_path = path + "report.html" # 判断是否定义的路径目录存在,不能存在则创建
if not os.path.exists(path):
os.makedirs(path)
else:
pass # 定义一个测试容器
testsuite = unittest.TestSuite() # 将测试用例添加到容器
testsuite.addTest(LoginCase("test_login_success"))
testsuite.addTest(LoginCase("test_login_pwd_error"))
testsuite.addTest(LoginCase("test_login_pwd_null"))
testsuite.addTest(LoginCase("test_login_user_error"))
testsuite.addTest(LoginCase("test_login_user_null")) # 将运行结果保存到report,名字为定义的路径和文件名,运行脚本
with open(report_path, 'wb') as report:
runner = HTMLTestRunner.HTMLTestRunner(stream=report, title=report_title, description=desc)
runner.run(testsuite) # 关闭report,脚本结束
report.close() 生成报告为: