接口测试基础——第6篇unittest模块(三)
今天是unittest最后一讲,我们解决一下如何只运行一次setUp和tearDown方法以及简单的数据驱动的知识。
1、只运行一次setUp和tearDown方法
很简单,只需要把setUp和tearDown分别替换为setUpClass和tearDownClass即可,但是用这两个方法必须加上 @classmethod 修饰
# coding: utf-8
import unittest
import time
class MyTest(unittest.TestCase):
@classmethod
def setUpClass(cls):
print "SetUp"
time.sleep(2)
@classmethod
def tearDownClass(cls):
print "teardown"
time.sleep(2)
def test01(self):
print "test01"
def test03(self):
print "test03"
def test02(self):
print "test02"
if __name__ == '__main__':
suite = unittest.TestLoader().loadTestsFromTestCase(MyTest)
unittest.TextTestRunner(verbosity=2).run(suite)
上面的代码运行以后,得到的输出便是:
至于@classmethod是什么意思,大家先不要管,后面在说装饰器的时候我会专门的和大家说。
2、数据驱动
unittest里的数据驱动也要用到装饰器的知识,还是直接先记住代码即可:
# coding: utf-8
import unittest
import time
import ddt
info = [{"username": "captain", "password": 123},
{"username": "warrior", "password": 456}]
@ddt.ddt
class MyTest(unittest.TestCase):
@classmethod
def setUpClass(cls):
print "SetUp"
time.sleep(2)
@classmethod
def tearDownClass(cls):
print "teardown"
time.sleep(2)
@ddt.data(*info)
def test01(self, mes):
print mes
@ddt.data(*info)
def test03(self, res):
print res["username"]
def test02(self):
print "test02"
if __name__ == '__main__':
unittest.main()
运行上面代码:
可以看到,我们的info是一个列表,里面套了两个字典,在打印res["username"]的时候回将两个username都打印出来;这也是数据驱动的好处,也就是将用到的数据放到一起(比如一个Excel中),然后通过数据驱动就可以读到Excel中的所有数据了。
以上内容就是我要讲的所有的unittest需要了解的知识了,其实东西并不多,unittest模块常用的基本上就这三篇的知识,将来无论你做接口测试还是UI自动化,都会用到以上知识,所以,现在就学会了记住了,将来会少走很多弯路~~
欢迎大家关注微信公众号“自动化测试实战”,我们一起进步!