如何写出优美的代码 - 如何测试 Django 应用

优质

小牛编辑

133浏览

2023-12-01

Django 的启动互相之间的依赖严重，很多参数和依赖都需要在运行的时候导入，导致大部分文件都不能单独执行。
不过 Django 的社区非常活跃，对于知名的测试框架都有进行封装，如： django.test、django_nose 等等，
以配合自身的测试命令使用。

doctest

在 Flask 中测试一个文件的 doctest 只需要运行：python filename.py，然而这在 Django 中行不通。
在 Django 中依赖自身的 test 命令：python manage.py test[ app_name]，其中 app_name 若为空
默认测试所有应用。在 1.6 及以后版本中，
需要首先在 settings.py 中指定 TEST_RUNNER：

INSTALLED_APPS = (
    ...
    'django_nose',
)
TEST_RUNNER = 'django_nose.NoseTestSuiteRunner'
NOSE_ARGS = ['--with-doctest']

TestCase

Django 的 TestCase 类是 unittest.TestCase 的子类，使用起来非常相似。

Fixture

Fixture 是 unittest 提供读取测试数据的一种方式，在 Django 的 TestCase 中也可以直接使用，使用前需要导出数据：

python manage.py dumpdata --format=yaml --indent=4 > fixtures_dir/filename.yaml

支持的数据格式包括 YAML、JSON 等等，YAML 可读性较高，不过需要安装额外的依赖。

配合 testserver 命令启动：

python manage.py testserver fixtures_dir/filename.yaml

在测试用例中指定：

from django.test import TestCase
from django.contrib.auth import authenticate
class LoginTest(TestCase):
    fixtures = ['mysite.yaml']
    def setUp(self):
        # 导入 fixture 中用户数据，省去创建用户的流程，也免去了清除用户数据的流程。
    def test_has_user(self):
        # 如果已导入 fixture 中数据，则可以使用其中的账号登录。
        self.assertIsNotNone(authenticate(username='windrunner', password='password'))

Client

Client 提供了用户代理的模拟，其使用类似于 requests 库，不过使用前需要先初始化：client = Client()，
Client 默认会提供 CSRF 认证，如果需要手动验证 CSRF，需要这样初始化：
csrf_enabled_client = Client(enforce_csrf_checks=True)。

import unittest
from django.test.client import Client
class PageTest(unittest.TestCase):
    def setUp(self):
        self.client = Client()
    def test_home(self):
        res = self.client.get('/')
        self.assertEqual(200, res.status_code)
    def test_login(self):
        """普通测试。client 实例会自动解决 csrf 问题。"""
        res = self.client.get('/login/')
        self.assertEqual(200, res.status_code)
        self.assertIn('Username', res.content)
        res_post = self.client.post('/login/', {'username': 'windrunner', 'password': 'password', })
        self.assertEqual(200, res_post.status_code)
        self.assertIn('windrunner', res_post.content)
    def test_login_csrf(self):
        """强制 csrf 检查"""
        self.client = Client(enforce_csrf_checks=True)          # 使用检查 CSRF 的 Client 示例代替默认实例
        res = self.client.get('/login/')
        csrf_token = '%s' % res.context['csrf_token']             # 获取 csrf_token
        res_fail = self.client.post('/login/', {'user': 'windrunner', 'pass': 'password', })
        self.assertEqual(403, res_fail.status_code)             # 没有处理 CSRF token 会返回 403 错误代码
        res_csrf = self.client.post('/login/', {'user': 'windrunner', 'pass': 'password', 'csrfmiddlewaretoken': csrf_token, })
        self.assertIn('windrunner', res_csrf.content)
    def test_logout(self):
        res = self.client.post('/logout/')
        self.assertEqual(302, res.status_code)

testserver

testserver 是 Django 提供的启动测试服务器的方法，会创建一个测试数据库来替代默认数据库，
通常会在启动时导入相应 fixture。命令如下：

python manage.py testserver --addrport 7000 fixture1 fixture2

Selenium

因为 Selenium 是控制浏览器测试 web 服务，因此并不会受到 Django 的干扰，这里有一段示例代码：

import unittest
from selenium import webdriver
from django.contrib.auth import get_user_model, authenticate
class LoginTest(unittest.TestCase):
    def setUp(self):
        self.browser = webdriver.Firefox()          # 初始化浏览器，也可以选择 Chrome 或者 PhanatomJS
    def tearDown(self):
        self.browser.quit()                         # 测试结束后关闭浏览器
    def _login(self):
        # 这个方法没有以 ``test`` 开始，因此并不会单独被执行。
        self.browser.get('http://localhost:8000/login')         # 发送 GET 请求并打开页面
        # 使用浏览器的选择权选中 HTML 元素，并发送浏览器事件，复杂的元素选择可以借助 XPath
        self.browser.find_element_by_id('username').send_keys('windrunner')
        self.browser.find_element_by_id('password').send_keys('password')
        self.browser.find_element_by_id('submit').click()   # 触发点击事件
    def test_login(self):
        self._login()
        self.assertIn('windrunner', self.browser.page_source)   # 断言登录后的页面内容
    def test_logout(self):
        self._login()
        self.assertIn('windrunner', self.browser.page_source)
        self.browser.get('http://localhost:8000/logout')
        self.assertIn('nobody', self.browser.page_source)
        self.assertNotIn('windrunner', self.browser.page_source)