1
0
mirror of https://github.com/django/django.git synced 2025-10-29 00:26:07 +00:00

Fixed #21102 -- pickling a QuerySet with prefetches twice

Fixed the bug that a QuerySet that prefetches related objects cannot be
pickled and unpickled more than once (The second pickling attempt
raises an exception).

Added a new test for the queryset pickling idempotency.

The bug was introduced by
bac187c0d8.
This commit is contained in:
Minjong Chung
2013-09-13 21:18:50 +09:00
committed by Anssi Kääriäinen
parent dbc2e8eb73
commit e66fe357b2
2 changed files with 14 additions and 2 deletions

View File

@@ -73,3 +73,15 @@ class PickleabilityTestCase(TestCase):
empty = pickle.loads(dumped)
self.assertQuerysetEqual(
empty, [])
def test_pickle_prefetch_related_idempotence(self):
p = Post.objects.create()
posts = Post.objects.prefetch_related('materials')
# First pickling
posts = pickle.loads(pickle.dumps(posts))
self.assertQuerysetEqual(posts, [p], lambda x: x)
# Second pickling
posts = pickle.loads(pickle.dumps(posts))
self.assertQuerysetEqual(posts, [p], lambda x: x)