Web项目如何做单元测试

你可能会用单元测试框架，python的unittest、pytest，Java的Junit、testNG等。

那么你会做单元测试么！当然了，这有什么难的？

test_demo.py

def inc(x):
return x + 1

def test_answer():
assert inc(3) == 4
inc() 是定义的一个被测函数，test_anserver() 用于测试上面的一段代码。

通过pytest运行上面的代码：

pytest test_demo.py
====================== test session starts ======================= platform win32 -- Python 3.7.1, pytest-5.0.1, py-1.8.0, pluggy-0.12.0
rootdir: D:vipcndemo
plugins: cov-2.7.1, forked-1.0.2, html-1.20.0, metadata-1.8.0, ordering-0.6, parallel-0.0.9, rerunfailures-7.0, xdist-1.28.0, seleniumbase-1.23.10
collected 1 item

test_demo.py . [100%]

==================== 1 passed in 0.08 seconds ====================
单元测试不就是这么单嘛！

那么Web项目中的单元测试如何做？

我们以Django Web框架为例，它是MTV开发模式。接下来会围绕着这个模式介绍如何做测试。

模型测试
M 指models，用于定义ORM，即对象关系映射，是通过使用描述对象和数据库之间映射的元数据，将面向对象语言程序中的对象自动持久化到关系数据库中。
models.py 中的代码是这样的：

from django.db import models

class Question(models.Model):
question_text = models.CharField(max_length=200)
pub_date = models.DateTimeField(auto_now=True)

这里定义了两个类，这两个类即没有入参，也没有return返回值。如何测试呢？

测试代码如下：

from django.test import TestCase
from myapp.models import Question

class QuestionTestCase(TestCase):

def setUp(self):
    Question.objects.create(id=1, question_text="你会做单元测试么？")

def test_question(self):
    """查询id=1的问题"""
    question = Question.objects.get(id=1)
    self.assertEqual(question.question_text, '你会做单元测试么？')

不知道你是否看懂了这段代码，django模型我们可以看作是数据库表，那么对于表的操作就是增删改查，这里先创建一条数据，再查询出这条数据，然后判断其字段是否正确。

参考：https://docs.djangoproject.com/en/2.2/topics/testing/overview/

视图测试
V 指views，用于接收前端发来的请求，可能需要调用数据库，把对应的数据处理之后，和HTML页面一同返回给前端。
views.py 代码如下：

from django.shortcuts import render
from .models import Question

def index(request):
latest_question_list = Question.objects.order_by('-pub_date')[:5]
context = {'latest_question_list': latest_question_list}
return render(request, 'polls/index.html', context)
index() 视图函数确实有入参，request包含的是客户端信息，比如请求的方法，请求的host，请求头Header等，这些客户端数据如何构造？ return返回的是HTML页面，以及查询数据库的数据，如何针对这些数据写断言呢？

测试代码如下：

from django.test import TestCase
from myapp.models import Question

class IndexTestCase(TestCase):

def setUp(self):
    Question.objects.create(id=1, question_text="你会做单元测试么？")

def test_index(self):
    """测试index视图"""
    response = self.client.get("/index")
    self.assertEqual(response.status_code, 200)
    self.assertTemplateUsed(response, "polls/index.html")

这里假定当浏览器访问 http://127.0.0.1:8000/index 时调用到index视图，返问题列表页面。

self.client.get() 可以模拟客户端浏览器发送 request GET 请求。拿到服务端的response,判断状态码是否为 200。 self.assertTemplateUsed() 断言返回的页面是否正确。

参考：https://docs.djangoproject.com/en/2.2/topics/testing/tools/

模板测试
T 指Teamplate，主要是HTML页面。用户在浏览器中输入URL地址，最终会得到一个HTML页面。
index.html代码如下：

{% if latest_question_list %}

{{ question.question_text }}

{% else %}

No polls are available.

{% endif %}
这里面的代码连个方法都没有，更别提入参和返回值了，请问怎么对HTML代码进行测试？

我们确实没有办法直接对HTML代码进行测试。不过，可以借助Selenium来做UI自动化测试，从而保证页面的正确性。

from django.contrib.staticfiles.testing import StaticLiveServerTestCase
from selenium import webdriver

class MySeleniumTests(StaticLiveServerTestCase):

@classmethod
def setUpClass(cls):
    super().setUpClass()
    cls.selenium = webdriver.Chrome()
    cls.selenium.implicitly_wait(10)

@classmethod
def tearDownClass(cls):
    cls.selenium.quit()
    super().tearDownClass()

def test_index_page(self):
    self.selenium.get('%s%s' % (self.live_server_url, '/index'))
    question_list = self.selenium.find_elements_by_name("q")
    for q in question_list:
        print(q.text)

Django封装了StaticLiveServerTestCase，让你在运行UI测试时会自动启动Django服务。所以，你可以直接使用self.live_server_url 访问django启动的服务地址。

参考：https://docs.djangoproject.com/en/2.2/topics/testing/tools/

本文是否刷新了你对项目中单元测试的理解？那么问题来了，我上面的写的哪部分代码属于单元测试？