mirror of
https://github.com/freqtrade/freqtrade.git
synced 2024-11-10 10:21:59 +00:00
131 lines
4.6 KiB
Python
131 lines
4.6 KiB
Python
import logging
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from freqtrade.exceptions import OperationalException
|
|
from freqtrade.loggers import (FTBufferingHandler, FTStdErrStreamHandler, set_loggers,
|
|
setup_logging, setup_logging_pre)
|
|
|
|
|
|
def test_set_loggers() -> None:
|
|
# Reset Logging to Debug, otherwise this fails randomly as it's set globally
|
|
logging.getLogger('requests').setLevel(logging.DEBUG)
|
|
logging.getLogger("urllib3").setLevel(logging.DEBUG)
|
|
logging.getLogger('ccxt.base.exchange').setLevel(logging.DEBUG)
|
|
logging.getLogger('telegram').setLevel(logging.DEBUG)
|
|
|
|
previous_value1 = logging.getLogger('requests').level
|
|
previous_value2 = logging.getLogger('ccxt.base.exchange').level
|
|
previous_value3 = logging.getLogger('telegram').level
|
|
|
|
set_loggers()
|
|
|
|
value1 = logging.getLogger('requests').level
|
|
assert previous_value1 is not value1
|
|
assert value1 is logging.INFO
|
|
|
|
value2 = logging.getLogger('ccxt.base.exchange').level
|
|
assert previous_value2 is not value2
|
|
assert value2 is logging.INFO
|
|
|
|
value3 = logging.getLogger('telegram').level
|
|
assert previous_value3 is not value3
|
|
assert value3 is logging.INFO
|
|
|
|
set_loggers(verbosity=2)
|
|
|
|
assert logging.getLogger('requests').level is logging.DEBUG
|
|
assert logging.getLogger('ccxt.base.exchange').level is logging.INFO
|
|
assert logging.getLogger('telegram').level is logging.INFO
|
|
assert logging.getLogger('werkzeug').level is logging.INFO
|
|
|
|
set_loggers(verbosity=3, api_verbosity='error')
|
|
|
|
assert logging.getLogger('requests').level is logging.DEBUG
|
|
assert logging.getLogger('ccxt.base.exchange').level is logging.DEBUG
|
|
assert logging.getLogger('telegram').level is logging.INFO
|
|
assert logging.getLogger('werkzeug').level is logging.ERROR
|
|
|
|
|
|
@pytest.mark.skipif(sys.platform == "win32", reason="does not run on windows")
|
|
def test_set_loggers_syslog():
|
|
logger = logging.getLogger()
|
|
orig_handlers = logger.handlers
|
|
logger.handlers = []
|
|
|
|
config = {'verbosity': 2,
|
|
'logfile': 'syslog:/dev/log',
|
|
}
|
|
|
|
setup_logging_pre()
|
|
setup_logging(config)
|
|
assert len(logger.handlers) == 3
|
|
assert [x for x in logger.handlers if type(x) == logging.handlers.SysLogHandler]
|
|
assert [x for x in logger.handlers if type(x) == FTStdErrStreamHandler]
|
|
assert [x for x in logger.handlers if type(x) == FTBufferingHandler]
|
|
# setting up logging again should NOT cause the loggers to be added a second time.
|
|
setup_logging(config)
|
|
assert len(logger.handlers) == 3
|
|
# reset handlers to not break pytest
|
|
logger.handlers = orig_handlers
|
|
|
|
|
|
@pytest.mark.skipif(sys.platform == "win32", reason="does not run on windows")
|
|
def test_set_loggers_Filehandler(tmpdir):
|
|
logger = logging.getLogger()
|
|
orig_handlers = logger.handlers
|
|
logger.handlers = []
|
|
logfile = Path(tmpdir) / 'ft_logfile.log'
|
|
config = {'verbosity': 2,
|
|
'logfile': str(logfile),
|
|
}
|
|
|
|
setup_logging_pre()
|
|
setup_logging(config)
|
|
assert len(logger.handlers) == 3
|
|
assert [x for x in logger.handlers if type(x) == logging.handlers.RotatingFileHandler]
|
|
assert [x for x in logger.handlers if type(x) == FTStdErrStreamHandler]
|
|
assert [x for x in logger.handlers if type(x) == FTBufferingHandler]
|
|
# setting up logging again should NOT cause the loggers to be added a second time.
|
|
setup_logging(config)
|
|
assert len(logger.handlers) == 3
|
|
# reset handlers to not break pytest
|
|
if logfile.exists:
|
|
logfile.unlink()
|
|
logger.handlers = orig_handlers
|
|
|
|
|
|
@pytest.mark.skip(reason="systemd is not installed on every system, so we're not testing this.")
|
|
def test_set_loggers_journald(mocker):
|
|
logger = logging.getLogger()
|
|
orig_handlers = logger.handlers
|
|
logger.handlers = []
|
|
|
|
config = {'verbosity': 2,
|
|
'logfile': 'journald',
|
|
}
|
|
|
|
setup_logging_pre()
|
|
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 type(x) == FTStdErrStreamHandler]
|
|
# reset handlers to not break pytest
|
|
logger.handlers = orig_handlers
|
|
|
|
|
|
def test_set_loggers_journald_importerror(import_fails):
|
|
logger = logging.getLogger()
|
|
orig_handlers = logger.handlers
|
|
logger.handlers = []
|
|
|
|
config = {'verbosity': 2,
|
|
'logfile': 'journald',
|
|
}
|
|
with pytest.raises(OperationalException,
|
|
match=r'You need the cysystemd python package.*'):
|
|
setup_logging(config)
|
|
logger.handlers = orig_handlers
|