Fix queryset batch_size that wasn't copied to cloned queryset
This commit is contained in:
parent
50ffa8014e
commit
b640c766db
@ -757,7 +757,7 @@ class BaseQuerySet(object):
|
||||
'_read_preference', '_iter', '_scalar', '_as_pymongo',
|
||||
'_limit', '_skip', '_hint', '_auto_dereference',
|
||||
'_search_text', 'only_fields', '_max_time_ms',
|
||||
'_comment')
|
||||
'_comment', '_batch_size')
|
||||
|
||||
for prop in copy_props:
|
||||
val = getattr(self, prop)
|
||||
|
@ -394,6 +394,16 @@ class QuerySetTest(unittest.TestCase):
|
||||
with self.assertRaises(ValueError):
|
||||
list(qs)
|
||||
|
||||
def test_batch_size_cloned(self):
|
||||
class A(Document):
|
||||
s = StringField()
|
||||
|
||||
# test that batch size gets cloned
|
||||
qs = A.objects.batch_size(5)
|
||||
self.assertEqual(qs._batch_size, 5)
|
||||
qs_clone = qs.clone()
|
||||
self.assertEqual(qs_clone._batch_size, 5)
|
||||
|
||||
def test_update_write_concern(self):
|
||||
"""Test that passing write_concern works"""
|
||||
self.Person.drop_collection()
|
||||
|
Loading…
x
Reference in New Issue
Block a user