fix: failing postgresql tests

django-commons · Jun 7, 2021 · d0f7cfa · d0f7cfa
1 parent 7d4c530
commit d0f7cfa
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/debug_toolbar/panels/sql/tracking.py b/debug_toolbar/panels/sql/tracking.py
@@ -57,11 +57,10 @@ def cursor(*args, **kwargs):
         def chunked_cursor(*args, **kwargs):
             # prevent double wrapping
             # solves https://github.com/jazzband/django-debug-toolbar/issues/1239
-            if hasattr(connection._djdt_cursor, "__wrapped__"):
-                return connection._djdt_cursor(*args, **kwargs)
-            return state.Wrapper(
-                connection._djdt_chunked_cursor(*args, **kwargs), connection, panel
-            )
+            cursor = connection._djdt_chunked_cursor(*args, **kwargs)
+            if not isinstance(cursor, NormalCursorWrapper):
+                return state.Wrapper(cursor, connection, panel)
+            return cursor
 
         connection.cursor = cursor
         connection.chunked_cursor = chunked_cursor