python unittest實(shí)現(xiàn)api自動(dòng)化測(cè)試

更新時(shí)間：2018年04月04日 14:31:41 作者：老鼠程序員

這篇文章主要為大家詳細(xì)介紹了python unittest實(shí)現(xiàn)api自動(dòng)化測(cè)試的方法，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下

項(xiàng)目測(cè)試對(duì)于一個(gè)項(xiàng)目的重要性，大家應(yīng)該都知道吧，寫python的朋友，應(yīng)該都寫過自動(dòng)化測(cè)試腳本。
最近正好負(fù)責(zé)公司項(xiàng)目中的api測(cè)試，下面寫了一個(gè)簡單的例子，對(duì)API 測(cè)試進(jìn)行梳理。

首先，編寫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:對(duì)于request的取值，我這里定義了常用的三種方法：

post方法：request.get_json(),在調(diào)用API時(shí)，傳值是json方式
get和put方法:request.args 或者reqparse.RequestParser(),調(diào)用API時(shí)，傳的是字符串

其次，定義Blueprint（藍(lán)圖）文件 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')

然后,編寫測(cè)試腳本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:調(diào)用的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，因?yàn)槲覀冊(cè)赼pi藍(lán)圖中，已經(jīng)通過api.add_resource(TestPost, ‘//postMeth')添加類的方式定義過

flask的url_for的使用說明

url_for(‘testPostb.testpost', PostData='adb', ddos='123'),'testPostb.testpost'這個(gè)字符串中
testPostb指的是藍(lán)圖的名稱，也就是testPostb = Blueprint(‘testPostb', name)中Blueprint(‘testPostb',name)中的testPostb。
testpost指的是藍(lán)圖下endpoit的端點(diǎn)名稱,flask_restful中，指的是api.add_resource(TestPost, ‘//postMeth')中類名TestPost的小寫

啟動(dòng)測(cè)試腳本：

C:\secauto3>python run.py test
test_post (testPostM.APITestCase) ... ok

----------------------------------------------------------------------
Ran 1 test in 0.056s

OK

小總結(jié)：url_for的傳值和request中的取值是有對(duì)應(yīng)關(guān)系的，最后就是flask_restful中端點(diǎn)的方式，一定要是api.add_resource中類名的小寫。

領(lǐng)取干貨：零基礎(chǔ)入門學(xué)習(xí)python視頻教程

以上就是本文的全部內(nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: