[ADD] Levels

django · Mar 26, 2021 · 4ad06b3 · 4ad06b3
1 parent a607915
commit 4ad06b3
Showing 1 changed file with 15 additions and 5 deletions.
diff --git a/django/test/runner.py b/django/test/runner.py
@@ -33,6 +33,11 @@
     tblib = None
 
 
+NOTSET = 0
+DEBUG = 10
+INFO = 20
+
+
 class DebugSQLTextTestResult(unittest.TextTestResult):
     def __init__(self, stream, descriptions, verbosity):
         self.logger = logging.getLogger('django.db.backends')
@@ -486,7 +491,12 @@ def __init__(self, pattern=None, top_level=None, verbosity=1,
 
         self.pattern = pattern
         self.top_level = top_level
-        self.verbosity = verbosity
+        if verbosity == 0:
+            self.level = NOTSET
+        elif verbosity == 1:
+            self.level = INFO
+        else:
+            self.level = DEBUG
         self.interactive = interactive
         self.failfast = failfast
         self.keepdb = keepdb
@@ -582,7 +592,7 @@ def add_arguments(cls, parser):
         )
 
     def log(self, msg, level=2):
-        if self.verbosity >= level:
+        if self.level >= level:
             print(msg)
 
     def setup_test_environment(self, **kwargs):
@@ -627,9 +637,9 @@ def build_suite(self, test_labels=None, extra_tests=None, **kwargs):
 
         if self.tags or self.exclude_tags:
             if self.tags:
-                self.log('Including test tag(s): %s.' % ', '.join(sorted(self.tags)), level=2)
+                self.log('Including test tag(s): %s.' % ', '.join(sorted(self.tags)), level=INFO)
             if self.exclude_tags:
-                self.log('Excluding test tag(s): %s.' % ', '.join(sorted(self.exclude_tags)), level=2)
+                self.log('Excluding test tag(s): %s.' % ', '.join(sorted(self.exclude_tags)), level=INFO)
             all_tests = filter_tests_by_tags(all_tests, self.tags, self.exclude_tags)
 
         all_tests = reorder_tests(all_tests, self.reorder_by, self.reverse)
@@ -719,7 +729,7 @@ def get_databases(self, suite):
 
         unused_databases = [alias for alias in connections if alias not in databases]
         if unused_databases:
-            self.log('Skipping setup of unused database(s): %s.' % ', '.join(sorted(unused_databases)), level=2)
+            self.log('Skipping setup of unused database(s): %s.' % ', '.join(sorted(unused_databases)), level=DEBUG)
         return databases
 
     def run_tests(self, test_labels, extra_tests=None, **kwargs):