From ebd8468b1c44cd19c3425c7d35e983f67b5c8b58 Mon Sep 17 00:00:00 2001 From: Ruslan Kuprieiev Date: Thu, 19 Jul 2018 07:05:28 +0300 Subject: [PATCH] tests: test CmdBase._set_loglevel Signed-off-by: Ruslan Kuprieiev --- tests/test_logger.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/tests/test_logger.py b/tests/test_logger.py index 8959bc462..596313245 100644 --- a/tests/test_logger.py +++ b/tests/test_logger.py @@ -1,6 +1,7 @@ import colorama import logging +from dvc.command.base import CmdBase from dvc.config import Config from dvc.logger import Logger @@ -32,3 +33,19 @@ class TestLogger(TestDvc): msg = Logger.colorize(name, name) # This is not a tty, so it should not colorize anything self.assertEqual(msg, name) + + +class TestLoggerCLI(TestDvc): + def test(self): + class A(object): + quiet = True + verbose = False + + args = A() + CmdBase._set_loglevel(args) + self.assertEqual(Logger.logger().getEffectiveLevel(), logging.CRITICAL) + + args.quiet = False + args.verbose = True + CmdBase._set_loglevel(args) + self.assertEqual(Logger.logger().getEffectiveLevel(), logging.DEBUG)