代码改变世界

Web项目如何做单元测试

2019-08-07 22:58  虫师  阅读(10151)  评论(2编辑  收藏  举报

你可能会用单元测试框架,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:\vipcn\demo
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 %}
    <ul>
    {% for question in latest_question_list %}
        <li><a name="q" href="/polls/{{ question.id }}/">{{ question.question_text }}</a></li>
    {% endfor %}
    </ul>
{% else %}
    <p>No polls are available.</p>
{% 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/

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

Web Page Counters
Computer Desks