mirror of
https://github.com/pre-commit/pre-commit.git
synced 2026-01-14 13:00:10 -06:00
24 lines
639 B
Python
24 lines
639 B
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
|
|
from pre_commit import color
|
|
from pre_commit.logging_handler import LoggingHandler
|
|
|
|
|
|
def _log_record(message, level):
|
|
return logging.LogRecord('name', level, '', 1, message, {}, None)
|
|
|
|
|
|
def test_logging_handler_color(cap_out):
|
|
handler = LoggingHandler(True)
|
|
handler.emit(_log_record('hi', logging.WARNING))
|
|
ret = cap_out.get()
|
|
assert ret == f'{color.YELLOW}[WARNING]{color.NORMAL} hi\n'
|
|
|
|
|
|
def test_logging_handler_no_color(cap_out):
|
|
handler = LoggingHandler(False)
|
|
handler.emit(_log_record('hi', logging.WARNING))
|
|
assert cap_out.get() == '[WARNING] hi\n'
|