Fix $comment tests for MongoDB 3.1.x.

mongodb · Jul 14, 2015 · 39c41b2 · 39c41b2
1 parent 77de378
commit 39c41b2
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/test/test_cursor.py b/test/test_cursor.py
@@ -1052,9 +1052,12 @@ def find():
 
         run_with_profiling(find)
 
+        # MongoDB 3.1.5 changed the ns for commands.
+        regex = {'$regex': 'pymongo_test.(\$cmd|test)'}
+
         def count():
             self.db.test.find().comment('foo').count()
-            op = self.db.system.profile.find({'ns': 'pymongo_test.$cmd',
+            op = self.db.system.profile.find({'ns': regex,
                                               'op': 'command',
                                               'command.count': 'test',
                                               'command.$comment': 'foo'})
@@ -1064,7 +1067,7 @@ def count():
 
         def distinct():
             self.db.test.find().comment('foo').distinct('type')
-            op = self.db.system.profile.find({'ns': 'pymongo_test.$cmd',
+            op = self.db.system.profile.find({'ns': regex,
                                               'op': 'command',
                                               'command.distinct': 'test',
                                               'command.$comment': 'foo'})