Make it easier to enable logs while running tests.

python-websockets · Aug 5, 2024 · 273db5b · 273db5b
1 parent 7bb18a6
commit 273db5b
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/tests/__init__.py b/tests/__init__.py
@@ -1,5 +1,14 @@
 import logging
+import os
 
 
-# Avoid displaying stack traces at the ERROR logging level.
-logging.basicConfig(level=logging.CRITICAL)
+format = "%(asctime)s %(levelname)s %(name)s %(message)s"
+
+if bool(os.environ.get("WEBSOCKETS_DEBUG")):  # pragma: no cover
+    # Display every frame sent or received in debug mode.
+    level = logging.DEBUG
+else:
+    # Hide stack traces of exceptions.
+    level = logging.CRITICAL
+
+logging.basicConfig(format=format, level=level)