django写单元测试的方法

   从网上找了很多django单元测试的案例，感觉不是很好用，于是自己写了一套测试方法，在测试环境我们只需要传uri 、请求方式、参数即可一键对所有接口进行测试。

一、使用requests模拟Http请求

   假设你执行成功的返回的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 、请求方式、参数(query_string,body或者rest都支持)即可，如下代码可在tests.py文件里执行。

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")

我们只需要一键执行tests.py文件就能看到效果，如下:

二、优化代码将测试结果优雅地输出到md文件里

优化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返回状态: " + status + "\n响应数据:\n" + response_data, file=file)

用md编辑器打开，查看结果也是非常的直观:

来源：https://blog.csdn.net/qq_33036061/article/details/115914998