selectively run CodeWScope group cmd tests based on db version

Author: Mike Dirolf

test/test_collection.py

@@ -21,6 +21,7 @@
 
 import qcheck
 from test_connection import get_connection
+import version
 from pymongo.objectid import ObjectId
 from pymongo.code import Code
 from pymongo.binary import Binary
@@ -582,29 +583,29 @@ def test_group_with_scope(self):
  self.assertEqual(2, db.test.group([], {}, {"count": 0},
  Code(reduce_function,
  {"inc_value": 1}))[0]['count'])
-
-
- self.assertEqual(2, db.test.group([], {}, {"count": 0},
- Code(reduce_function,
- {"inc_value": 1}),
- command=True)[0]['count'])
-
  self.assertEqual(4, db.test.group([], {}, {"count": 0},
  Code(reduce_function,
  {"inc_value": 2}))[0]['count'])
- self.assertEqual(4, db.test.group([], {}, {"count": 0},
- Code(reduce_function,
- {"inc_value": 2}),
- command=True)[0]['count'])
 
  self.assertEqual(1, db.test.group([], {}, {"count": 0},
  Code(reduce_function,
  {"inc_value": 0.5}))[0]['count'])
- self.assertEqual(1, db.test.group([], {}, {"count": 0},
- Code(reduce_function,
- {"inc_value": 0.5}),
- command=True)[0]['count'])
 
+ if version.at_least(db.connection(), (1, 1)):
+ self.assertEqual(2, db.test.group([], {}, {"count": 0},
+ Code(reduce_function,
+ {"inc_value": 1}),
+ command=True)[0]['count'])
+
+ self.assertEqual(4, db.test.group([], {}, {"count": 0},
+ Code(reduce_function,
+ {"inc_value": 2}),
+ command=True)[0]['count'])
+
+ self.assertEqual(1, db.test.group([], {}, {"count": 0},
+ Code(reduce_function,
+ {"inc_value": 0.5}),
+ command=True)[0]['count'])
 
  def test_large_limit(self):
  db = self.db

test/version.py

@@ -0,0 +1,44 @@
+# Copyright 2009 10gen, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Some tools for running tests based on MongoDB server version."""
+
+def _padded(iter, length, padding=0):
+ l = list(iter)
+ if len(l) < length:
+ for _ in range(length - len(l)):
+ l.append(0)
+ return l
+
+def _parse_version_string(version_string):
+ mod = 0
+ if version_string.endswith("+"):
+ version_string = version_string[0:-1]
+ mod = 1
+ elif version_string.endswith("-"):
+ version_string = version_string[0:-1]
+ mod = -1
+
+ version = [int(part) for part in version_string.split(".")]
+ version = _padded(version, 3)
+ version.append(mod)
+
+ return tuple(version)
+
+# Note this is probably broken for very old versions of the database...
+def version(connection):
+ return _parse_version_string(connection.server_info()["version"])
+
+def at_least(connection, min_version):
+ return version(connection) >= tuple(_padded(min_version, 4))