响应结果统一处理
解决问题:响应结果返回的数据有时候是text,有时候是JSON格式,若是指定返回一种格式,那么不支持的格式返回就会报错,所以得加个判断,如果是dict则返回JSON,如果是其他格式一并返回text
创建 handle_response.py 文件
创建HandleResponse类(翻译:处理响应)
class HandleResponse:
def __init__(self):
self.my_assert = TestCase()
#响应结果统一处理
def handle_response(self,response):
try:
#判断类型,如果是字典(dict),则返回json类型和数据
if isinstance(response.json(),dict):
return {"response_type":"json","response":response.json()}
#如果是其他类型,则返回text类型
except Exception as e:
return {"response_type": "str", "response": response.text}
调用方法,在test_1_login中使用:
#前置处理
@classmethod
def setUpClass(cls) -> None:
#响应结果处理
cls.handle_response = HandleResponse()