def maybe_skip(self):
if (yield server_is_mongos(self.cx)):
raise SkipTest("mongos has no maxTimeAlwaysTimeOut fail point")
cmdline = yield get_command_line(self.cx)
if '1' != safe_get(cmdline, 'parsed.setParameter.enableTestCommands'):
if 'enableTestCommands=1' not in cmdline['argv']:
raise SkipTest("testing maxTimeMS requires failpoints")
def maybe_skip(self):
if (yield server_is_mongos(self.cx)):
raise SkipTest("mongos has no maxTimeAlwaysTimeOut fail point")
cmdline = yield get_command_line(self.cx)
if '1' != safe_get(cmdline, 'parsed.setParameter.enableTestCommands'):
if 'enableTestCommands=1' not in cmdline['argv']:
raise SkipTest("testing maxTimeMS requires failpoints")
def maybe_skip(self):
if (yield server_is_mongos(self.cx)):
raise SkipTest("mongos has no maxTimeAlwaysTimeOut fail point")
if not (yield at_least(self.cx, (2, 5, 3, -1))):
raise SkipTest("maxTimeMS requires MongoDB >= 2.5.3")
cmdline = yield get_command_line(self.cx)
if "1" != safe_get(cmdline, "parsed.setParameter.enableTestCommands"):
if "enableTestCommands=1" not in cmdline["argv"]:
raise SkipTest("testing maxTimeMS requires failpoints")
def test_exhaust(self):
if (yield server_is_mongos(self.cx)):
self.assertRaises(InvalidOperation,
self.db.test.find, exhaust=True)
return
self.assertRaises(TypeError, self.db.test.find, exhaust=5)
cur = self.db.test.find(exhaust=True)
self.assertRaises(InvalidOperation, cur.limit, 5)
cur = self.db.test.find(limit=5)
self.assertRaises(InvalidOperation, cur.add_option, 64)
cur = self.db.test.find()
cur.add_option(64)
self.assertRaises(InvalidOperation, cur.limit, 5)
yield self.db.drop_collection("test")
# Insert enough documents to require more than one batch.
yield self.db.test.insert([{} for _ in range(150)])
client = self.motor_client(max_pool_size=1)
# Ensure a pool.
yield client.db.collection.find_one()
socks = client._get_primary_pool().sockets
# Make sure the socket is returned after exhaustion.
cur = client[self.db.name].test.find(exhaust=True)
has_next = yield cur.fetch_next
self.assertTrue(has_next)
self.assertEqual(0, len(socks))
while (yield cur.fetch_next):
cur.next_object()
self.assertEqual(1, len(socks))
# Same as previous but with to_list instead of next_object.
docs = yield client[self.db.name].test.find(exhaust=True).to_list(None)
self.assertEqual(1, len(socks))
self.assertEqual(
(yield self.db.test.count()),
len(docs))
# If the Cursor instance is discarded before being
# completely iterated we have to close and
# discard the socket.
sock = one(socks)
cur = client[self.db.name].test.find(exhaust=True).batch_size(1)
has_next = yield cur.fetch_next
self.assertTrue(has_next)
self.assertEqual(0, len(socks))
if 'PyPy' in sys.version:
# Don't wait for GC or use gc.collect(), it's unreliable.
cur.close()
del cur
yield self.pause(0.1)
# The exhaust cursor's socket was discarded, although another may
# already have been opened to send OP_KILLCURSORS.
self.assertNotIn(sock, socks)
self.assertTrue(sock.closed)
def test_exhaust(self):
if sys.version_info < (3, 4):
raise SkipTest("requires Python 3.4")
if (yield server_is_mongos(self.cx)):
self.assertRaises(InvalidOperation,
self.db.test.find, cursor_type=CursorType.EXHAUST)
return
cur = self.db.test.find(cursor_type=CursorType.EXHAUST)
self.assertRaises(InvalidOperation, cur.limit, 5)
cur = self.db.test.find(limit=5)
self.assertRaises(InvalidOperation, cur.add_option, 64)
cur = self.db.test.find()
cur.add_option(64)
self.assertRaises(InvalidOperation, cur.limit, 5)
yield self.db.drop_collection("test")
# Insert enough documents to require more than one batch.
yield self.db.test.insert_many([{} for _ in range(150)])
client = self.motor_client(maxPoolSize=1)
# Ensure a pool.
yield client.db.collection.find_one()
socks = get_primary_pool(client).sockets
# Make sure the socket is returned after exhaustion.
cur = client[self.db.name].test.find(cursor_type=CursorType.EXHAUST)
has_next = yield cur.fetch_next
self.assertTrue(has_next)
self.assertEqual(0, len(socks))
while (yield cur.fetch_next):
cur.next_object()
self.assertEqual(1, len(socks))
# Same as previous but with to_list instead of next_object.
docs = yield client[self.db.name].test.find(
cursor_type=CursorType.EXHAUST).to_list(None)
self.assertEqual(1, len(socks))
self.assertEqual(
(yield self.db.test.count_documents({})),
len(docs))
# If the Cursor instance is discarded before being
# completely iterated we have to close and
# discard the socket.
sock = one(socks)
cur = client[self.db.name].test.find(
cursor_type=CursorType.EXHAUST).batch_size(1)
has_next = yield cur.fetch_next
self.assertTrue(has_next)
self.assertEqual(0, len(socks))
if 'PyPy' in sys.version:
# Don't wait for GC or use gc.collect(), it's unreliable.
yield cur.close()
del cur
yield gen.sleep(0.1)
# The exhaust cursor's socket was discarded, although another may
# already have been opened to send OP_KILLCURSORS.
self.assertNotIn(sock, socks)
self.assertTrue(sock.closed)
