QuerySet limit function now returns all docs in cursor when 0 is passed

This commit is contained in:
Ali
2017-08-04 11:31:29 +01:00
parent 3794b181d5
commit 2a795e9138
3 changed files with 10 additions and 3 deletions

View File

@@ -124,6 +124,11 @@ class QuerySetTest(unittest.TestCase):
self.assertEqual(len(people2), 1)
self.assertEqual(people2[0], user_a)
# Test limit with 0 as parameter
people = self.Person.objects.limit(0)
self.assertEqual(people.count(with_limit_and_skip=True), 2)
self.assertEqual(len(people), 2)
# Test chaining of only after limit
person = self.Person.objects().limit(1).only('name').first()
self.assertEqual(person, user_a)