python用unittest+HTMLTestRunner的框架测试并生成测试报告

时间:2021-10-01 14:35:43

直接贴代码:

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() 生成报告为:
python用unittest+HTMLTestRunner的框架测试并生成测试报告