Merge branch 'master' of https://github.com/mjtamlyn/mongoengine into transform_update

This commit is contained in:
Ross Lawley 2011-11-01 01:19:30 -07:00
commit aa9ed614ad
2 changed files with 7 additions and 0 deletions

View File

@ -1365,6 +1365,7 @@ class QuerySet(object):
raise OperationError(u'Update failed [%s]' % unicode(e))
def __iter__(self):
self.rewind()
return self
def _sub_js_fields(self, code):

View File

@ -570,7 +570,13 @@ class QuerySetTest(unittest.TestCase):
people1 = [person for person in queryset]
people2 = [person for person in queryset]
# Check that it still works even if iteration is interrupted.
for person in queryset:
break
people3 = [person for person in queryset]
self.assertEqual(people1, people2)
self.assertEqual(people1, people3)
def test_repr_iteration(self):
"""Ensure that QuerySet __repr__ can handle loops