fix queryset tests

2018-05-21 09:46:23 +08:00 · 2018-05-21 09:46:23 +08:00 · 3f9ff7254f
commit 3f9ff7254f
parent f7a3acfaf4
1 changed files with 4 additions and 1 deletions
--- a/tests/queryset/queryset.py
+++ b/tests/queryset/queryset.py
@ -970,6 +970,9 @@ class QuerySetTest(unittest.TestCase):
        org = Organization.objects.get(id=o1.id)
        with query_counter() as q:
            org.save(cascade=False)
+            if mongodb_version >= (3, 0):
+                self.assertEqual(q, 1)
+            else:
                self.assertEqual(q, 0)

        # Saving a doc after you append a reference to it should result in