怎样从0开始搭建一个测试框架_7——接口

草帽路飞UU

我们先在ReadMe.md中补上新加的依赖库。然后在utils中创建一个client.py的文件，在其中创建一个HTTPClient类：

1. """
   
2. 添加用于接口测试的client，对于HTTP接口添加HTTPClient，发送http请求。
   
3. 还可以封装TCPClient，用来进行tcp链接，测试socket接口等等。
   
4. """
   
5. 
   
6. import requests
   
7. from utils.log import logger
   
8. 
   
9. METHODS = ['GET', 'POST', 'HEAD', 'TRACE', 'PUT', 'DELETE', 'OPTIONS', 'CONNECT']
   
10. 
    
11. 
    
12. class UnSupportMethodException(Exception):
    
13.     """当传入的method的参数不是支持的类型时抛出此异常。"""
    
14.     pass
    
15. 
    
16. 
    
17. class HTTPClient(object):
    
18.     """
    
19.     http请求的client。初始化时传入url、method等，可以添加headers和cookies，但没有auth、proxy。
    
20. 
    
21.     >>> HTTPClient('http://www.baidu.com').send()
    
22.     <Response [200]>
    
23. 
    
24.     """
    
25.     def __init__(self, url, method='GET', headers=None, cookies=None):
    
26.         """headers: 字典。 例：headers={'Content_Type':'text/html'}，cookies也是字典。"""
    
27.         self.url = url
    
28.         self.session = requests.session()
    
29.         self.method = method.upper()
    
30.         if self.method not in METHODS:
    
31.             raise UnSupportMethodException('不支持的method:{0}，请检查传入参数！'.format(self.method))
    
32. 
    
33.         self.set_headers(headers)
    
34.         self.set_cookies(cookies)
    
35. 
    
36.     def set_headers(self, headers):
    
37.         if headers:
    
38.             self.session.headers.update(headers)
    
39. 
    
40.     def set_cookies(self, cookies):
    
41.         if cookies:
    
42.             self.session.cookies.update(cookies)
    
43. 
    
44.     def send(self, params=None, data=None, **kwargs):
    
45.         response = self.session.request(method=self.method, url=self.url, params=params, data=data, **kwargs)
    
46.         response.encoding = 'utf-8'
    
47.         logger.debug('{0} {1}'.format(self.method, self.url))
    
48.         logger.debug('请求成功: {0}\n{1}'.format(response, response.text))
    
49.         return response

复制代码

接下来写个用例，但是我们接口的用例跟UI混在一起总是不好，所以我们可以在test下创建一个interface的目录，里面创建test_baidu_http.py的用例文件。

这里你也可以在test下分成API和UI两层，分别在其中再进行分层，看情况而定吧。

test_baidu_http.py：

1. import unittest
   
2. from utils.config import Config, REPORT_PATH
   
3. from utils.client import HTTPClient
   
4. from utils.log import logger
   
5. from utils.HTMLTestRunner import HTMLTestRunner
   
6. 
   
7. 
   
8. class TestBaiDuHTTP(unittest.TestCase):
   
9.     URL = Config().get('URL')
   
10. 
    
11.     def setUp(self):
    
12.         self.client = HTTPClient(url=self.URL, method='GET')
    
13. 
    
14.     def test_baidu_http(self):
    
15.         res = self.client.send()
    
16.         logger.debug(res.text)
    
17.         self.assertIn('百度一下，你就知道', res.text)
    
18. 
    
19. 
    
20. if __name__ == '__main__':
    
21.     report = REPORT_PATH + '\\report.html'
    
22.     with open(report, 'wb') as f:
    
23.         runner = HTMLTestRunner(f, verbosity=2, title='从0搭建测试框架 灰蓝', description='接口html报告')
    
24.         runner.run(TestBaiDuHTTP('test_baidu_http'))

复制代码

这里我们加了一句断言，没有断言怎么能叫用例，我们之前写的UI用例，也可以自己动手加上断言。

现在我们的框架既可以做UI测试，也能做接口测试了。如果你的接口类型不是HTTP的，请自己封装对应的Client类。socket库测TCP接口、suds库测SOAP接口，不论你是什么类型的接口，总能找到对应的Python库的。

ljc24787

也可以自己动手加上断言