mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
import colorama
|
|
import logging
|
|
|
|
from dvc.config import Config
|
|
from dvc.logger import Logger
|
|
|
|
from tests.basic_env import TestDvc
|
|
|
|
|
|
class TestLogger(TestDvc):
|
|
def test_config(self):
|
|
config1 = {Config.SECTION_CORE: {'LogLevel': 'debug'}}
|
|
Logger(config1)
|
|
self.assertEqual(Logger.logger().getEffectiveLevel(), logging.DEBUG)
|
|
|
|
config2 = {Config.SECTION_CORE: {'LogLevel': 'error'}}
|
|
Logger(config2)
|
|
self.assertEqual(Logger.logger().getEffectiveLevel(), logging.ERROR)
|
|
|
|
def test_set_level(self):
|
|
Logger.set_level('debug')
|
|
self.assertEqual(Logger.logger().getEffectiveLevel(), logging.DEBUG)
|
|
|
|
def test_be_quiet(self):
|
|
Logger.be_quiet()
|
|
self.assertEqual(Logger.logger().getEffectiveLevel(), logging.CRITICAL)
|
|
|
|
def test_be_verbose(self):
|
|
Logger.be_verbose()
|
|
self.assertEqual(Logger.logger().getEffectiveLevel(), logging.DEBUG)
|
|
|
|
def test_colorize(self):
|
|
for name, color in Logger.COLOR_MAP.items():
|
|
msg = Logger.colorize(name, name)
|
|
# This is not a tty, so it should not colorize anything
|
|
self.assertEqual(msg, name)
|