pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mongoengine/queryset.py
View File

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

6
tests/queryset.py
View File

@ -570,7 +570,13 @@ class QuerySetTest(unittest.TestCase):
people1 = [person for person in queryset] people1 = [person for person in queryset]
people2 = [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, people2)
self.assertEqual(people1, people3)
def test_repr_iteration(self): def test_repr_iteration(self):
"""Ensure that QuerySet __repr__ can handle loops """Ensure that QuerySet __repr__ can handle loops