辅助测试工具
# -*- coding: utf-8 -*-
# filename: objectdict.py
class ObjectDict(dict):
def __getattr__(self, key):
if key in self:
return self[key]
return None
def __setattr__(self, key, value):
self[key] = value
View decorators 测试
View decorators 即视图函数装饰器，常用来进行权限控制等等。如：
# -*- coding: utf-8 -*-
# filename: decorators.py
from functools import wraps
from django.http import HttpResponseForbidden
def active_required(func):
@wraps(func)
def wrapper(request, *args, **kwargs):
if not request.user.is_active:
return HttpResponseForbidden("账户被禁言！")
return func(request, *args, **kwargs)
return wrapper
上面的装饰器对当前登录用户的 active 状态进行了限制，如何写 test case 呢？方法如下：
# -*- coding: utf-8 -*-
from django.test import TestCase
from django.test.client import RequestFactory
from django.http import HttpResponse
from objectdict import ObjectDict
from decorators import active_required
def view(request):
return HttpResponse('Hello World!')
class DecoratorsTest(TestCase):
def setUp(self):
self.factory = RequestFactory()
self.request = self.factory.get('/')
def test_active_required(self):
# fake active user
self.request.user = ObjectDict(is_active=True)
decorated = active_required(view)
response = decorated(self.request)
self.assertEqual(response.status_code, 200)
# fake inactive user
self.request.user = ObjectDict(is_active=False)
response = decorated(self.request)
self.assertEqual(response.status_code, 403)
如上代码，使用 django.test.client.RequestFactory 构造了一个 factory，调用 factory 的 get 方法获得了一个模拟的 request 对象。前面定义了一个测试视图函数 view ，通过调用 view(self.request) 即可模拟视图函数的调用过程，而通过 active_required(view)(request) 即模拟了加了装饰器之后的视图函数的调用过程。调用后返回模拟的 HttpResponse 对象，可以通过 response 的 status_code、content 等属性测试结果是否和预期一致。
Middleware 测试
假设我们有如下 middleware 定义：
# -*- coding: utf-8 -*-
# filename: middlewares.py
from django.http import HttpResponseRedirect
class HelloMiddleware(object):
def process_request(self, request):
msg = request.GET.get('msg', None)
if msg:
print(msg)
else:
return HttpResponseRedirect(request.get_full_path() + '?msg=Hello')
则 Test case 写法：
# -*- coding: utf-8 -*-
from django.test import TestCase
from django.test.client import RequestFactory
from django.http import HttpResponseRedirect
from middlewares import HelloMiddleware
class HelloMiddlewareTest(TestCase):
def setUp(self):
self.middleware = HelloMiddleware()
self.factory = RequestFactory()
def test_with_msg(self):
request = self.factory.get('/?msg=World')
self.assertEqual(self.middleware.process_request(request), None)
def test_without_msg(self):
request = self.factory.get('/')
self.assertIsInstance(self.middleware.process_request(request), HttpResponseRedirect)
通过调用 middleware 的 process_* 方法即可模拟 middleware 的处理过程，然后根据返回值进行测试。
Template tags 测试
假设有如下 template tag 定义：
# -*- coding: utf-8 -*-
# filename: helloworld.py
from django.template import Library
register = Library()
@register.simple_tag()
def hello():
return 'hello world!'
@register.simple_tag(takes_context=True)
def world(context, msg):
return 'hello %s' % msg
则测试方法为：
# -*- coding: utf-8 -*-
from django.test import TestCase
from django.template import Template, Context
class TemplateTagsTest(TestCase):
def test_hello(self):
out = Template(
'{% load helloword %}'
'{% hello %}'
).render(Context({}))
self.assertEqual('hello world!', out)
def test_world(self):
out = Template(
'{% load helloword %}'
'{% world msg %}'
).render(Context({
'msg': 'world!',
}))
self.assertEqual('hello world!', out)
如上，通过构造 Template 对象并调用 render 方法来获得渲染结果并予以测试。如果 template tag 函数中需要用到 request 对象，可将上面的 Context 换成 RequestContext（先用 RequestFactory 构造 request）。
Template filter 测试
Template filter 其实就是个普通的 Python 函数，可以直接通过调用测试返回值即可。当然，如果你不嫌麻烦，也可以用和测试 template tags 一样的方法构造 Template 对象后 render 测试渲染结果和预期是否一致。
Management Commands 测试
Management commands 是提供给 manage.py 进行命令行调用的，在代码中可通过 django.core.management.call_command 函数调用相应的命令，然后测试预期的产生的影响。
这里的问题主要在于 management commands 中可能使用到了 input/raw_input/getpass.getpass 等需要用户交互输入的函数导致不好写 testcase.
解决方法也简单，利用和 gevent.monkey 类似的 Python 模块的 monkey patch 方法即可。在代码中不直接调用 input/raw_input/getpass.getpass 而是先在其所在模块中定义它们的 wrapper 后通过调用这些 wrapper 实现功能，此后在 test case 中就可以先将这些 wrapper monkey patch 后再调用 call_command.
View Function 测试
直接通过 TestCase 内置的 client 模拟请求测试即可，不多说。
MongoDB 作为数据库的测试
如果使用 MongoDB 作为 Django 的后端数据库，在 TestCase 中要自动建立测试数据库的话，可继承 TestCase 写个自定义的新 TestCase 解决：
# -*- coding: utf-8 -*-
from django.conf import settings
from django.test import TestCase as DjangoTestCase
from pymongo import MongoClient
class TestCase(DjangoTestCase):
__dbname__ = 'test'
def _pre_setup(self):
super(TestCase, self)._pre_setup()
client = MongoClient(settings.MONGODB_HOST, settings.MONGODB_PORT)
self.db = client[self.__dbname__]
def _post_teardown(self):
super(TestCase, self)._post_teardown()
for collection in self.db.collection_names():
if collection == 'system.indexes':
continue
self.db.drop_collection(collection)
然后需要用到测试数据库的 test case 类从这个自定义的 TestCase 继承即可。
