從網(wǎng)上找了很多django單元測試的案例，感覺不是很好用，于是自己寫了一套測試方法，在測試環(huán)境我們只需要傳uri 、請求方式、參數(shù)即可一鍵對所有接口進行測試。
一、使用requests模擬Http請求
假設你執(zhí)行成功的返回的json格式如下:
{
"code": 0,
"message": "OK",
"data": {
"first": false,
"token": "3eeeb5bdad75cbe442fd9c6df5373550"
},
"elapsed": 96
}
我寫了一個公共的測試方法test(),def test(method, url, body_data=None, query_string=None, rest_query_string=None): pass, 傳uri 、請求方式、參數(shù)(query_string,body或者rest都支持)即可，如下代碼可在tests.py文件里執(zhí)行。
from django.test import TestCase
# Create your tests here.
# coding:utf-8
from django.test import TestCase, Client
import os
import requests
import json
user = "1234567"
host = "http://localhost:8006/app"
false = False
true = True
null = None
token = None
POST = "POST"
GET = "GET"
DELETE = "DELETE"
PUT = "PUT"
headers = {'content-Type': 'application/json', 'Accept': '*/*'}
login_data = json.dumps({"phone": user,
"pwd": "e10adc3949ba59abbe56e057f20f883e",
"login_type": 0,
"identifier": "",
"role": 0})
login = requests.post(host + "/login", data=login_data, headers=headers)
login_content = eval(login.content.decode("utf-8"))
if login_content["code"] == 0:
print("login 成功")
token = login_content["data"]["token"]
print("token:" + token)
else:
print("login fail")
if not token:
raise Exception("登錄異常")
headers["user-token"] = token
def test(method, url, body_data=None, query_string=None, rest_query_string=None):
if query_string:
url = host + url + (str(rest_query_string) if rest_query_string is not None else "") + "?" + query_string
else:
url = host + url + (str(rest_query_string) if rest_query_string is not None else "")
if method in [POST, DELETE, PUT] and body_data:
body_data = json.dumps(body_data)
response_data = requests.request(method, url, data=body_data, headers=headers)
response_data = response_data.content.decode("utf-8")
if response_data.find("\"code\": 0") != -1:
print(url + " 成功!")
else:
print(url + " 失敗!" + response_data)
test(GET, "/check_token/", rest_query_string=token)
test(GET, "/get/child")
我們只需要一鍵執(zhí)行tests.py文件就能看到效果，如下:
二、優(yōu)化代碼將測試結(jié)果優(yōu)雅地輸出到md文件里
優(yōu)化test方法, 添加樣式，md文件支持讀取樣式。
def test(method, url, body_data=None, query_string=None, rest_query_string=None):
if query_string:
url = host + url + (str(rest_query_string) if rest_query_string is not None else "") + "?" + query_string
else:
url = host + url + (str(rest_query_string) if rest_query_string is not None else "")
if method in [POST, DELETE, PUT] and body_data:
body_data = json.dumps(body_data)
response_data = requests.request(method, url, data=body_data, headers=headers)
response_data = response_data.content.decode("utf-8")
status = "font color='red'>失敗/font>"
if response_data.find("\"code\": 0") != -1:
status = "font color='green'>成功/font>"
print(url + " 成功!")
else:
print(url + " 失敗!")
response_data = "```json\n" + response_data + "\n```"
print("url: " + url + "\n返回狀態(tài): " + status + "\n響應數(shù)據(jù):\n" + response_data, file=file)
用md編輯器打開，查看結(jié)果也是非常的直觀:
到此這篇關(guān)于django寫單元測試的方法的文章就介紹到這了,更多相關(guān)django單元測試內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
您可能感興趣的文章:- Django單元測試中Fixtures的使用方法
- Django單元測試中Fixtures用法詳解
- Django REST framework 單元測試實例解析
- Django單元測試工具test client使用詳解
- Django使用unittest模塊進行單元測試過程解析
- Python Django框架單元測試之文件上傳測試示例
- 對Python的Django框架中的項目進行單元測試的方法
