python框架实例_[学习笔记]Python单元测试框架-unittest-实例

让我们用官网上的一个简单的例子来先看看Unittest的使用方式：

import unittest

class TestStringMethods(unittest.TestCase):

def test_upper(self):

self.assertEqual('foo'.upper(), 'FOO')

def test_isupper(self):

self.assertTrue('FOO'.isupper())

self.assertFalse('Foo'.isupper())

def test_split(self):

s = 'hello world'

self.assertEqual(s.split(), ['hello', 'world'])

# check that s.split fails when the separator is not a string

with self.assertRaises(TypeError):

s.split(2)

if __name__ == '__main__':

unittest.main()

一个测试用例是由unittest.Testcase来创建的，上面的代码里面三个测试用例分别对应的三种方式都是以'test'来开头，这种命名方式可以明确的告诉test runner需要运行的测试用例是哪些。

每一个测试用例里面用来做断言的函数就是assert类的函数，包括assertEqual(判断相等),assertTrue(判断为真),assertFalse(判断为假)等等，关于assert的所有用法可以参见：https://docs.python.org/3.5/library/unittest.html#assert-methods

如果要想抛出异常(RuntimeError, TypeError, NameError)，则会使用到函数assertRaise的方式。

接下来我们再看一个例子，现在有一个测试的类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

# 减法

def sub(self):

return self.a - self.b

# 乘法

def mul(self):

return self.a * self.b

# 加法

def div(self):

return self.a / self.b

如果我们想对里面的4个函数进行测试，下面是用unittest框架写出来的代码：

from Unittest.calculator import Count

import unittest

class TestCount(unittest.TestCase):

#测试的初始化

def setUp(self):

print("test start")

#测试用例add的编写

def test_add(self):

self.Count = Count(2, 3)

self.assertEqual(self.Count.add(),5)

# 测试用例sub的编写

def test_sub(self):

self.Count = Count(6, 3)

self.assertEqual(self.Count.sub(), 3)

# 测试用例mul的编写

def test_mul(self):

self.Count = Count(6, 3)

self.assertEqual(self.Count.mul(), 18)

# 测试用例div的编写

def test_div(self):

self.Count = Count(6, 3)

self.assertEqual(self.Count.div(), 2)

#测试的回收-测试的环境还原

def tearDown(self):

print("test end")

if __name__ == "__main__":

# 构造测试集

suite = unittest.TestSuite()

suite.addTest(TestCount("test_add"))

suite.addTest(TestCount("test_sub"))

suite.addTest(TestCount("test_mul"))

suite.addTest(TestCount("test_div"))

# 测试

runner =unittest.TextTestRunner()

runner.run(suite)

setup函数和tearDown函数分别对应的是单元测试用例之前的初始化和之后的回收。

在Unittest中通过addTest这个函数来构造测试集，在这里可以将需要测试的测试用例一一加入

最后通过TextTestRunner运行应该执行的测试用例

我们再来看一下Unitttest的整个静态类图：

state.png

从上图可以看出整个书写Unittest的流程为：首先是要写好TestCase，然后由TestLoader加载TestCase到TestSuite，然后由TextTestRunner来运行TestSuite，运行的结果保存在TextTestResult中，整个过程集成在unittest.main模块中。

在上面的例子中，我们已经将相似的功能放到了同一个类中，如果我们将相似的功能放到同一个文件中时，可能就会有以下的结构方式：

testcount/

|-count.py

|-testadd.py

|-testsub.py

|-testmul.py

|-testdiv.py

|-runtest.py

当有多个测试用例需要运行，需要添加新的用例时就可以到相应的文件中去添加，但是此时的runtest.py文件中通过addTest()来添加/删除测试用例还是会很麻烦，Unittest中有个功能可以很好的解决这个问题，这个方式就是discover(),此方法是来自于TestLoader.discover()

discover(start_dir, pattern='test*.py', top_level_dir=None)

start_dir:要测试的模块名或测试用例目录。

pattern=‘test*.py’:表示用例文件名的匹配原件

top_level_dir=None:测试模块的顶层目录，如果没有顶层目录，默认为None

现在通过discover()方法重新实现runtest.py文件的功能。

import unittest

#定义测试用例的目录为当前目录

test_dir ='./'

discover= unittest.defaultTestLoader.discover(test_dir,pattern='test*.py')

编写Web测试用例

以一个查询的具体的单元测试用例为实例，目录如下：

unittest.png

baidu.py 实现的是百度搜索和更改百度设置的功能

youdao.py实现的是有道词典中查询的功能

runtest.py实现将日志输出到report目录下的log.txt文件中

.C:\Program Files\Python35-32\lib\site-packages\selenium\webdriver\remote\webdriver.py:585: DeprecationWarning: use driver.switch_to.alert instead

warnings.warn("use driver.switch_to.alert instead", DeprecationWarning)

.

----------------------------------------------------------------------

Ran 2 tests in 27.463s

OK

runtest_html.py 中引入了HTMLTestRunner的第三方包，使日志可以用html的格式进行展示