Flask 单元测试 unittest

时间 2019-12-11

原文原文链接

import  unittest 单元测试

app = Flask(__name__)



--------------------------------------------
import unitest
class TestClass(unittest.TestCase):
    
    #该方法会首先执行，至关于作测试前的准备工做
    def setUp(self):
        pass


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

    #测试代码
    def test_app_exists(self):
        pass




def num_div(num1,num2):
    #assert 断言 后面是一个表达式 若是表示返回真，则断言成功，程序可以继续往下执行
    #若是表达式返回假，则断言失败，assert会抛出异常AssertionError 终止程序继续往下执行
    assert isinstance(num1,int)
    assert isinstance(num2,int)
    assert num2 != 0

    print(num1 / num2)

if __name__ == '__main__':
    num_div(100,10)





# -*- coding: utf-8 -*-
# @Time    : 2019/2/18 20:31
# @Author  : yanxiatingyu
# @File    : unittest_study.py
# @Software: PyCharm


import unittest
from flask import Flask
from flask import request
from flask import jsonify

app = Flask(__name__)


@app.route('/')
def index():
    return 'test'


@app.route('/login', methods=["POST"])
def login():
    user_name = request.form.get("user_name")
    user_pwd = request.form.get("user_pwd")
    print(request.form)
    print(request.args)

    resp = ''
    if not all([user_name, user_pwd]):
        resp = {
            "code": 300,
            "message": "invalid params"
        }

        return jsonify(resp)

    if user_pwd == 'a' and user_name == 'a':
        resp = {
            "code": 200,
            "msessage": "login success"
        }
    else:
        resp = {
            "code": 300,
            "message": "login faild"
        }
    return jsonify(resp)


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



import unittest
from unittest_study import app
import json


class LoginTest(unittest.TestCase):
    """构造单元测试案例"""

    def test_empty_user_name_password(self):
        """测试用户名密码不完整的状况"""
        # 建立进行web请求的客户端，使用flask提供的
        client = app.test_client()

        # 利用client客户端模拟发送web请求
        ret = client.post("/login", data={
            "user_name": "a",
            "user_pwd": "a"
        })

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

        # 由于login视图返回的json字符串
        #
        resp = json.loads(resp)

        # 拿到返回值后进行断言测试
        self.assertIn("code", resp)  # code 是否 包含在resp里面

        self.assertEqual(resp["code"], 200)


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


-------------------------------------------------
from flask_mail import Mail,Message

# 配置邮件: 服务器 、 端口 、 传输层安全协议 、 邮箱名 、密码
app.config.update(
    DEBUG=True,
    MAIL_SERVER='smtp.qq.com',
    MAIL_PROT=465,
    MAIL_USE_TLS=True,
    MAIL_USERNAME='799209502@qq.com',
    MAIL_PASSWORD='cyfgpsjasdqibaii'
)
#//MAIL_PASSWORD  受权码
#IMAP/SMTP服务 (什么是 IMAP，它又是如何设置？)  开启QQ邮箱此服务 获取受权码


#因为QQ邮箱不支持非加密的协议，那么使用加密协议，分为两种加密协议，选择其中之一便可
    #1.mail_use_tls    端口号是587

    #2.mail_use_ssl    端口号是465
    #我选择的mail_use_tls  协议，因此MAIL_USE_TLS设置为True,端口号587




mail = Mail(app)

import json
@app.route('/')
def index():

    try:
        # sender 发送方，recipients 接收方列表
        msg = Message("This is a test !", sender='799209502@qq.com', recipients=["changanmingji@126.com"])
        # 邮箱内容
        msg.body = "Flask test mail !"
        # 发送邮件
        mail.send(msg)
        return json.dumps('send success !'), 200, {
            "Content-Type": "application/json"
        }
    except Exception as e:
        print(e)
        return json.dumps("send failed !"),200,{
            "Content-Type":"application/json"
        }

# -*- coding: utf-8 -*-
# @Time    : 2019/2/18 20:41
# @Author  : yanxiatingyu
# @File    : test.py.py
# @Software: PyCharm


import unittest
from unittest_study import app
import json


class LoginTest(unittest.TestCase):
    """构造单元测试案例"""

    def setUp(self):
        """在进行测试以前，先被执行"""
        # 设置flask工做在测试模式下
        # app.config['TESTINg'] = True
        # 或
        app.testing = True# 打开后程序 具体出现什么错误等，告诉你细节问题，不开就不会体现细节问题

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

    def test_empty_user_name_password(self):
        """测试用户名密码不完整的状况"""

        # 利用client客户端模拟发送web请求
        ret = self.client.post("/login", data={
            "user_name": "a",
            "user_pwd": "a"
        })

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

        # 由于login视图返回的json字符串
        #
        resp = json.loads(resp)

        # 拿到返回值后进行断言测试
        self.assertIn("code", resp)  # code 是否 包含在resp里面

        # 是否相等
        self.assertEqual(resp["code"], 200)

    def test_wrong_user_name_password(self):
        """测试用户名或密码错误"""
        ret = self.client.post('/login', data={
            "user_name": 'b',
            "user_pwd": 'b'
        })

        resp = ret.data
        resp = json.loads(resp)

        self.assertIn("code", resp)
        self.assertEqual(resp["code"], 300)


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

# -*- coding: utf-8 -*-
# @Time    : 2019/2/19 12:54
# @Author  : yanxiatingyu
# @File    : app.py
# @Software: PyCharm

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)


class Config(object):
    """配置参数"""
    # sqlalchemy 配置参数
    SQLALCHEMY_DATABASE_URI = "mysql://root:@127.0.0.1:3306/test"
    # 设置sqlalchemy自动跟踪数据库
    SQLALCHEMY_TRACK_MODIFICATIONS = True


app.config.from_object(Config)
db = SQLAlchemy(app)
db.init_app(app)


class User(db.Model):
    """用户表"""
    __tablename__ = "tbl_user"

    id = db.Column(db.Integer, primary_key=True)  # 指明主键
    name = db.Column(db.String(64))
    email = db.Column(db.String(128))

    def __repr__(self):
        return "User object : name = %s " % self.name


if __name__ == '__main__':
    db.drop_all()
    db.create_all()

    user1 = User(**{
        "name": "姓名1",
        "email": "邮箱1"
    })
    user2 = User(**{
        "name": "姓名2",
        "email": "邮箱2"
    })
    user3 = User(name="姓名3", email="邮箱3")
    # db.session.add(user1)
    # or
    db.session.add_all([user1, user2, user3])
    db.session.commit()

# -*- coding: utf-8 -*-
# @Time    : 2019/2/19 13:32
# @Author  : yanxiatingyu
# @File    : test.py
# @Software: 数据库 单元测试


import unittest
from app import User, db, app

class DataBaseTeset(unittest.TestCase):
    """测试用户的数据库操做"""

    def setUp(self):
        app.testing = True
        #    SQLALCHEMY_DATABASE_URI = "mysql://root:@127.0.0.1:3306/test"
        # 测试 数据库的修改
        app.config["SQLALCHEMY_DATABASE_URI"] = "mysql://root:@127.0.0.1:3306/test"

    def test_add_user(self):
        user = User(**{
            "name": "姓名4",
            "email": '邮箱4'
        })
        db.session.add(user)
        db.session.commit()

        res = User.query.filter_by(name="姓名1").first()
        self.assertIsNotNone(res)

    def tearDown(self):
        """在全部的测试执行后 执行，经过用来进行清理操做"""
        db.session.remove()#删除数据，和链接状态
        # db.drop_all() 删除数据库表，慎用直接运行