From 3f9ff7254f206e0926c6ab31cc3a1a1efe39ad2d Mon Sep 17 00:00:00 2001
From: Erdenezul Batmunkh <erdenezul.b@gmail.com>
Date: Mon, 21 May 2018 09:46:23 +0800
Subject: [PATCH] fix queryset tests

---
 tests/queryset/queryset.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/queryset/queryset.py b/tests/queryset/queryset.py
index 9c7ac0e4..dea5b110 100644
--- a/tests/queryset/queryset.py
+++ b/tests/queryset/queryset.py
@@ -970,7 +970,10 @@ class QuerySetTest(unittest.TestCase):
         org = Organization.objects.get(id=o1.id)
         with query_counter() as q:
             org.save(cascade=False)
-            self.assertEqual(q, 0)
+            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
         # two db operations (a query for the reference and an update).