测试Flask应用

Posted 2022-06-02 suncunxu

测试骨架

import os
import flaskr
import unittest
import tempfile

class FlaskrTestCase(unittest.TestCase):

    def setUp(self):
        self.db_fd, flaskr.app.config[‘DATABASE‘] = tempfile.mkstemp()
        flaskr.app.config[‘TESTING‘] = True
        self.app = flaskr.app.test_client()
        flaskr.init_db()

    def tearDown(self):
        os.close(self.db_fd)
        os.unlink(flaskr.app.config[‘DATABASE‘])

if __name__ == ‘__main__‘:
    unittest.main()
在setUP()方法中的代码创建了一个新的测试客户端并且初始化一个新的数据库,这个函数在每个单独的测试函数运行之前被调用,为了在测试后删除数据库


,我们在tearDown()方法中关闭了这个文件并且从文件系统中删除了它,此外在初始化过程中,TESTING配置标志被激活,这使得处理请求时的
错误捕捉失效,以便获得更好的错误反馈

class FlaskrTestCase(unittest.TestCase):

    def setUp(self):
        self.db_fd, flaskr.app.config[‘DATABASE‘] = tempfile.mkstemp()
        self.app = flaskr.app.test_client()
        flaskr.init_db()

    def tearDown(self):
        os.close(self.db_fd)
        os.unlink(flaskr.DATABASE)

    def test_empty_db(self):
        rv = self.app.get(‘/‘)
        assert ‘No entries here so far‘ in rv.data
测试函数以test开头,这使得unittest能够自动识别要运行的测试方法.
通过使用self.app.get,我们可以发送一个HTTP GET请求到一个给定路径的应用程序.返回的值将会是一个response_class对象,我们
现在可以用data属性检查从应用中返回的值,


登录和注销

应用的大部分功能只允许具有管理员资格的用户访问,所以我们需要一种方法来帮助我们的测试客户端登录和注销,为此我们想登录和注销页面发送一些请求,
这些请求都携带了表单数据,因为登录和注销页面都会重定向,我们将客户端设置为follow_redirects.
添加如下方法:
def login(self,username,password):
　　return self.app.post(‘/login‘,data=dict(username=username,password=password),follow_redirect=true)
def logout(self):
　　return self.app.get(‘/logout‘,follow_redirects=true)

其他测试技巧
test-request-context()方法可以用于配合with声明,用于触发一个临时的请求上下文,通过它可以访问requets,g,和session类的
对象,
app = flask.Flask(__name__)
with app.test_request_context(‘/?name=Peter‘):
　　assert flask.request.path == ‘/‘
　　assert flask.request.args[‘name‘] == ‘Peter‘
值得注意的是如果你使用的是一个测试请求上下文，before_request() 同 after_request() 一样不会自动地被调用。然而 
teardown_request() 函数在测试请求的上下文离开with块确实会被执行。如果你 要 before_request() 函数同样被调用，
你必须自己调用 preprocess_request():
app = flask.Flask(__name__)
with app.test_request_context(‘/?name=Peter‘):
　　app.preprocess_request()
如果希望调用after_request()函数,需要使用process_response()方法,该方法需要传入response对象

伪造资源和上下文
一个非常普遍的模式就是在用于的上下文或者flask.g对象中存储用户的认证信息以及数据库连接,一般的模式是第一次使用的时候把它存入对象,

然后在关闭的时候将其删除.
def get_user():
　　user=  getattr(g,‘user‘,None)
　　if user is None:
　　　　user = fetch_current-user_from_database()
　　　　g.user = user
　　return user