怎样从0开始搭建一个测试框架_8——断言

草帽路飞UU

这里我们简单封装一个断言，在utils中创建assertion.py文件，在其中创建断言：

1. """
   
2. 在这里添加各种自定义的断言，断言失败抛出AssertionError就OK。
   
3. """
   
4. 
   
5. 
   
6. def assertHTTPCode(response, code_list=None):
   
7.     res_code = response.status_code
   
8.     if not code_list:
   
9.         code_list = [200]
   
10.     if res_code not in code_list:
    
11.         raise AssertionError('响应code不在列表中！')  # 抛出AssertionError，unittest会自动判别为用例Failure，不是Error

复制代码

这个断言传入响应，以及期望的响应码列表，如果响应码不在列表中，则断言失败。

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

复制代码

我们添加断言，响应码在[400]中，执行会发现fail掉了。

在assertion.py中你可以添加更多更丰富的断言，响应断言、日志断言、数据库断言等等，请自行封装。

ljc24787

日志断言、数据库断言等等，请自行封装。