python接口测试

python+request+unittest+HTMLTestRunner

首先介绍一下python的requests模块：

requests的使用介绍：requests快速入门

环境说明：

1.WIN 7,   64位

2.Python3.4.3 （pip-8.1.2）

3.Requests  --->pip install requests

4.Unittest   --->unittest 框架是python自带的单元测试框架，python2.1及其以后的版本已将unittest作为一个标准块放入python开发包中，所以unittest不用单独安装。

5.测试报告利用HTMLTestRunner生成。

测试思路：

1、先把每个http接口一个一个写脚本测试。（提交的json串直接放在data字典中，这里没有用到excel等写测试用例，测试用例直接用脚本实现。）

2、写完所有接口的测试脚本后，由于一个接口有好几个测试用例，所有要把同一个接口的py脚本封装成方法，每一个接口封装成一个接口类。

3、用testsuite直接调用这些接口类，构造测试集；或利用unittest自动识别测试用例，TestLoader类中提供的discover（）方法。

  （命名规则：接口名称要以test_XXX开头）

   就好把所有的接口测试用例连起来构建自动化测试了。

4、最后利用HTMLTestRunner生成测试报告。

如图：

单个接口测试例子：

[python]  view plain  copy

1. import requests  
2. import json  
3.   
4. def test_qualification_add():  
5.     url = "http://xxx.xxx.xxx/audit/api/xxx/get"    #测试的接口url  
6.     headers = {"Content-Type":"application/json"}     
7.     data = {                                        #接口传送的参数  
8.         "token": "abcdefg",  
9.         "id": 1,  
10.         "param": {  
11.             "QuId":1    
12.         }  
13.     }  
14.     r = requests.post(url = url,json = data,headers = headers)    #发送请求  
15.     #return r.json  
16.     print (r.text)                                                #获取响应报文  
17.     print (r.status_code)  
18.       
19. if __name__=="__main__":  
20.     test_qualification_add()  

在进行接口测试的过程中对传入的参数进行边界值测试、错误推导测试、等价类测试等测试方法进行测试。那么一个接口就要设很多测试用例。

下面是封装为接口类的例子：

（一个接口地址一个类，每一个方法是一条测试用例）

（setUp()、tearDown()方法会分别作用于每个测试用例的开始和结束，如果每个类中的setUp()、tearDown()，所做的事情一样，那么可以封装一个自己的测试类，如代码：）

[python]  view plain  copy

1. import requests  
2. import json  
3. import unittest  
4. class MyTest(unittest.TestCase):     #封装测试环境的初始化和还原的类  
5.     def setUp(self):                 #放对数据可操作的代码，如对mysql、momgodb的初始化等,这里不对数据库进行操作！  
6.         print("start test")  
7.         pass  
8.     def tearDown(self):             #与setUp()相对  
9.         print("end test")  
10.         pass  
11. class test_xxx_get(MyTest):         #把这个接口封装一个类，下面的方法是具体的测试用例  
12.     '''''接口名称：获取资质'''    #这个描述接口名称  
13.     def test_xxx_get(self):  
14.         '''''测试用例1：哈哈'''   #这个描述接口用例名称  
15.         self.url = "http://xxx.xxx.xxx/audit/api/xxx/get"  #请求url  
16.         self.headers = {"Content-Type":"application/json"}  
17.         self.data = {                            #请求参数  
18.             "token": "abcdefg",  
19.             "id": 1,  
20.             "param": {  
21.                 "QuId": 14   
22.             }  
23.         }    #self.用在方法属性中，表示是该方法的属性，不会影响其他方法的属性。                                       
24.         r = requests.post(url = self.url,json = self.data,headers = self.headers)  
25.         #return r.json()  
26.         print (self.r.text)  
27.         print (self.r.status_code)  
28.         self.assertIn("true",self.r.text)     #断言判断接口返回是否符合要求，可以写多个断言！    
29.           
30. if __name__=="__main__":  
31.     unittest.main()  

unittest提供了全局的main()方法，使用它可以方便的将一个单元测试木块变成可以直接运行的测试脚本；

main()方法使用TestLoader类来搜索所有包含在该木块中以“test”命名开头的测试方法，并自动执行它；

执行方法的默认顺序是：根据ASCII码的顺序加载测试用例，数字与字母的顺序为：0-9，A-Z，a-z。所以以A开头的测试用例方法会优先执行，以a开头会后执行。

所以每一个接口类、测试用例要以test*命名，如下图：

（测试用例放在同一个文件夹：）

runtest.py

[python]  view plain  copy

1. import unittest  
2. import json  
3. import requests  
4. from HTMLTestRunner import HTMLTestRunner  
5. import time  
6.   
7. #加载测试文件 (有几多个接口就加载几多个，一个一个加进来)  
8. import test_creative_add  
9. import test_creative_get  
10. import test_qualification_add  
11. import test_qualification_get  
12. import test_qualification_reflesh  
13.   
14. #构造测试集  
15. suite = unittest.TestSuite() #实例化  
16.   
17. #TestSuite类的addTest()方法把不同测试类中的测试方法组装到测试套件中。  
18. #增加测试用例==》接口文件名.接口类(方法也就是这个接口的其他用例),要把每一个测试用例都增加进来！！！  
19.   
20. suite.addTest(test_creative_add.test_creative_add("test_creative_add")) #增加创意  
21. suite.addTest(test_creative_get.test_creative_get("test_creative_get")) #获取创意  
22. suite.addTest(test_qualification_add.test_qualification_add("test_qualification_add"))#增加资质  
23. suite.addTest(test_qualification_get.test_qualification_get("test_qualification_get"))#获取资质  
24. suite.addTest(test_qualification_reflesh.test_qualification_reflesh("test_qualification_reflesh"))#更新资质  
25. if __name__=="__main__":   
26.            testunit = unittest.TestSuite()   
27.            testunit.addTest(suite)   
28.   
29.            #按照一定的格式获取当前的时间   
30.            now = time.strftime("%Y-%m-%d %H_%M_%S")  
31.              
32.            #定义报告存放路径     
33.            filename = './' + now + 'test_result.html'  
34.            fp = open(filename,"wb")   
35.              
36.            #定义测试报告  
37.            runner = HTMLTestRunner(stream = fp,   
38.                                    title = "xxx接口测试报告",  
39.                                    description = "测试用例执行情况：")   
40.   
41.            #运行测试   
42.            runner.run(testunit)  
43.            fp.close()              #关闭文件对象把数据写进磁盘  

如果有几百条测试用例，在runtest脚本中一个个增加测试用例，那么就很麻烦啦，其实用discover()一步就搞掂了。

discover函数介绍：

discover(start_dir,pattern='test*.py',top_level_dir=None)
找到指定目录下所有测试模块，并可递归查到子目录下的测试木块，只有匹配到的文件名才会被加载。如果启动的不是顶层目录，那么顶层目录必然单独指定。

• start_dir：要测试的木块名或测试用例的目录。
• pattent=‘test*.py’：表示用例文件名的匹配原则。此处匹配文件名一test开头的所有.py类型文件，*表示任意多个字符。
• top_level_dir=None ：测试木块的顶层目录，如果没有顶层目录，默认为None。

这里测试用例都放在同一个目录，都以test*.py的形式！！(如上图)

runtest2.py

[python]  view plain  copy

1. import unittest  
2. import json  
3. import requests  
4. from HTMLTestRunner import HTMLTestRunner  
5. import time  
6.   
7. #定义测试用例的目录为当前目录  
8. test_dir = './'  
9. discover = unittest.defaultTestLoader.discover(test_dir,pattern = 'test*.py')  
10.   
11. if __name__=="__main__":  
12.   
13.   
14.     #按照一定的格式获取当前的时间  
15.     now = time.strftime("%Y-%m-%d %H-%M-%S")  
16.       
17.     #定义报告存放路径  
18.     filename = './' + now + 'test_result.html'  
19.       
20.     fp = open(filename,"wb")  
21.     #定义测试报告  
22.     runner = HTMLTestRunner(stream =  fp,  
23.                             title = "xxx接口测试报告",  
24.                             description = "测试用例执行情况：")  
25.     #运行测试  
26.     runner.run(discover)  
27.     fp.close() #关闭报告文件  

最后生成的测试报告的样子：

整合自动化测试发送测试报告：

runtest_mail.py

[python]  view plain  copy

1. import unittest  
2. import requests  
3. from HTMLTestRunner import HTMLTestRunner  
4. import time  
5. import os  
6. import smtplib  
7. from email.mime.text import MIMEText  
8. from email.header import Header  
9.   
10. #======定义发送邮件========  
11. def send_mail(file_new):  
12.     f = open(file_new,'rb')  
13.     mail_body = f.read()  
14.     f.close()  
15.   
16.     msg = MIMEText(mail_body,'html','utf-8')  
17.     msg['Subject'] = Header('xxx接口自动化测试报告','utf-8')  
18.   
19.     smtp = smtplib.SMTP()  
20.     smtp.connect('smtp.sina.com')  
21.     smtp.login('xxx@sina.com','xxx336..')  
22.     smtp.sendmail('xxx@sina.com','10xxx6@qq.com',msg.as_string())  
23.     smtp.quit()  
24.     print('邮件已发出！注意查收。')  
25.   
26. #======查找测试目录，找到最新生成的测试报告======  
27. def new_report(test_report):  
28.     lists = os.listdir(test_report)  
29.     lists.sort(key=lambda fn:os.path.getmtime(test_report + '\\' + fn))  
30.     file_new = os.path.join(test_report,lists[-1])  
31.     print(file_new)  
32.     return file_new  
33.   
34. if __name__ == "__main__":  
35.     test_dir = "D:\\dsp_testpro\\test_case"  
36.     test_report = "D:\\dsp_testpro\\test_report"  
37.   
38.     discover = unittest.defaultTestLoader.discover(test_dir,  
39.                                                    pattern = 'test*.py')  
40.     #按照一定的格式获取当前的时间  
41.     now = time.strftime("%Y-%m-%d_%H-%M-%S-")  
42.   
43.     #定义报告存放路径  
44.     filename = test_report + "\\" + now + 'result.html'  
45.     fp = open(filename,'wb')  
46.     #定义测试报告  
47.     runner = HTMLTestRunner(stream =  fp,  
48.                             title = "xxx接口测试报告",  
49.                             description = "测试用例执行情况：")  
50.     #运行测试  
51.     runner.run(discover)  
52.     fp.close() #关闭报告文件  
53.   
54.     new_report = new_report(test_report)  
55.     send_mail(new_report)  

邮件如下：