diff options
Diffstat (limited to 'django/test')
| -rw-r--r-- | django/test/testcases.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/django/test/testcases.py b/django/test/testcases.py index 4538716ce7..8203a3837d 100644 --- a/django/test/testcases.py +++ b/django/test/testcases.py @@ -175,6 +175,7 @@ class SimpleTestCase(unittest.TestCase): for alias in connections: connection = connections[alias] connection.cursor = _CursorFailure(cls.__name__, connection.cursor) + connection.chunked_cursor = _CursorFailure(cls.__name__, connection.chunked_cursor) @classmethod def tearDownClass(cls): @@ -182,6 +183,7 @@ class SimpleTestCase(unittest.TestCase): for alias in connections: connection = connections[alias] connection.cursor = connection.cursor.wrapped + connection.chunked_cursor = connection.chunked_cursor.wrapped if hasattr(cls, '_cls_modified_context'): cls._cls_modified_context.disable() delattr(cls, '_cls_modified_context') |