【python】python的unittest使用

对于后台测试中的接口测试，因为结果确定，所以实现自动化相对而言比较容易，同时自动化用例方便维护也可以降低回归成本。

python提供了unittest类用来方便的实现case的自动调用，接着昨天的socket例子，实现了一个有3个测试case的用例，代码如下：

import socket
import sys
import time
import unittest

class testClient(unittest.TestCase):
def setUp(self):
pass
def tearDown(self):
pass
def testInputInt(self):
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.connect(('127.0.0.1', 62001))
sock.send('1')
strBuf = sock.recv(1024)
strReturn = strBuf.split(':')[1]
try:
intReturn = int(strReturn)
except:
pass
self.assertEqual(intReturn, 2)
sock.close()
def testTimeout(self):
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.connect(('127.0.0.1', 62001))
time.sleep(6)
try:
sock.send('1')
strBuf = sock.recv(1024)
self.assertEqual(0, 1)
except:
pass
sock.close()
def testInputString(self):
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.connect(('127.0.0.1', 62001))
sock.send('abcd')
strBuf = sock.recv(1024)
strReturn = strBuf.split(':')[1]
self.assertEqual(strReturn, 'abcd')
sock.close()

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

跑完之后结果如下：

...

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

Ran 3 tests in 6.003s

OK

3个点表示是3个成功的case，失败的话会打E，然后打印出失败的信息。当然，也可以自己加一些log打印。

实现的基本过程是这样的：

1.自定义一个类继承unittest；

2.定义setUp和tearDown方法，分别用来执行调度用例前和调度用例后需要执行的操作；

3.所有的用例需要用test来开头；

4.通过执行unittest.main()来调度所有case。