持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第21天，点击查看活动详情

单元测试对于软件测试人员来说是个“高端”的技能，甚至有一些开发人员也会写单元测试。然而，我 们已经学会了开发Web系统，那么，顺带着来讨论一下Django的单元测试。

unittest单元测试框架

在学习Django单元测试之前，我们先来学习一下 unittest单元测试框架。首先，更新一个误区，单元测 试框架不只适用于程序单元级别的测试。

一般单元测试框架主要完成以下几件事儿：

提供用例组织与执行：当你的测试用例只有几条时，可以不必考虑用例的组织，但是，当测试用例达到 成百上千条时，大量的测试用例堆砌在一起，就产生了扩展性与维护性等问题，需要考虑用例的规范与组织 问题了。单元测试框架就是用来解决这个问题的。

提供丰富的比较方法：不论是功能测试，还是单元测试，在用例执行完成之后都需要将实际结果与进行 预期结果的进行比较（断言），从而断定用例是否执行通过。所以，作为单元测试框架一般也会提供丰富的 断言方法。例如，判断相等／不等、包含／不包含、Trure／False 的断言方法等。

提供丰富的日志：当测试用例执行失败时能抛出清晰的失败原因，当所有用例执行完成后能提供丰富的 执行情况结果信息。例如，总执行时间、失败用例数、成功用例数等。

从这几点来看，单元测试框架可以帮助我们完成任何类型测试的自动化。

单元测试：unittest

HTTP接口自动化测试：unittest＋Requests

Web UI自动化测试：unitte+Selenium

移动自动化测试：unittest＋Appium

假设有如下功能。

"""Describe：实现简单计算器：＋、—、＊、八"""


class Calculator():
    """实现两个数的加、减、乘、除口"""

    def _init_(self, a, b):
        self.a = int(a)
        self.b = int(b)  # 加法

    def add(self):
        return self.a + self.b  # 减法

    def sub(self):
        return self.a - self.b  # 乘法

    def mul(self):
        return self.a * self.b  # 除法

    def div(self):
        return self.a / self.b

首先，需要声明的是，不用单元测试框架一样可以做单元测试。接下看看不用单元测试框架，如何做单 元测试。

from count import Calculator

class CountTest:

    def test_add_4_5(self):
        c=Calculator(4,5)

        result=c.add()

        try:

            assert result == 9, "assertion failure."

        except AssertionError as e:

            print(e) 

if __name__== "_main_":
    t= CountTest()

    t.test_add_4_5()

这里只针对Calculator类的add（方法编写了一条测试用例。

首先，导入count.py文件中的Calculator类，Calculator类需要初始化两个参数，然后，调用add（方法， 得到两个参数相加的返回结果；通过assert断言如果返回结果等于预期结果9，那么测试通过，否则，抛出 AssertionError异常，并打印“assertion failure.”的异常信息。

那么，通过unittest单元测试框架如何编写测试用例。

from count import Calculator import unittest

class CountTest(unittest.TestCase):

    def setUp(self):

        self.c=Calculator(8, 4)

    def tearDown(self):

        pass

    def test_add(self):
        result = self.c.add()
        self.assertEqual(result,12)

    def test_sub(self):

        result=self.c.sub()
        self.assertEqual(result,4)

    def test_mul(self):

        result = self.c.mu1()

        self.assertEqual(result, 32)

    def test_div(self):

        result = self.c

        self.c.div()

        self.assertEqual(result, 2)

if __name__ == "_main_":
    unittest.main()

从感官上来看，通过unittest单元测试框架编写测试更新规范和整洁。接下来，分析一下unittest单元测 试框架的结构。

首先，通过import 导入 unittest单元测试框架。 创建 CountTest类继承 unittest.TestCase类。 setUpO

和tearDo

在单元测试框架中比较特别，它们分别在每一个测试用例的开始和结束执行。setUp0 方法用于测试用例执行前的初始化工作，例如初始化变量、生成数据库测试数据、打开浏览器等。tearDown（） 方法与setUpO方法相呼应，用于测试用例执行之后的善后工作，例如清除数据库测试数据、关闭浏览器等。 unittest要求测试方法必须以“test”开头。例如，test＿add、test＿sub等。

最后，通过unittest.main（方法查询并执行继承了unittest.TestCase的类下面，并且以“test”开头的方法。 执行结果如下：

从执行结果可以看到通过一个小点“．”来表示一条运行通过的用例，总共运行4条测试用例，用时0.001 秒。

关于unittest单元测试框架，我在《Selenium2 自动化测试实战——基于Python 语言》一书中花一个章来介 绍，读者可以参考此书来学习。另外，Python 官方文档也是非常不错的参考资料，如图