Flask介绍Flask中如何使用单元测试并检测登陆

Posted 2021-08-31 Vax_Loves_1314

为什么要测试？

Web程序开发过程一般包括以下几个阶段：[需求分析，设计阶段，实现阶段，测试阶段]。其中测试阶段通过人工或自动来运行测试某个系统的功能。目的是检验其是否满足需求，并得出特定的结果，以达到弄清楚预期结果和实际结果之间的差别的最终目的。

测试的分类

测试从软件开发过程可以分为：

• 单元测试
  • 对单独的代码块(例如函数)分别进行测试,以保证它们的正确性
• 集成测试
  • 对大量的程序单元的协同工作情况做测试
• 系统测试
  • 同时对整个系统的正确性进行检查,而不是针对独立的片段

在众多的测试中，与程序开发人员最密切的就是单元测试，因为单元测试是由开发人员进行的，而其他测试都由专业的测试人员来完成。所以我们主要学习单元测试。

什么是单元测试？

程序开发过程中，写代码是为了实现需求。当我们的代码通过了编译，只是说明它的语法正确，功能能否实现则不能保证。 因此，当我们的某些功能代码完成后，为了检验其是否满足程序的需求。可以通过编写测试代码，模拟程序运行的过程，检验功能代码是否符合预期。

单元测试就是开发者编写一小段代码，检验目标代码的功能是否符合预期。通常情况下，单元测试主要面向一些功能单一的模块进行。

举个例子：一部手机有许多零部件组成，在正式组装一部手机前，手机内部的各个零部件，CPU、内存、电池、摄像头等，都要进行测试，这就是单元测试。

在Web开发过程中，单元测试实际上就是一些“断言”（assert）代码。

断言就是判断一个函数或对象的一个方法所产生的结果是否符合你期望的那个结果。 python中assert断言是声明布尔值为真的判定，如果表达式为假会发生异常。单元测试中，一般使用assert来断言结果。

断言方法的使用

def num(num1, num2):
# 两个数都必须为整数
# num2不能为0

# 断言: num1是整形, 不是的话, 就抛出错误.
# 错误的类型是AssertionError

# assert 判断条件 如果出错了的提示信息
assert isinstance(num1, int), 'num1 must is int'
assert isinstance(num2, int), 'num2 must is int'
assert num2 != 0, 'num2 not 0'

print num1 / num2

# 这里可以随便输入内容测试, 查看断言结果
num(1, 1)

断言语句类似于：

if not expression:    
    raise AssertionError
 AssertionError

常用的断言方法：

assertEqual     如果两个值相等，则pass
assertNotEqual  如果两个值不相等，则pass
assertTrue      判断bool值为True，则pass
assertFalse     判断bool值为False，则pass
assertIsNone    不存在，则pass
assertIsNotNone 存在，则pass

单元测试的基本写法

首先，定义一个类，继承自unittest.TestCase

import unittest
class TestClass(unitest.TestCase):
    pass

其次，在测试类中，定义两个测试方法

import unittest
class TestClass(unittest.TestCase):

    #该方法会首先执行，方法名为固定写法
    def setUp(self):
        pass

    #该方法会在测试代码执行完后执行，方法名为固定写法
    def tearDown(self):
        pass

最后，在测试类中，编写测试代码

import unittest
class TestClass(unittest.TestCase):

    #该方法会首先执行，相当于做测试前的准备工作
    def setUp(self):
        pass

    #该方法会在测试代码执行完后执行，相当于做测试后的扫尾工作
    def tearDown(self):
        pass
    #测试代码
    def test_app_exists(self):
        pass

示例:检测登录的单元测试

编写登录接口

# -*- coding:utf-8 -*-

from flask import Flask, request, jsonify

app = Flask(__name__)


# 当参数不全的话，返回errcode=-2
# 当登录名和密码错误的时候，返回 errcode = -1
# 当登录成功之后，返回 errcode = 0
@app.route('/login', methods=['POST'])
def index():

    username    = request.form.get('username')
    password = request.form.get('password')

    # 判断参数是否为空
    if not all([username, password]):
        result = {
            "errcode": -2,
            "errmsg": "params error"
        }
        return jsonify(result)
    # a = 1 / 0
    # 如果账号密码正确
    # 判断账号密码是否正确
    if username == 'itheima' and password == 'python':
        result = {
            "errcode": 0,
            "errmsg": "success"
        }
        return jsonify(result)
    else:
        result = {
            "errcode": -1,
            "errmsg": "wrong username or password"
        }
        return jsonify(result)


if __name__ == '__main__':
    app.run(debug=True)

编写一个测试用例

import unittest
from demo5_login_assert import app
import json


class LoginTestCase(unittest.TestCase):

    def test_empty_username_password(self):
        client = app.test_client()
        response = client.post('/login', data={"username": 'aaaa'})
        response_data = response.data
        response_json = json.loads(response_data)

        self.assertIn('errcode', response_json, 'None errcode')
        errcode = response_json['errcode']
        print errcode
        self.assertEqual(errcode, -2, 'error code is %s' % errcode)

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

数据库测试

#coding=utf-8
import unittest
from author_book import *

#自定义测试类，setUp方法和tearDown方法会分别在测试前后执行。以test_开头的函数就是具体的测试代码。
class DatabaseTestCase(unittest.TestCase):
    def setUp(self):
        app.config['TESTING'] = True
        app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:mysql@localhost/test0'
        self.app = app
        db.create_all()

    def tearDown(self):
        db.session.remove()
        db.drop_all()

    #测试代码
    def test_append_data(self):
        au = Author(name='itcast')
        bk = Book(info='python')
        db.session.add_all([au,bk])
        db.session.commit()
        author = Author.query.filter_by(name='itcast').first()
        book = Book.query.filter_by(info='python').first()
        #断言数据存在
        self.assertIsNotNone(author)
        self.assertIsNotNone(book)