add test for quiet mode (#586)

Author: Niraj-Kamdar

cve_bin_tool/cli.py

@@ -298,6 +298,9 @@ def main(argv=None):
     if argv is None:
         argv = sys.argv
 
+    # Reset logger level to info
+    LOGGER.setLevel(logging.INFO)
+
     parser = argparse.ArgumentParser(
         prog="cve-bin-tool",
         description=textwrap.dedent(

test/test_cli.py

@@ -1,14 +1,14 @@
 """
 CVE-bin-tool CLI tests
 """
-import os
-import unittest
 import logging
+import os
 import subprocess
+import unittest
+
 from cve_bin_tool.cli import main
 from cve_bin_tool.cvedb import DISK_LOCATION_DEFAULT
 from cve_bin_tool.extractor import Extractor
-
 from .utils import (
     TempDirTest,
     download_file,
@@ -219,3 +219,20 @@ def test_unknown_warning(self):
         warnings = [i for i in cm.output if "WARNING" in i]
         self.assertTrue(len(warnings) > 0)
         self.assertTrue("was detected with version UNKNOWN" in "".join(warnings))
+
+    def test_quiet_mode(self):
+        """ Test that an quite mode isn't generating any output """
+        logger = logging.getLogger()
+
+        # build the test file in test/binaries
+        binaries_path = os.path.join(
+            os.path.abspath(os.path.dirname(__file__)), "binaries"
+        )
+        filename = "test-python-3.7.1.out"
+        subprocess.call(["make", filename], cwd=binaries_path)
+
+        with self.assertLogs(logger, logging.INFO) as cm:
+            main(["cve-bin-tool", "-q", os.path.join(binaries_path, filename)])
+            logger.info("test")
+
+        self.assertEqual(["INFO:root:test"], cm.output)