VerboseDebuggingLiveServer

Author: mpasternak

docs/helpers.rst

@@ -378,6 +378,9 @@ Take note, logs will be written to standard output, which is being supressed
 by pytest by default, so you will need to use ``-s`` or ``--capture=no``
 parameter too.
 
+And if your live server still keeps acting weird, there is a switch called
+``--liveserver-debug`` which drops into debugger in the console
+whenever traceback occurs in a live server page.
 
 .. note:: Combining database access fixtures.
 

pytest_django/fixtures.py

@@ -540,6 +540,9 @@ def live_server(request):
     if request.config.getvalue("liveserver_verbose") is True:
         live_server_class = live_server_helper.VerboseLiveServer
 
+        if request.config.getvalue("liveserver_debug") is True:
+            live_server_class = live_server_helper.VerboseDebuggingLiveServer
+
     server = live_server_class(addr)
     request.addfinalizer(server.stop)
     return server

pytest_django/live_server_helper.py

@@ -90,3 +90,9 @@ class VerboseLiveServer(LiveServer):
     def get_live_server_thread_class(self):
         from .verbose_live_server import VerboseLiveServerThread
         return VerboseLiveServerThread
+
+
+class VerboseDebuggingLiveServer(LiveServer):
+    def get_live_server_thread_class(self):
+        from .verbose_live_server import VerboseDebuggingLiveServerThread
+        return VerboseDebuggingLiveServerThread

pytest_django/plugin.py

@@ -132,6 +132,15 @@ def pytest_addoption(parser) -> None:
         default=False,
         help="Enable verbose logging for live_server fixture",
     )
+    group.addoption(
+        "--liveserver-debug",
+        "--liveserver_debug",
+        action="store_true",
+        dest="liveserver_debug",
+        default=False,
+        help="Drops into debugger on tracebacks in live_server "
+             "(only if used with --liveserver-verbose)"
+    )
 
     parser.addini(
         "django_find_project",

pytest_django/verbose_live_server.py

@@ -6,16 +6,32 @@
 
 
 class VerboseWSGIRequestHandler(WSGIRequestHandler):
+    debug_on_traceback = False
+
     def handle_uncaught_exception(self, request, resolver, exc_info):
         traceback.print_tb(exc_info[2], file=sys.stderr)
+        if self.debug_on_traceback:
+            import pytest
+            pytest.set_trace()
+
         return super(VerboseWSGIRequestHandler, self).handle_uncaught_exception(
             request, resolver, exc_info)
 
 
+class VerboseDebuggingWSGIRequestHandler(VerboseWSGIRequestHandler):
+    debug_on_traceback = True
+
+
 class VerboseLiveServerThread(LiveServerThread):
+    request_handler_class = VerboseWSGIRequestHandler
+
     def _create_server(self):
         return ThreadedWSGIServer(
             (self.host, self.port),
-            VerboseWSGIRequestHandler,
+            self.request_handler_class,
             allow_reuse_address=False
         )
+
+
+class VerboseDebuggingLiveServerThread(VerboseLiveServerThread):
+    request_handler_class = VerboseDebuggingWSGIRequestHandler