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-8from django.test import TestCase, Clientimport osimport requestsimport jsonuser = "1234567"host = "http://localhost:8006/app"false = Falsetrue = Truenull = Nonetoken = NonePOST = "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"] = tokendef 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编辑器打开,查看结果也是非常的直观:
到此这篇关于django写单元测试的方法的文章就介绍到这了,更多相关django单元测试内容请搜索 以前的文章或继续浏览下面的相关文章希望大家以后多多支持 !
最后更新于 2021-11-22 08:21:30 并被添加「」标签,已有 位童鞋阅读过。
本站使用「署名 4.0 国际」创作共享协议,可自由转载、引用,但需署名作者且注明文章出处
相关文章
- html5 自定义播放器核心代码
- 【老师推荐】三年级上册同步作文 人教版3年级上小学生语文必读三上同步作文书小学入门起步专项训练作文大全人教部编版荣恒教育
- Python opencv缺陷检测的实现及问题解决
- 张大酥绿豆糕中式糕点心结婚订婚专用喜饼伴手礼茶点小吃零食甜品
- 馋小赖下饭菜卤味脆藕藕片香辣土豆片海带豆角素菜小零食休闲食品