iframe-proxy | Sunbelt Computer Software

27 lines (19 loc) · 765 Bytes

from __future__ import unicode_literals

from pre_commit import color

from pre_commit.logging_handler import LoggingHandler

class FakeLogRecord(object):

def __init__(self, message, levelname, levelno):

self.message = message

self.levelname = levelname

self.levelno = levelno

def getMessage(self):

return self.message

def test_logging_handler_color(cap_out):

handler = LoggingHandler(True)

handler.emit(FakeLogRecord('hi', 'WARNING', 30))

ret = cap_out.get()

assert ret == color.YELLOW + '[WARNING]' + color.NORMAL + ' hi\n'

def test_logging_handler_no_color(cap_out):

handler = LoggingHandler(False)

handler.emit(FakeLogRecord('hi', 'WARNING', 30))

assert cap_out.get() == '[WARNING] hi\n'

Sunbelt Computer Software