python unittest做api自動化測試
阿新 • • 發佈:2018-12-30
專案測試對於一個專案的重要性,大家應該都知道吧。
寫python的朋友,應該都寫過自動化測試指令碼。
最近正好負責公司專案中的api測試,
下面寫了一個簡單的例子,對API 測試進行梳理。
首先,編寫restful api介面檔案 testpost.py,包含了get,post,put方法
#!/usr/bin/env python
# -*- coding: utf-8 -*-
from flask import request
from flask_restful import Resource
from flask_restful import reqparse
test_praser = reqparse.RequestParser()
test_praser.add_argument('ddos' )
class TestPost(Resource):
def post(self, PostData):
data = request.get_json()
user = User('wangjing')
if data['ddos']:
return {'hello': 'uese', "PostData": PostData, 'ddos': 'data[\'ddos\']'}
return {'hello': 'uese', "PostData": PostData}
def get (self, PostData):
data = request.args
if data and data['ddos']:
return "hello" + PostData + data['ddos'], 200
return {'hello': 'uese', "PostData": PostData}
def put(self, PostData):
data = test_praser.parse_args()
if data and data['ddos']:
return "hello" + PostData + data['ddos'], 200
return {'hello': 'uese', "PostData": PostData}
ps:對於request的取值,我這裡定義了常用的三種方法:
post方法:request.get_json(),在呼叫API時,傳值是json方式
get和put方法:request.args 或者reqparse.RequestParser(),呼叫API時,傳的是字串
其次,定義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中類名的小寫