Skip to content

Commit

Permalink
Fixed #13227 -- Ensure that the query cache is flushed when a QuerySe…
Browse files Browse the repository at this point in the history
…t is deepcopied, avoiding problems when an evaluated queryset is used as a subquery. Thanks to claudep for the report.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@12970 bcc190cf-cafb-0310-a4f2-bffc1f526a37
  • Loading branch information
freakboy3742 committed Apr 13, 2010
1 parent b3390fe commit e93f56b
Show file tree
Hide file tree
Showing 2 changed files with 24 additions and 5 deletions.
9 changes: 5 additions & 4 deletions django/db/models/query.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,11 +45,12 @@ def __deepcopy__(self, memo):
"""
Deep copy of a QuerySet doesn't populate the cache
"""
obj_dict = deepcopy(self.__dict__, memo)
obj_dict['_iter'] = None

obj = self.__class__()
obj.__dict__.update(obj_dict)
for k,v in self.__dict__.items():
if k in ('_iter','_result_cache'):
obj.__dict__[k] = None
else:
obj.__dict__[k] = deepcopy(v, memo)
return obj

def __getstate__(self):
Expand Down
20 changes: 19 additions & 1 deletion tests/regressiontests/queries/tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
from django.db.models import Count
from django.test import TestCase

from models import Tag, Annotation, DumbCategory
from models import Tag, Annotation, DumbCategory, Note, ExtraInfo

class QuerysetOrderedTests(unittest.TestCase):
"""
Expand Down Expand Up @@ -63,3 +63,21 @@ def test_sliced_delete(self):
# This prevents us from even evaluating this test case at all.
# Refs #10099
self.assertFalse(connections[DEFAULT_DB_ALIAS].features.allow_sliced_subqueries)

class CloneTests(TestCase):
def test_evaluated_queryset_as_argument(self):
"#13227 -- If a queryset is already evaluated, it can still be used as a query arg"
n = Note(note='Test1', misc='misc')
n.save()
e = ExtraInfo(info='good', note=n)
e.save()

n_list = Note.objects.all()
# Evaluate the Note queryset, populating the query cache
list(n_list)
# Use the note queryset in a query, and evalute
# that query in a way that involves cloning.
try:
self.assertEquals(ExtraInfo.objects.filter(note__in=n_list)[0].info, 'good')
except:
self.fail('Query should be clonable')

0 comments on commit e93f56b

Please sign in to comment.