Pytest项目_day20(log日志)
Log日志
-
优点:记录程序运行信息,方便定位问题
-
python日志模块logging,日志等级如下:
- DEBUG
- INFO(正常)
- WARNING
- ERROR(报错)
-
示例代码如下:
import logging
import os.path
import timeroot_path = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
log_path = os.path.join(root_path, "log")class Logger:def __init__(self):# 定义日志位置和文件名self.logname = os.path.join(log_path, "{}.log".format(time.strftime("%Y-%m-%d")))# 定义一个日志容器self.logger = logging.getLogger("log")# 设置日志打印的级别# 让所有级别的日志都打印输出self.logger.setLevel(logging.DEBUG)# 创建日志输入的格式self.formater = logging.Formatter('[%(asctime)s][%(filename)s %(lineno)d][%(levelname)s]:%(message)s')# 创建日志处理器,用来存放日志文件self.filelogger = logging.FileHandler(self.logname, mode='a', encoding='utf8')# 创建日志处理器,在控制台打印self.console = logging.StreamHandler()# 设置控制台打印日志界别self.console.setLevel(logging.DEBUG)# 文件存放日志级别self.filelogger.setLevel(logging.DEBUG)# 文件存放日志格式self.filelogger.setFormatter(self.formater)# 控制台打印日志格式self.console.setFormatter(self.formater)# 将日志输出渠道添加到日志收集器中self.logger.addHandler(self.filelogger)self.logger.addHandler(self.console)logger = Logger().logger
if __name__ == '__main__':logger.debug("打印debug日志")logger.info("打印info日志")logger.warning("打印warning日志")logger.error("打印error日志")
- 输出结果如下:
- 我们可以在测试用例中引入日志,如下:
import pytestfrom utils.log_util import logger
from utils.yaml_util import func_yaml
from utils.read_data import read_data@pytest.mark.parametrize("data", read_data()["person"])
def test_person(data):logger.info("开始执行test_person测试用例")result = func_yaml(data)print(result)logger.info("test_person测试用例执行完毕")
- 输出结果如下:
完整的get、post、delete、put请求优化
- 我们需要将这些请求都加上日志,需要记录url和参数信息,为了简化操作,放在最后一个函数处
- 代码如下:
import jsonimport requestsfrom utils.log_util import logger
from utils.read import get_dataapi_root_url = get_data.read_ini()["host"]["api_sit_url"]class RestClient:def __init__(self):self.api_root_url = api_root_urldef get(self, url, **kwargs):return self.request(self.api_root_url + url, "GET", **kwargs)def post(self, url, **kwargs):return self.request(self.api_root_url + url, "POST", **kwargs)def put(self, url, **kwargs):return self.request(self.api_root_url + url, "PUT", **kwargs)def delete(self, url, **kwargs):return self.request(self.api_root_url + url, "DELETE", **kwargs)def request(self, url, method, **kwargs):self.request_log(url, method, **kwargs)if method == "GET":return requests.get(url, **kwargs)if method == "POST":return requests.post(url, **kwargs)if method == "PUT":return requests.put(url, **kwargs)if method == "DELETE":return requests.delete(url, **kwargs)def request_log(self, url, method, **kwargs):logger.info("接口请求的地址>>>>{}".format(url))logger.info("接口请求的方法>>>>{}".format(method))data = dict(**kwargs).get("data")json_data = dict(**kwargs).get("json")params = dict(**kwargs).get("params")headers = dict(**kwargs).get("headers")if data is not None:logger.info("接口请求的data参数>>>>\n{}".format(json.dumps(data, indent=2)))if json_data is not None:logger.info("接口请求的json参数>>>>\n{}".format(json.dumps(json_data, indent=2)))if params is not None:logger.info("接口请求的params参数>>>>\n{}".format(json.dumps(params, indent=2)))if headers is not None:logger.info("接口请求的headers参数>>>>\n{}".format(json.dumps(headers, indent=2)))
- 同时,我们使用conftest.py来设置前置和后置操作
import pytestfrom utils.log_util import logger@pytest.fixture(scope="function", autouse=True)
def func():logger.info("开始执行测试用例~")yieldlogger.info("测试用例执行完毕~")
- 最后的效果如下:
请求返回的日志优化
- 我们可以在请求的返回函数中加上log日志记录
- 还可以定义一个类,用于拼接状态码和body
- 注意,此处可以使用python的类属性动态赋值特性,不需要在类中显式定义success和body,而是直接赋值,python会自动创建该类属性
- 代码如下:
# response_util
import jsonfrom core.ResultBase import ResultResponse
from utils.log_util import loggerdef process_response(response):if response.status_code ==200 or response.status_code ==201:ResultResponse.success = TrueResultResponse.body = response.json()logger.info("接口返回的内容>>>>: "+json.dumps(response.json(), ensure_ascii=False))else:ResultResponse.success = Falselogger.info("接口状态码不是2开头,请检查")return ResultResponse# ResultBase
class ResultResponse:pass# Testcase
def test_mobile():print("测试手机归属地get请求")param = get_data.read_data()["mobile_belong_dir"]result = mobile_query(param)assert result.success == Trueassert result.body['status'] == 0assert result.body['msg'] == "ok"assert result.body['result']["shouji"] == "13906191369"assert result.body['result']["province"] == "浙江"assert result.body['result']["city"] == "杭州"assert result.body['result']["company"] == "中国移动"assert result.body['result']["areacode"] == "0571"
- 输出结果如下:
小优化
- 我们可以使用
requests.request()
来简化代码,直接传入method,从而可以自动切换请求类型 - 为了能够保持登陆状态,我们还可以使用session来进行请求,代码如下:
import jsonimport requestsfrom utils.log_util import logger
from utils.read import get_dataapi_root_url = get_data.read_ini()["host"]["api_sit_url"]class RestClient:def __init__(self):self.api_root_url = api_root_url# 新增self.session = requests.Session()def get(self, url, **kwargs):return self.request(url, "GET", **kwargs)def post(self, url, **kwargs):return self.request(url, "POST", **kwargs)def put(self, url, **kwargs):return self.request(url, "PUT", **kwargs)def delete(self, url, **kwargs):return self.request(url, "DELETE", **kwargs)def request(self, url, method, **kwargs):self.request_log(url, method, **kwargs)# if method == "GET":# return requests.get(url, **kwargs)# if method == "POST":# return requests.post(url, **kwargs)# if method == "PUT":# return requests.put(url, **kwargs)# if method == "DELETE":# return requests.delete(url, **kwargs)# 新增response = self.session.request(method, self.api_root_url + url, **kwargs)return responsedef request_log(self, url, method, **kwargs):logger.info("接口请求的地址>>>>{}".format(url))logger.info("接口请求的方法>>>>{}".format(method))data = dict(**kwargs).get("data")json_data = dict(**kwargs).get("json")params = dict(**kwargs).get("params")headers = dict(**kwargs).get("headers")if data is not None:logger.info("接口请求的data参数>>>>\n{}".format(json.dumps(data, indent=2)))if json_data is not None:logger.info("接口请求的json参数>>>>\n{}".format(json.dumps(json_data, indent=2)))if params is not None:logger.info("接口请求的params参数>>>>\n{}".format(json.dumps(params, indent=2)))if headers is not None:logger.info("接口请求的headers参数>>>>\n{}".format(json.dumps(headers, indent=2)))