【使用unittest框架】使用unittest重构测试用例

前言

一直想学习自动化测试，但是都没行动，业余时间学习零零碎碎并记录20210418。

 

5、使用unittest框架

1. unittest框架简介

2. Unittest中类方法和实例方法

3. 测试用例和测试套件

4. 使用unittest重构测试用例

5. 为项目生产测试报告

 

重构步骤

1. 拷贝上一章测试用例
2. 继承测试用例基础把object改为unittest.TestCase
3. 将__init__修改为setup
4. 将assert断言，修改为unittest断言
5. 关掉浏览器放到teardown
6. 注意：第二次测试清空数据
7. 注意：顺序测试要加序号，例如：test1、test2

实际操作

1、拷贝上一章测试用例

2、继承测试用例基础把object改为unittest.TestCase

3、将__init__修改为setup

4、将assert断言，python断言修改为unittest断言，不改也不影响，强迫症想统一。

5、关掉浏览器放到teardown

注意：第二次测试清空数据

注意：顺序测试要加序号，例如：test1、test2

 

代码分享

1、用户登录重构

import unittest
from time import sleep

from selenium import webdriver
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.wait import WebDriverWait


class TestUserLogin(unittest.TestCase):

    @classmethod
    def setUp(cls) -> None:
        cls.driver = webdriver.Chrome()
        cls.driver.get('http://localhost:8080/jpress/user/login')
        cls.driver.maximize_window()
    # def __init__(self):
    #     self.driver = webdriver.Chrome()
    #     self.driver.get('http://localhost:8080/jpress/user/login')
    #     # self.driver.maximize_window()

    # 测试用户登录，用户名错误
    def test1_user_login_userError(self):
        # 用户名为空
        user = ''  #实际正确的是amdin
        pwd = 'admin'
        expected = '账号不能为空'

        # 输入用户名
        self.driver.find_element_by_name('user').send_keys(user)
        # 输入密码
        self.driver.find_element_by_name('pwd').send_keys(pwd)
        # 点击登录
        self.driver.find_element_by_xpath('/html/body/div/div/div/form/div[3]/div/button').click()  #千万别忘记click

        # 因为有弹窗所以需要切换到弹窗上,并等待弹窗
        WebDriverWait(self.driver, 5).until(EC.alert_is_present())

        alert = self.driver.switch_to.alert
        # python 的断言
        # 获取弹窗上的文字来比较
        assert alert.text == expected
        sleep(3)
        alert.accept()

        sleep(5)


        # self.driver.quit()

    def tearDown(self) -> None:
        self.driver.quit()

    # 测试用户登录成功
    def test2_user_login_ok(self):
        # 用户名为空
        user = 'admin'
        pwd = 'admin'
        expected = '用户中心'

        # 输入用户名
        self.driver.find_element_by_name('user').clear()    # 为了把上面案例输入的值清空
        self.driver.find_element_by_name('user').send_keys(user)
        # 输入密码
        self.driver.find_element_by_name('pwd').clear()
        self.driver.find_element_by_name('pwd').send_keys(pwd)
        # 点击登录
        # self.driver.find_element_by_xpath('/html/body/div/div/div/form/div[3]/div/button').click()
        self.driver.find_element_by_css_selector('body > div > div > div > form > div.row > div > button').click()
        # 等待标题
        WebDriverWait(self.driver, 5).until(EC.title_is(expected))

        sleep(3)
        #验证 因为没有错误弹窗  所以用到了title断言判断
        # assert self.driver.title == expected
        self.assertEqual(self.driver.title,expected)

if __name__ == '__main__':
    unittest.main()

运行结果：

2、管理员登录重构

代码

import os
import unittest
from time import sleep, strftime, localtime

from PIL import Image
from selenium import webdriver
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.wait import WebDriverWait

from lib.ShowapiRequest import ShowapiRequest
from util import util

class TestAdminLogin(unittest.TestCase):

    def setUp(self) -> None:
        self.driver = webdriver.Chrome()
        self.driver.get('http://localhost:8080/jpress/admin/login')
        print('尝试管理员登录')

    # def __init__(self):
    #     self.driver = webdriver.Chrome()
    #     self.driver.get('http://localhost:8080/jpress/admin/login')
    #     # self.driver.maximize_window()

    # 测试管理员用户登录，验证码错误
    def test1_admin_login_Error(self):
        # 用户名为空
        user = 'admin'  #实际正确的是amdin
        pwd = 'admin'
        captcha = '123456'
        expected = '验证码不正确，请重新输入'

        # 输入用户名
        self.driver.find_element_by_name('user').send_keys(user)
        # 输入密码
        self.driver.find_element_by_name('pwd').send_keys(pwd)
        # 输入验证码
        self.driver.find_element_by_name('captcha').send_keys(captcha)
        # 点击登录
        self.driver.find_element_by_xpath('//*[@id="form"]/div[4]/div/button').click()  #千万别忘记click

        # 因为有弹窗所以需要切换到弹窗上,并等待弹窗
        WebDriverWait(self.driver, 5).until(EC.alert_is_present())

        alert = self.driver.switch_to.alert
        # python 的断言
        # 获取弹窗上的文字来比较
        assert alert.text == expected
        sleep(2)
        alert.accept()

        sleep(2)

    # def tearDown(self) -> None:
    #     self.driver.quit()

    # 测试用户登录成功
    def test2_admin_login_ok(self):

        # 用户名为空
        user = 'admin'
        pwd = 'admin'
        expected = 'Jpress后台'

        # 输入用户名
        self.driver.find_element_by_name('user').clear()    # 为了把上面案例输入的值清空
        self.driver.find_element_by_name('user').send_keys(user)
        # 输入密码
        self.driver.find_element_by_name('pwd').clear()
        self.driver.find_element_by_name('pwd').send_keys(pwd)
    # 自动识别验证码test_yzmPro
    # captcha = util.test_yzmPro()  # 不知道为啥直接调用参数还是什么问题，直接把原来写的调用AI库识别复杂验证码拷贝过来调试了
        path = '/Users/ff/PycharmProjects_py3/Selenium_project/'
        path = os.path.join(path, 'screenshots')
        if not os.path.exists(path):  # 如果找不到/Users/ff/PycharmProjects_py3/Selenium_project/screenshots就创建
            os.mkdir(path)
        # 设置要截图的文件名：自定义名称+系统时间命名+后缀.png
        picture_name1 = '验证码未切' + strftime('%Y_%m_%d_%H_%M_%S', localtime()) + '.png'
        path = os.path.join(path, picture_name1)  # 截图1的路径+名称
        self.driver.get_screenshot_as_file(path)  # 截图保存
        print(path)

        # 找到验证码坐标，利用pil模块中的抠图方法crop，把图片抠出来，保存为picture_name2
        # yzm_img = browser.find_element_by_xpath('//*[@id="captchaimg"]')
        yzm_img = self.driver.find_element_by_id("captchaImg")
        print(yzm_img.location)  # 打印左上角坐标{'x': 547, 'y': 447}

        left = yzm_img.location['x']
        top = yzm_img.location['y']
        width = yzm_img.size['width'] + left
        height = yzm_img.size['height'] + top

        dpr = self.driver.execute_script(
            'return window.devicePixelRatio')  # execute_script（）：执行js，使devicePixelRatio 能够返回当前显示设备的物理像素分辨率与 CSS 像素分辨率的比率
        print(dpr)

        # # 打开截图1
        im = Image.open(path)
        # 把图片抠出来，保存为picture_name2
        img = im.crop((left * dpr, top * dpr, width * dpr, height * dpr))
        picture_name2 = '验证码切图' + strftime('%Y_%m_%d_%H_%M_%S', localtime()) + '.png'
        print(picture_name2)
        img.save(picture_name2)  # 这种方式图片默认是保存到main.py当前路径下

        # 这里是复杂验证码获取

        r = ShowapiRequest("http://route.showapi.com/184-4", "602469", "cff386bf98294645ab8c1acf2fc2c385")  # 密钥和密码用自己的

        r.addFilePara("image", picture_name2)
        r.addBodyPara("typeId", "34")
        r.addBodyPara("convert_to_jpg", "0")
        r.addBodyPara("needMorePrecise", "0")
        res = r.post()
        print(res.text)  # 返回信息
        # 取showapi_res_body中的result
        body = res.json()['showapi_res_body']
        print(body)

        # 取出验证码
        print(body['Result'])
    # 完成验证码识别，取出后，进去输入

        # 输入验证码
        self.driver.find_element_by_name('captcha').clear()
        self.driver.find_element_by_name('captcha').send_keys(captcha)
        # 点击登录
        self.driver.find_element_by_xpath('//*[@id="form"]/div[4]/div/button').click()  #千万别忘记click
        # 等待标题
        WebDriverWait(self.driver, 5).until(EC.title_is(expected))

        sleep(3)
        #验证 因为没有错误弹窗  所以用到了title断言判断
        # assert self.driver.title == expected
        self.assertEqual(self.driver.title,expected)
        sleep(2)

if __name__ == '__main__':
    unittest.main()

运行结果：登录失败运行成功；登录成功的由于没充钱，失败验证码的时候运行失败了

 

“永不放弃，总有希望在前面等待！”送给自己，也送给正在阅读文章的博友们～