（三十六）Unittest单元测试框架之认识unittest-认识单元测试

随笔记录方便自己和同路人查阅。

#------------------------------------------------我是可耻的分割线-------------------------------------------

　　学习selenium自动化之前，最好先学习HTML、CSS、JavaScript等知识，有助于理解定位及操作元素的原理。关于python和selenium安装请自行搜索别的资料，这里就不多做介绍了，所有例子均使用python3.6+selenium执行的。

#------------------------------------------------我是可耻的分割线-------------------------------------------

对于不熟悉变成的测试新手来讲，单元测试是个听起来高大上的话题，貌似只有高级测试或开发人员才能胜任这项工作。其实，它并非想象得那么高级，我们通过实例慢慢揭开它神秘的面纱。

可能读者还有个疑问，我们不是在学Web自动化么？为什么要去学习单元测试框架，其实单元测试框架并非只能用于代码级别的测试，对于单元测试框架来讲，笔者认为它主要完成以下三件事。

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

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

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

一般的单元测试框架都会提供这些功能，从范元测试框架的这些特性来看，它同样适用于Web自动化用例的开发与执行。

认识unittest

什么是单元测试？单元测试负责对最小的软件设计单元（模块）进行验证，它使用软件设计文档中对模块的描述作为指南，对重要的程序分支进行测试以发现模块中的错误。在Python语言下有诸多单元测试框架，如doctest、unittest、pytest、nose等，unittest框架（原名PyUnit框架）为Python语言自带的单元测试框架，Python2.1及其以后的版本已将unittest作为一个标准模块放入Python开发包中。

认识单元测试

可能读者会问不用单元测试框架能写单元测试么？答案是坑定的，单元测试本身就是通过一段代码去验证另一段打开，所以不用单元测试框架也可以写单元测试，下面就通过例子演示不用测试框架的单元测试。

首先创建一个被测试类calculator.py

#计算器
class Count():
    def __init__(self,a,b):
        self.a = int(a)
        self.b = int(b)
    
    #计算加法
    def add(self):
        return self.a + self.b

程序非常简单，创建一个Count类用于两个整数的计算，通过__init__()方法对两个数进行初始化，接着创建add()方法返回两个数相加的结果。

根据上面所实现的功能，不用测试框架所编写的但会员测试如test.py

from calculator import Count

#测试两个整数相加
class TestCount():
    def test_add(self):
        try:
            j = Count(2,3)
            add =j.add()
            assert(add == 5),'Integer addition result error!'
        except AssertionError as msg:
            print(msg)
        else:
            print('Test paa!')

#执行测试类的测试方法
mytest = TestCount()
mytest.test_add()

首先，引入calculator文件中的Conut类；然后在test_add()方法中调用Count类并传入两个参数2和3；最后调用Count类中的add()方法对两个参数做假发运行，并通过assert()方法判断add()的返回值是否等于5.如果不相等则抛出自定义的“Integer addition result error!”异常信息，如果相等则打印“Test pass!”

运行结果：

不难发现这种测试存在许多问题。首先，测试程序的写法没有一定的规范可以遵循，是个程序员完全可以能写出始终不同的测试程序来，不统统一的代码维护起来十分麻烦。其次，需要编写大量的辅助代码才能进行单元测试，在test.py中用于测试的代码甚至比被测试的代码还要多，而且这仅仅是一个测试用，对一个单元模块来说，只编写一条测试用例显然是不够的。

为了让单元测试代码更容易维护和编写，最好的方式是遵循一定的规范来编写测试用例，这也是单元测试框架诞生的初衷。接下来讲如何通过unittest单元测试框架编写单元测试用例。

from calculator import Count
import unittest

#测试两个整数相加
class TestCount(unittest.TestCase):
    def setUp(self):
        print('test start')
    def test_add(self):
        j = Count(2,3)
        self.assertEqual(j.add(),5)
    def tearDown(self):
        print("test end")

#执行测试类的测试方法
if __name__ == '__main__':
    unittest.main()

分析上面的代码，首先引入unittest模块，创建TestCount类继承unittest的TestCase类，我们可以将TestCase类堪称是对特定类进行测试的集合。

setUp()方法用于测试用例执行前的初始化工作，这里只简单打印“test start”信息。tearDown()方法与setUp()方法相呼应，用于测试用例执行之后的善后工作（如退出操作），这里打印“test end”信息。

在test_add()中首先调用Count类并传入要计算的数，通过调用add()方法得到两数相加的返回值。这里不再使用繁琐的异常处理，而是调用unittest框架所提供的assertEqual()方法对add()的返回值进行断言，判断两者是否相等，assertEqual()方法由TestCase类继承而来。

unittest提供了全局的main()方法使用它可以方便地将一个单元测试模块变成可以直接运行的测试脚本。main()方法使用TestLoader类来搜索所有半酣在该模块中的”test”命名开头的测试方法，并自动执行它们。

运行结果：