其次,定義Blueprint(藍圖)文件 init.py
#!/usr/bin/env python # -*- coding: utf-8 -*- from flask import Blueprint from flask_restful import Api from testpost import TestPost testPostb = Blueprint('testPostb', name) api = Api(testPostb) api.add_resource(TestPost, '/<string:PostData>/postMeth')
然后,編寫測試腳本testPostM.py
#!/usr/bin/env python # -*- coding: utf-8 -*- import unittest import json from secautoApp.api.testPostMeth import api from flask import url_for from run import app from secautoApp.api.testPostMeth import TestPost headers = {'Accept': 'application/json', 'Content-Type': 'application/json' } class APITestCase(unittest.TestCase): def setUp(self): # self.app = create_app(os.getenv("SECAUTOCFG") or 'default') self.app = app # self.app_context = self.app.app_context() # self.app_context.push() self.client = self.app.test_client() # # def tearDown(self): # self.app_context.pop() def test_post(self): # with app.test_request_context(): response = self.client.get(api.url_for(TestPost, PostData='adb', ddos='123')) self.assertTrue(response.status_code == 200) response = self.client.get(url_for('testPostb.testpost', PostData='adb', ddos='123')) self.assertTrue(response.status_code == 200) self.assertTrue(json.loads(response.data)['PostData'] =='adb') response = self.client.post(url_for('testPostb.testpost', PostData='adb'), headers=headers, data=json.dumps({"ddos": '123'})) print json.loads(response.data) self.assertTrue(response.status_code == 200) response = self.client.put(url_for('testPostb.testpost', PostData='adb', ddos='123')) self.assertTrue(json.loads(response.data) == 'helloadb123') response = self.client.put(url_for('testPostb.testpost', PostData='adb')) print json.loads(response.data)['PostData'] self.assertTrue(response.status_code == 200)
ps:調用的api url 主要用的是flask_restful 的api.url_for,或者是flask的url_for,下面我來說下這2種方法的具體使用
flask_restful 的api.url_for說明
api.url_for(TestPost,PostData='adb'),這里的TestPost指的是restful api接口文件中定義的class,因為我們在api藍圖中,已經通過api.add_resource(TestPost, ‘//postMeth')添加類的方式定義過
flask的url_for的使用說明
url_for(‘testPostb.testpost', PostData='adb', ddos='123'),'testPostb.testpost'這個字符串中
testPostb指的是藍圖的名稱,也就是testPostb = Blueprint(‘testPostb', name)中Blueprint(‘testPostb',name)中的testPostb。
testpost指的是藍圖下endpoit的端點名稱,flask_restful中,指的是api.add_resource(TestPost, ‘//postMeth')中 類名TestPost的小寫
啟動測試腳本:
C:\secauto3>python run.py test test_post (testPostM.APITestCase) ... ok ---------------------------------------------------------------------- Ran 1 test in 0.056s OK
小總結:url_for的傳值和request中的取值是有對應關系的,最后就是flask_restful中端點的方式,一定要是api.add_resource中類名的小寫。
相信看了本文案例你已經掌握了方法,更多精彩請關注Gxl網其它相關文章!
推薦閱讀:
python使用unittest測試接口步奏詳解
Python怎么統計字母出現的次數
聲明:本網頁內容旨在傳播知識,若有侵權等問題請及時與本網聯系,我們將在第一時間刪除處理。TEL:177 7030 7066 E-MAIL:11247931@qq.com