Pytest 记录日志输出到控制台和写入文件

Pytest 记录日志输出到控制台和写入文件

引用

CSDN

1.

https://blog.csdn.net/rhx_qiuzhi/article/details/139558489

Pytest是出色的测试框架，但它不会自动显示print语句或日志的输出，这在尝试调试失败的测试或了解流程时可能会成为问题。本文将详细介绍如何在Pytest中配置日志记录，包括输出到控制台和写入文件，帮助开发者更好地观测执行过程和定位问题。

自定义日志记录器和内置的日志记录器

日志记录器的重要性在于可以随时设置不同级别的日志，Python内置的日志记录器分类了5类的日志级别，标识日志的严重级别：

• NOTSET（0）：此级别捕获所有消息，无论其严重性如何。
• DEBUG（10）：此级别用于任何可以帮助识别潜在问题的内容，例如变量值或程序执行的步骤。
• INFO（20）：此级别用于确认事情是否按预期工作。
• WARNING（30）：此级别表示发生了意外情况，或者在不久的将来可能会出现一些问题（例如“磁盘空间不足”）。但是，该软件仍按预期工作。
• ERROR（40）：此级别表示存在更严重的问题，导致软件无法执行某项功能。
• CRITICAL（50）：此级别表示一个非常严重的错误，可能会阻止程序继续运行。

自定义记录器允许您更方便地定义和处理这些级别，从而进一步提高记录过程的精度和控制。

项目代码

项目目录树

temperature_convertor
├── requirements.txt
├── src
│   ├── __init__.py
│   ├── __pycache__
│   │   ├── __init__.cpython-310.pyc
│   │   └── custom_logger.cpython-310.pyc
│   ├── custom_logger.py
│   └── temp_convertor.py
└── tests
    ├── __init__.py
    └── test_temp_convertor.py

运行环境：

% which python3
/usr/local/bin/python3
macOS 14.0

自定义日志记录器

我们使用了一个自定义记录器，配置为在DEBUG级别显示日志消息。通过创建自己的记录器，您可以控制记录器的行为，从而可以自定义它以满足您的特定要求。

# src/custom_logger.py
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import logging
from enum import Enum

class LogLevel(Enum):
    DEBUG = logging.DEBUG
    INFO = logging.INFO
    WARNING = logging.WARNING
    ERROR = logging.ERROR
    CRITICAL = logging.CRITICAL

def console_logger(name: str, level: LogLevel) -> logging.Logger:
    logger = logging.getLogger(f"__{name}__")
    logger.setLevel(level.value)
    # Create a console handler and set its level
    console_handler = logging.StreamHandler()
    console_handler.setLevel(level.value)
    # Set the formatter for the console handler
    formatter = logging.Formatter(
        "%(asctime)s - %(name)s - %(levelname)s - %(message)s",
        datefmt="%m/%d/%Y %I:%M:%S%p",
    )
    console_handler.setFormatter(formatter)
    # Add the console handler to the logger
    logger.addHandler(console_handler)
    return logger

函数源代码

这里通过温度的转换的两个程序来调用设置的日志记录器。

# src/temp_convertor.py
#!/usr/bin/env python
# -*- coding: utf-8 -*-
from src.custom_logger import console_logger, LogLevel

c_logger = console_logger(name="temp_convertor", level=LogLevel.DEBUG)

def fahrenheit_to_celsius(fahrenheit: float) -> float:
    """
    将华氏度转换成摄氏度
    :param fahrenheit: 华氏度
    :return:摄氏度
    """
    c_logger.debug(f"Coverting {fahrenheit}°F to Celsius.")
    celsius = round((fahrenheit - 32) * 5 / 9, 2)
    c_logger.info(f"Result: {celsius}°C")
    return celsius

def celsius_to_fahrenheit(celsius: float) -> float:
    """
    将摄氏度转换成华氏度
    :param celsius:
    :return:
    """
    c_logger.debug(f"Coverting {celsius}°F to fahrenheit.")
    fahrenheit = round((celsius * 9 / 5) + 32, 2)
    c_logger.info(f"Result: {fahrenheit}°C")
    return fahrenheit

if __name__ == '__main__':
    print(fahrenheit_to_celsius(100))
    print(celsius_to_fahrenheit(100))

执行上述代码，检查代码基本逻辑正确。源代码没问题之后，编写一些单元测试来验证实用程序的功能。

# tests/test_temp_convertor.py
#!/usr/bin/env python
# -*- coding: utf-8 -*-
from src.temp_convertor import fahrenheit_to_celsius, celsius_to_fahrenheit

def test_fahrenheit_to_celsius():
    assert fahrenheit_to_celsius(90) == 32.22

def test_celsius_to_fahrenheit():
    assert celsius_to_fahrenheit(19) == 66.2

pytest中定义和覆盖日志记录信息

Pytest提供了多种方法来控制测试过程中的日志记录。其中一种方法是通过CLI定义和覆盖默认日志记录格式。

使用CLI定义Logging

pytest允许通过命令行参数的方式自定义日志输出格式和时间格式，这种灵活性使您能够根据自己的具体需求调整日志显示方式。这里给出一个示例：

pytest -s -v --log-cli-level=INFO --log-format="%(asctime)s %(levelname)s %(message)s" --log-date-format=" %Y-%m-%d %H:%M"

在该命令中，log format设置日志消息本身的格式，而log date format则设置日志消息中时间戳的格式。

使用pytest.ini定义Logging

如果使用持久化的配置，可以在pytest.ini文件中设置日志参数。此文件允许您在默认情况下启用CLI日志记录，并指定默认日志级别。以下是pytest.ini配置的示例：

[pytest]
log_cli = true
log_cli_level = DEBUG
log_cli_format = %(asctime)s %(levelname)s %(message)s
log_cli_date_format = %Y-%m-%d %H:%M:%S

在这里的配置中，设置了几个参数如下所示：

• log_cli：启用到控制台的日志记录。
• log_cli_level：将日志级别设置为DEBUG。
• log_cli_format：设置记录消息的格式。
• log_cli_date_format：设置日志消息中时间戳的格式。

有关配置日志记录pytest.ini，当然也可以使用pyproject.toml，tox.ini，setup.config，但是在Pytest中pytest.ini优先于其他几个配置文件，即使为空也是如此。

运行此操作，我们从关卡中获取每个测试的实时日志。

DEBUG

上面定义了将日志输出到控制台，还可以将其写入文件供以后查询或查看执行信息。需要将文件处理器添加到配置文件中的【日志记录器 】，在Pytest.ini中

[pytest]
log_file = logs/temp_convertor_tests_run.log
log_file_date_format = %Y-%m-%d %H:%M:%S
log_file_format = %(asctime)s - %(name)s %(levelname)s %(message)s
log_file_level = DEBUG

执行pytest命令

pytest -s -v

此时会在tests目录下生成一个log文件

logs/temp_convertor_tests_run.log

2024-06-10 19:49:12 - __temp_convertor__ DEBUG Coverting 90°F to Celsius.
2024-06-10 19:49:12 - __temp_convertor__ INFO Result: 32.22°C
2024-06-10 19:49:12 - __temp_convertor__ DEBUG Coverting 19°F to fahrenheit.
2024-06-10 19:49:12 - __temp_convertor__ INFO Result: 66.2°C

当然还有其他的诸如禁用日志的功能，但如果在测试中可能只有小概率会使用禁用日志的功能

pytest -s -v --show-capture=no

修改单个测试级别的日志

某些时候，可能希望更改特定测试的日志级别，可能是为了调试该测试或减少日志输出中的干扰。Pytest的fixture允许您执行此操作。

caplog

该fixture可以用于控制测试中的日志并与之交互，使用caplog，实现临时修改日志级别、捕获日志信息进行断言，在测试用例中新增

def test_celsius_to_fahrenheit_caplog_ex(caplog):
    caplog.set_level(logging.DEBUG, logger="__temp_conertor__")
    assert celsius_to_fahrenheit(19) == 66.2
    print("printing caplong records...")
    for record in caplog.records:
        print(record.levelname, record.message)

再次执行

pytest -s -v

输出显示测试的日志级别设置为 INFO，并且可以使用 和 访问和打印日志消息。

record.levelname
record.message

在其他的文章中给出了配置控制台和记录到日志文件时，可以使用

logger.config
或
pytest.ini
来实现。

logger.config
是通过编程的方式设置日志记录器的配置，而
pytest.ini
是通过配置文件来设置
pytest
的全局选项，包括日志选项。

pytest.ini
方式相对更简单和方便，因为它可以在一个地方集中管理所有的
pytest
配置，并且不需要在每个测试文件中重复设置日志配置。通过在
pytest.ini

文件中添加相应的日志配置选项，可以轻松地控制日志的输出级别、格式和文件路径等。

然而，如果你需要更细粒度的日志控制，或者需要在不同的测试用例中使用不同的日志配置，那么使用
logger.config
可能更合适。通过编程方式设置日志记录器，可以根据具体的需求动态地调整日志配置。

综上所述，如果你希望简单地配置日志并在整个项目中使用相同的日志设置，推荐使用
pytest.ini
。如果你需要更灵活的日志控制或在不同的测试用例中有不同的日志需求，可以考虑使用
logger.config
。最终的选择取决于你的具体需求和项目的特点。

# pytest.ini
[pytest]
log_cli = True
log_cli_level = INFO
log_cli_format = %(asctime)s (%(levelname)s) | %(filename)s:%(lineno)s | %(message)s
log_cli_date_format = %Y-%m-%d %H:%M:%S
log_file = pytest_log.txt
log_file_level = INFO
log_file_date_format = %Y-%m-%d %H:%M:%S
log_file_format = %(asctime)s ( %(levelname)s ) %(filename)s:%(lineno)s | %(message)s

在上述代码中，
log_cli = True
表示在执行过程中启动实时监测日志，
log_cli_level
指定了监测日志的等级显示，
log_cli_format
定义了输出日志的显示格式，
log_cli_date_format
指定了显示日志的时间格式。

log_file
指定了存放日志文件的路径，
log_file_level
表示文件中显示的日志等级，
log_file_date_format
表示文件中显示的日志时间格式，
log_file_format
表示文件中显示的日志格式。

要使用上述配置，需要在测试用例文件中导入
logging
库，并在需要记录日志的地方使用
logging
模块的方法输出日志信息，例如:

def test_demo_ini():
    import logging
    logging.info(f'这是测试用例的info')
    logging.warning(f'这是测试用例的warning')
    logging.error(f'这是测试用例的error')
    assert 1

控制台输出：

在temp_convertor_tests_run.log日志文件中