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編輯器打開,查看結果也是非常的直觀:
到此這篇關於django寫單元測試的方法的文章就介紹到這瞭,更多相關django單元測試內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!
推薦閱讀:
- None Found