Fix #877: Fix tests for pymongo 2.8+

2015-02-15 10:02:22 +02:00
parent 3df3d27533
commit 25298c72bb
2 changed files with 7 additions and 3 deletions
--- a/tests/document/indexes.py
+++ b/tests/document/indexes.py
@@ -495,9 +495,12 @@ class IndexesTest(unittest.TestCase):

        self.assertEqual(BlogPost.objects.hint([('ZZ', 1)]).count(), 10)

-        def invalid_index():
-            BlogPost.objects.hint('tags')
-        self.assertRaises(TypeError, invalid_index)
+        if pymongo.version >= '2.8':
+            self.assertEqual(BlogPost.objects.hint('tags').count(), 10)
+        else:
+            def invalid_index():
+                BlogPost.objects.hint('tags')
+            self.assertRaises(TypeError, invalid_index)

        def invalid_index_2():
            return BlogPost.objects.hint(('tags', 1))