arti/asyncio-mongo_tailablecursor
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix multiple host issue at pool level

Browse Source
This commit is contained in:
Don Brown
2014-04-01 11:56:36 -06:00
parent 26845c7eb4
commit 9f5bdb3926
3 changed files with 26 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tests/test_connection.py
View File

@@ -43,9 +43,20 @@ class TestMongoConnectionMethods(MongoTest):
rapi.close()
@async
def test_pool(self):
def test_pool_multiple_hosts(self):
# MongoConnectionPool returns deferred, which gets MongoAPI
pool = asyncio_mongo.Pool.create(mongo_host, "%s,blah:333" % mongo_port, poolsize=2)
pool = asyncio_mongo.Pool.create(url="mongodb://{host}:{port},otherhost:333".format(
host=mongo_host, port=mongo_port), poolsize=2)
self.assertTrue(inspect.isgenerator(pool))
rapi = yield from pool
self.assertEqual(isinstance(rapi, asyncio_mongo.Pool), True)
rapi.close()
@async
def test_pool_with_url(self):
# MongoConnectionPool returns deferred, which gets MongoAPI
pool = asyncio_mongo.Pool.create(url="mongodb://{host}:{port}".format(
host=mongo_host, port=mongo_port), poolsize=2)
self.assertTrue(inspect.isgenerator(pool))
rapi = yield from pool
self.assertEqual(isinstance(rapi, asyncio_mongo.Pool), True)