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

add test coverage for no_cursor_timeout to support recent fix

Browse Source
This commit is contained in:
Bastien Gérard 2019-09-11 21:53:30 +02:00
parent 47f8a126ca
commit 7d94af0e31
2 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mongoengine/queryset/base.py
View File

@ -1576,6 +1576,7 @@ class BaseQuerySet(object):
if self._snapshot: if self._snapshot:
msg = "The snapshot option is not anymore available with PyMongo 3+" msg = "The snapshot option is not anymore available with PyMongo 3+"
warnings.warn(msg, DeprecationWarning) warnings.warn(msg, DeprecationWarning)
cursor_args = {} cursor_args = {}
if not self._timeout: if not self._timeout:
cursor_args["no_cursor_timeout"] = True cursor_args["no_cursor_timeout"] = True

12
tests/queryset/test_queryset.py
View File

@ -5809,9 +5809,19 @@ class TestQueryset(unittest.TestCase):
self.Person.objects.create(name="Baz") self.Person.objects.create(name="Baz")
self.assertEqual(self.Person.objects.count(with_limit_and_skip=True), 3) self.assertEqual(self.Person.objects.count(with_limit_and_skip=True), 3)
newPerson = self.Person.objects.create(name="Foo_1") self.Person.objects.create(name="Foo_1")
self.assertEqual(self.Person.objects.count(with_limit_and_skip=True), 4) self.assertEqual(self.Person.objects.count(with_limit_and_skip=True), 4)
def test_no_cursor_timeout(self):
qs = self.Person.objects()
self.assertEqual(qs._cursor_args, {}) # ensure no regression of #2148
qs = self.Person.objects().timeout(True)
self.assertEqual(qs._cursor_args, {})
qs = self.Person.objects().timeout(False)
self.assertEqual(qs._cursor_args, {"no_cursor_timeout": True})
if __name__ == "__main__": if __name__ == "__main__":
unittest.main() unittest.main()