From 0f2d3f2638fae0869af0c6fc6d6ea07514f46a31 Mon Sep 17 00:00:00 2001 From: Matthias Date: Sun, 29 Dec 2024 11:47:51 +0100 Subject: [PATCH] test: Update tests to use rich Handler --- tests/test_log_setup.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_log_setup.py b/tests/test_log_setup.py index 6dc88b79d..d4bc63193 100644 --- a/tests/test_log_setup.py +++ b/tests/test_log_setup.py @@ -6,7 +6,7 @@ import pytest from freqtrade.exceptions import OperationalException from freqtrade.loggers import ( FTBufferingHandler, - FTStdErrStreamHandler, + FtRichHandler, set_loggers, setup_logging, setup_logging_pre, @@ -72,7 +72,7 @@ def test_set_loggers_syslog(): setup_logging(config) assert len(logger.handlers) == 3 assert [x for x in logger.handlers if isinstance(x, logging.handlers.SysLogHandler)] - assert [x for x in logger.handlers if isinstance(x, FTStdErrStreamHandler)] + assert [x for x in logger.handlers if isinstance(x, FtRichHandler)] assert [x for x in logger.handlers if isinstance(x, FTBufferingHandler)] # setting up logging again should NOT cause the loggers to be added a second time. setup_logging(config) @@ -96,7 +96,7 @@ def test_set_loggers_Filehandler(tmp_path): setup_logging(config) assert len(logger.handlers) == 3 assert [x for x in logger.handlers if isinstance(x, logging.handlers.RotatingFileHandler)] - assert [x for x in logger.handlers if isinstance(x, FTStdErrStreamHandler)] + assert [x for x in logger.handlers if isinstance(x, FtRichHandler)] assert [x for x in logger.handlers if isinstance(x, FTBufferingHandler)] # setting up logging again should NOT cause the loggers to be added a second time. setup_logging(config) @@ -145,7 +145,7 @@ def test_set_loggers_journald(mocker): setup_logging(config) assert len(logger.handlers) == 3 assert [x for x in logger.handlers if type(x).__name__ == "JournaldLogHandler"] - assert [x for x in logger.handlers if isinstance(x, FTStdErrStreamHandler)] + assert [x for x in logger.handlers if isinstance(x, FtRichHandler)] # reset handlers to not break pytest logger.handlers = orig_handlers