Commit 74b91b38 authored by Anssi Kääriäinen's avatar Anssi Kääriäinen
Browse files

Added tests for double-pickling a QuerySet 

Refs #21102.
parent c89d80e2

tests/queryset_pickle/tests.py

+12 −0
Original line number Diff line number Diff line
@@ -94,3 +94,15 @@ class PickleabilityTestCase(TestCase): 
    def test_specialized_queryset(self):

        self.assert_pickles(Happening.objects.values('name'))

        self.assert_pickles(Happening.objects.values('name').dates('when', 'year'))



    def test_pickle_prefetch_related_idempotence(self):

        g = Group.objects.create(name='foo')

        groups = Group.objects.prefetch_related('event_set')



        # First pickling

        groups = pickle.loads(pickle.dumps(groups))

        self.assertQuerysetEqual(groups, [g], lambda x: x)



        # Second pickling

        groups = pickle.loads(pickle.dumps(groups))

        self.assertQuerysetEqual(groups, [g], lambda x: x)