Django接口自动化测试:一键执行与Markdown报告生成
本文介绍一种基于Django和requests库的接口自动化测试方案。通过封装公共测试方法,仅需传入URI、请求方式及参数即可批量执行接口测试。同时,优化代码以支持将测试结果优雅地输出为带有样式的Markdown文件,实现测试过程的一键化与可视化,提升接口测试效率。
在网络上搜索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_query_string),即可在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)


使用Markdown编辑器打开生成的文件,查看结果非常直观:

至此,关于Django编写单元测试方法的介绍完毕。如需了解更多Django单元测试相关内容,请搜索本站历史文章或继续浏览后续文章,感谢大家的支持!
本站声明
本文内容仅供参考,不代表本站观点。转载或引用请注明出处。如发现内容问题,请通过联系我们页面反馈。













