python mysql单元测试_python中的单元测试和数据库测试

登录测试

被测试的代码逻辑

@app.route('/login', methods=['POST'])

def login():

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)

单元测试代码

import json

import unittest

from demo1_login import app

class LoginTest(unittest.TestCase):

"""为登录逻辑编写测试案例"""

def setUp(self):

app.testing = True

self.client = app.test_client()

def test_empty_username_password(self):

"""测试用户名与密码为空的情况[当参数不全的话，返回errcode=-2]"""

response = app.test_client().post('/login', data={})

json_data = response.data

json_dict = json.loads(json_data)

self.assertIn('errcode', json_dict, '数据格式返回错误')

self.assertEqual(json_dict['errcode'], -2, '状态码返回错误')

# TODO 测试用户名为空的情况

# TODO 测试密码为空的情况

def test_error_username_password(self):

"""测试用户名和密码错误的情况[当登录名和密码错误的时候，返回 errcode = -1]"""

response = app.test_client().post('/login', data={"username": "aaaaa", "password": "12343"})

json_data = response.data

json_dict = json.loads(json_data)

self.assertIn('errcode', json_dict, '数据格式返回错误')

self.assertEqual(json_dict['errcode'], -1, '状态码返回错误')

# TODO 测试用户名错误的情况

# TODO 测试密码错误的情况

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='xxxxx')

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)