Django单元测试

本文主要介绍基于Django框架开发的web程序进行单元测试。

因为使用django程序的view函数的参数一般为request形式，那么如何进行单元测试呢？我们分为以下两部分介绍：

实现原理
简单示例

实现原理

关于测试，我们可以使用Django的库文件TestCase, Client 即可。

from django.test import TestCase, Client

TestCase类

在Django的单元测试中，我们使用使用python的unittest模块，django.test.TestCase继承于python的unittest.TestCase。
我们可以通过下面代码执行目录下所有的以“test”开头的文件。

python manage.py test

这里的TestCase类相当于是建立一个虚拟的运行环境，会建立一个新的数据库（以“test_”开头），基于新建立的数据库运行test程序。
同时每一个测试用例，均会有新的数据库。所以每个测试用例的开始，如果需要数据，则需要重新进行建库、插入、登录等操作。

Client类

因为很多view函数接收的都是requests，那么我们就需要去模拟网页的点击。这里就需要用到Client来发送post、get请求。

from django.test import Client
c = Client()
# post参数如下，第一个是url，第二个是传递的参数
response = c.post('/login/', {'username': 'john', 'password': 'smith'})
print(response.content)
c.get('/customers/details/', {'name': 'fred', 'age': 7})
print(resp.content)

上面简单的实现了模拟post和get请求。

1.get请求的参数（{'name': 'fred', 'age': 7}）是缺省的，可以不写
2.post的请求参数是必须的。否则会报错：django.http.request.RawPostDataException这与django处理post的方式有关。
详情请见：https://github.com/encode/django-rest-framework/issues/2774

当然也可以对Client进行一些修改，比如：模拟网页端操作：

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.https://www.jianshu.com/p/5533c866453a