-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
5 changed files
with
79 additions
and
52 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file was deleted.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,75 @@ | ||
import pytest | ||
import logging | ||
from argparse import ArgumentParser | ||
from fixattiosync.logger import add_args, setup_logger, get_fix_logger, FixLogger, setLoggerClass | ||
|
||
|
||
@pytest.fixture(autouse=True) | ||
def set_custom_logger_class(): | ||
# Ensure FixLogger is used globally for all loggers | ||
setLoggerClass(FixLogger) | ||
|
||
|
||
@pytest.fixture | ||
def arg_parser(): | ||
return ArgumentParser() | ||
|
||
|
||
def test_add_args_verbose(arg_parser): | ||
add_args(arg_parser) | ||
args = arg_parser.parse_args(["--verbose"]) | ||
assert args.verbose is True | ||
assert args.trace is False | ||
assert args.quiet is False | ||
|
||
|
||
def test_add_args_trace(arg_parser): | ||
add_args(arg_parser) | ||
args = arg_parser.parse_args(["--trace"]) | ||
assert args.trace is True | ||
assert args.verbose is False | ||
assert args.quiet is False | ||
|
||
|
||
def test_add_args_quiet(arg_parser): | ||
add_args(arg_parser) | ||
args = arg_parser.parse_args(["--quiet"]) | ||
assert args.quiet is True | ||
assert args.verbose is False | ||
assert args.trace is False | ||
|
||
|
||
def test_setup_logger_verbose(monkeypatch): | ||
monkeypatch.setattr("sys.argv", ["test", "--verbose"]) | ||
setup_logger("fix", force=True) | ||
logger = get_fix_logger("fix") | ||
assert logger.level == logging.DEBUG | ||
|
||
|
||
def test_setup_logger_trace_env(monkeypatch): | ||
monkeypatch.setenv("FIX_TRACE", "true") | ||
setup_logger("fix", force=True) | ||
logger = get_fix_logger("fix") | ||
assert logger.level == logging.DEBUG - 5 # TRACE level | ||
|
||
|
||
def test_setup_logger_quiet(monkeypatch): | ||
monkeypatch.setattr("sys.argv", ["test", "--quiet"]) | ||
setup_logger("fix", force=True) | ||
logger = get_fix_logger("fix") | ||
assert logger.level == logging.CRITICAL | ||
|
||
|
||
@pytest.mark.parametrize( | ||
"env_var,expected_level", | ||
[ | ||
({"FIX_TRACE": "true"}, logging.DEBUG - 5), # TRACE level | ||
({"FIX_VERBOSE": "true"}, logging.DEBUG), | ||
({"FIX_QUIET": "true"}, logging.CRITICAL), | ||
], | ||
) | ||
def test_setup_logger_with_env_vars(monkeypatch, env_var, expected_level): | ||
monkeypatch.setenv(list(env_var.keys())[0], list(env_var.values())[0]) | ||
setup_logger("fix", force=True) | ||
logger = get_fix_logger("fix") | ||
assert logger.level == expected_level |
This file was deleted.
Oops, something went wrong.