0x01 單元測試
# 測試文件
import server
import unittest
class ServerTestCase(unittest.TestCase):
def setUp(self):
'test begin'
self.app = server.app.test_client()
def tearDown(self):
'test end'
pass
def test_index(self):
rv = self.app.get('/')
# 從這里看出，傳送時的都是二進制流，想要拿到符合語言的數(shù)據(jù)類型要再轉換
assert '666' in rv.data.decode()
if __name__ == '__main__':
unittest.main()
# server.py
from flask import Flask
app = Flask(__name__)
@app.route('/)
def foo():
data = {
'p1': 'pt1',
'p2': 'pt2'
}
return jsonify(data)
- 測試類中，需要執(zhí)行的測試函數(shù)以 * test *開頭
- 測試時，server 不必執(zhí)行，只要在測試文件中導入 server 模塊即可
- 官方文檔中用到了tempfile.mkstemp() 好像是用來創(chuàng)建臨時文件句柄的，暫時忽略，用到的時候再看。
- 簡單的測試，用 requests 庫封裝的HTTP請求函數(shù)來做也不錯。
0x02 g
# server.py
from flask import Flask
app = Flask(__name__)
@app.route('/)
def foo():
data = {
'p1': 'pt1',
'p2': 'pt2'
}
return jsonify(data)
@app.after_request
def add_something(resp):
resp.headers['Something'] = getattr(g, 'user', 'None')
if __name__ == '__main__':
app.run(debug=True)
# 測試文件
import requests
resp =requests.get('http://127.0.0.1:5000/')
print(resp.headers)
輸出如下：
{
'Server': 'Werkzeug/0.11.10 Python/3.5.2',
'Date': 'Sat, 15 Jul 2017 04:17:17 GMT',
'Content-Type': 'text/html; charset=utf-8',
'Something': '666',
'Content-Length': '5'
}
可見應用成功。
因為 g 對象是為了在同一請求中傳遞數(shù)據(jù)，所以在不同請求中，g 并不是同一個 g；換言之，當前請求完全結束時，當前的 g 也就銷毀了。
