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

fix a unit test that fails from time to time

Browse Source
This commit is contained in:
Stefan Wojcik 2017-05-07 23:46:08 -04:00
parent b09698e926
commit 5f1670ffa2
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/queryset/queryset.py
View File

@ -4646,7 +4646,6 @@ class QuerySetTest(unittest.TestCase):
def test_no_cache(self): def test_no_cache(self):
"""Ensure you can add meta data to file""" """Ensure you can add meta data to file"""
class Noddy(Document): class Noddy(Document):
fields = DictField() fields = DictField()
@ -4664,15 +4663,19 @@ class QuerySetTest(unittest.TestCase):
self.assertEqual(len(list(docs)), 100) self.assertEqual(len(list(docs)), 100)
# Can't directly get a length of a no-cache queryset.
with self.assertRaises(TypeError): with self.assertRaises(TypeError):
len(docs) len(docs)
# Another iteration over the queryset should result in another db op.
with query_counter() as q: with query_counter() as q:
self.assertEqual(q, 0)
list(docs) list(docs)
self.assertEqual(q, 1) self.assertEqual(q, 1)
# ... and another one to double-check.
with query_counter() as q:
list(docs) list(docs)
self.assertEqual(q, 2) self.assertEqual(q, 1)
def test_nested_queryset_iterator(self): def test_nested_queryset_iterator(self):
# Try iterating the same queryset twice, nested. # Try iterating the same queryset twice, nested.