本文主要介紹基于Django框架開發(fā)的web程序進行單元測試洛心。
因為使用django程序的view函數(shù)的參數(shù)一般為request形式，那么如何進行單元測試呢蚌铜？我們分為以下兩部分介紹：
- 實現(xiàn)原理
- 簡單示例
實現(xiàn)原理
關(guān)于測試秕噪，我們可以使用Django的庫文件TestCase, Client 即可。
from django.test import TestCase, Client
TestCase類
在Django的單元測試中陷虎，我們使用使用python的unittest模塊，django.test.TestCase繼承于python的unittest.TestCase杠袱。
我們可以通過下面代碼執(zhí)行目錄下所有的以“test”開頭的文件尚猿。
python manage.py test
這里的TestCase類相當于是建立一個虛擬的運行環(huán)境，會建立一個新的數(shù)據(jù)庫（以“test_”開頭）楣富，基于新建立的數(shù)據(jù)庫運行test程序凿掂。
同時每一個測試用例，均會有新的數(shù)據(jù)庫纹蝴。所以每個測試用例的開始庄萎，如果需要數(shù)據(jù)，則需要重新進行建庫塘安、插入惨恭、登錄等操作。
Client類
因為很多view函數(shù)接收的都是requests耙旦，那么我們就需要去模擬網(wǎng)頁的點擊。這里就需要用到Client來發(fā)送post、get請求免都。
from django.test import Client
c = Client()
# post參數(shù)如下锉罐，第一個是url，第二個是傳遞的參數(shù)
response = c.post('/login/', {'username': 'john', 'password': 'smith'})
print(response.content)
c.get('/customers/details/', {'name': 'fred', 'age': 7})
print(resp.content)
上面簡單的實現(xiàn)了模擬post和get請求绕娘。
1.get請求的參數(shù)（{'name': 'fred', 'age': 7}）是缺省的脓规，可以不寫
2.post的請求參數(shù)是必須的。否則會報錯：django.http.request.RawPostDataException這與django處理post的方式有關(guān)险领。
詳情請見：https://github.com/encode/django-rest-framework/issues/2774
當然也可以對Client進行一些修改侨舆，比如：模擬網(wǎng)頁端操作：
from django.test import Client
c = Client(HTTP_USER_AGENT='Mozilla/5.0')
Client的類還可以模擬進行用戶的登錄和登出：
from django.test import Client
c = Client()
c.login(username='fred', password='secret')
c.logout()
簡單示例
我們給出一個示例如下，包括urls.py绢陌，test.py挨下，views.py
# urls.py
from django.conf.urls import url
from adminpage.views import *
urlpatterns = [
url(r'^login/?$', AdminLogin.as_view()),
]
# test.py
c = Client(HTTP_USER_AGENT='Mozilla/5.0')
class AuthLoginGet(TestCase):
def test_admin_login_get(self):
# c = Client(HTTP_USER_AGENT='Mozilla/5.0')
resp = c.get('/api/a/login')
mess = json.loads(str(resp.content, encoding="utf-8"))
self.assertEqual(mess['code'], 4)
self.assertEqual(mess['msg'], '')
# views.py
# 繼承基類
class AdminLogin(APIView):
def get(self):
if self.request.user.is_authenticated():
return
else:
raise LoginError('')
參考資料
1.https://docs.djangoproject.com/en/dev/topics/testing/tools/#the-test-client
2.http://www.reibang.com/p/5533c866453a
