mongodb/mongo-python-driver
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add more tests for maxTimeMS PYTHON-550

Browse Source
This commit is contained in:
Bernie Hackett 2014-03-04 10:35:45 -08:00
parent 502b7d4c0c
commit 22caa27a72
2 changed files with 46 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
test/test_cursor.py
View File

@ -102,19 +102,24 @@ class TestCursor(unittest.TestCase):
self.assertTrue(coll.find_one(max_time_ms=1000))
reducer = Code("""function(obj, prev){prev.count++;}""")
coll.group(key={"amalia": 1}, condition={}, initial={"count": 0},
reduce=reducer, maxTimeMS=1000)
if "enableTestCommands=1" in get_command_line(self.client)["argv"]:
self.client.admin.command("configureFailPoint",
"maxTimeAlwaysTimeOut",
mode="alwaysOn")
self.assertRaises(ExecutionTimeout,
coll.find_one, max_time_ms=1)
self.client.admin.command("configureFailPoint",
"maxTimeAlwaysTimeOut",
mode="off")
try:
cursor = coll.find().max_time_ms(1)
try:
cursor.next()
except ExecutionTimeout:
pass
else:
self.fail("ExecutionTimeout not raised")
self.assertRaises(ExecutionTimeout,
coll.find_one, max_time_ms=1)
finally:
self.client.admin.command("configureFailPoint",
"maxTimeAlwaysTimeOut",
mode="off")
def test_explain(self):
a = self.db.test.find()

34
test/test_database.py
View File

@ -40,14 +40,15 @@ from pymongo.collection import Collection
from pymongo.database import Database
from pymongo.errors import (CollectionInvalid,
ConfigurationError,
ExecutionTimeout,
InvalidName,
OperationFailure)
from pymongo.son_manipulator import (AutoReference,
NamespaceInjector,
ObjectIdShuffler)
from test import version
from test.utils import (is_mongos, server_started_with_auth,
remove_all_users)
from test.utils import (get_command_line, is_mongos,
remove_all_users, server_started_with_auth)
from test.test_client import get_client
@ -959,6 +960,35 @@ class TestDatabase(unittest.TestCase):
warnings.resetwarnings()
warnings.simplefilter("ignore")
def test_command_max_time_ms(self):
if not version.at_least(self.client, (2, 5, 3, -1)):
raise SkipTest("MaxTimeMS requires MongoDB >= 2.5.3")
if "enableTestCommands=1" not in get_command_line(self.client)["argv"]:
raise SkipTest("Test commands must be enabled.")
self.client.admin.command("configureFailPoint",
"maxTimeAlwaysTimeOut",
mode="alwaysOn")
try:
db = self.client.pymongo_test
db.command('count', 'test')
self.assertRaises(ExecutionTimeout, db.command,
'count', 'test', maxTimeMS=1)
pipeline = [{'$project': {'name': 1, 'count': 1}}]
# Database command helper.
db.command('aggregate', 'test', pipeline=pipeline)
self.assertRaises(ExecutionTimeout, db.command,
'aggregate', 'test',
pipeline=pipeline, maxTimeMS=1)
# Collection helper.
db.test.aggregate(pipeline=pipeline)
self.assertRaises(ExecutionTimeout,
db.test.aggregate, pipeline, maxTimeMS=1)
finally:
self.client.admin.command("configureFailPoint",
"maxTimeAlwaysTimeOut",
mode="off")
if __name__ == "__main__":
unittest.main()