assertEqual     如果两个值相等，则passassertNotEqual  如果两个值不相等，则passassertTrue      判断bool值为True，则passassertFalse     判断bool值为False，则passassertIsNone    不存在，则passassertIsNotNone 存在，则pass

简单的单元测试案例

这里给大家写了一个简单的登录案例，大致情况是测试登录过程中有无填写账号密码、账号密码有无错误，具体大家可以看代码。


# coding:utf-8

from flask import Flask, request, jsonify

app = Flask(__name__)

@app.route("/login", methods=["POST"])
def login():
    # 接收参数
    username = request.form.get("username")
    password = request.form.get("password")

    # 参数判断
    if not all([username, password]):
        resp = {
            "code":1,
            "message":"invalid params"
        }
        return jsonify(resp)

    if username == "admin" and password =="javaandpython":
        resp = {
            "code": 0,
            "message": "login success"
        }

        return jsonify(resp)
    else:
        resp = {
            "code": 2,
            "message": "login failed"
        }

        return jsonify(resp)

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

test.py


import unittest
from login import app
import json

class LoginTest(unittest.TestCase):
    '''构造单元测试'''
    def test_user_pass_isempty(self):
        '''测试用户和密码是否完整'''
        # 创建进行web请求的客户端，使用flask提供的
        client = app.test_client()

        # 利用client客户端模拟发送web请求
        ret = client.post("/login", data={}) # 用户名和密码都为空

        # ret是视图返回的响应对象，data属性是响应体的数据
        resp = ret.data
        # 解析json
        resp = json.loads(resp)

        # 开始进行断言
        self.assertIn("code", resp)
        self.assertEqual(resp["code"], 1)

if __name__ == '__main__':
    # 直接通过下面方法进行测试
    unittest.main()

我们可以直接通过运行test.py文件来测试login方法是否编写正确。

小白学Flask第十五天| 重要环节---单元测试！

例如上图就说明整个测试是成功的。

大家可能注意到，我们上面的代码只是测试了用户名和密码都为空的情况，作为单元测试，我们需要把其他几种情况都列举出来，如密码为空，用户名不为空。这样才能够体现出测试的完整性。

在测试类中，有两个固定的方法：


import unittest
class TestClass(unittest.TestCase):

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

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

还是按照上面登录的例子来看


import unittest
from login import app
import json

class LoginTest(unittest.TestCase):
    '''构造单元测试'''
    def setUp(self):
        # 设置flask工作在测试模式下
        app.testing = True

        # 创建进行web请求的客户端，使用flask提供的
        self. client = app.test_client()

    def test_user_pass_isempty(self):
        '''测试用户和密码是否完整'''
        # 利用client客户端模拟发送web请求
        ret = self.client.post("/login", data={})

        # ret是视图返回的响应对象，data属性是响应体的数据
        resp = ret.data

        # 解析json
        resp = json.loads(resp)

        # 开始进行断言
        self.assertIn("code", resp)
        self.assertEqual(resp["code"], 1)

    def test_user_pass_isright(self):
        '''测试用户名或密码是否正确'''
        # 利用client客户端模拟发送web请求
        ret = self.client.post("/login", data={"username" : "admin"})
        # ret是视图返回的响应对象，data属性是响应体的数据
        resp = ret.data

        # 解析json
        resp = json.loads(resp)

        # 开始进行断言
        self.assertIn("code", resp)
        self.assertEqual(resp["code"], 1)

if __name__ == '__main__':
    # 直接通过下面方法进行测试
    unittest.main()

可以从上面的代码中发现，setUp里面的所写的是多个测试函数中可能重复出现的代码。

码农公寓

相关文章