From 26845c7eb45be5d065a9fdf6e6d93ba49db19f4f Mon Sep 17 00:00:00 2001
From: Don Brown <mrdon@twdata.org>
Date: Tue, 1 Apr 2014 11:44:50 -0600
Subject: [PATCH] Strip additional hosts in the connection string

---
 asyncio_mongo/connection.py |  8 ++++++++
 tests/test_connection.py    | 10 +++++++++-
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/asyncio_mongo/connection.py b/asyncio_mongo/connection.py
index e35b304..1ed02d5 100644
--- a/asyncio_mongo/connection.py
+++ b/asyncio_mongo/connection.py
@@ -30,6 +30,14 @@ class Connection:
         connection = cls()
 
         connection.host = host
+
+        # strip any additional hosts in the connection string for now
+        try:
+            if "," in port:
+                port = port[:port.find(',')]
+        except TypeError:
+            pass
+
         connection.port = port
         connection._loop = loop
         connection._retry_interval = .5
diff --git a/tests/test_connection.py b/tests/test_connection.py
index 175304b..8cd07c6 100644
--- a/tests/test_connection.py
+++ b/tests/test_connection.py
@@ -39,6 +39,14 @@ class TestMongoConnectionMethods(MongoTest):
         pool = asyncio_mongo.Pool.create(mongo_host, mongo_port, poolsize=2)
         self.assertTrue(inspect.isgenerator(pool))
         rapi = yield from pool
-        print('rapi %s' % rapi.__class__)
+        self.assertEqual(isinstance(rapi, asyncio_mongo.Pool), True)
+        rapi.close()
+
+    @async
+    def test_pool(self):
+        # MongoConnectionPool returns deferred, which gets MongoAPI
+        pool = asyncio_mongo.Pool.create(mongo_host, "%s,blah:333" % mongo_port, poolsize=2)
+        self.assertTrue(inspect.isgenerator(pool))
+        rapi = yield from pool
         self.assertEqual(isinstance(rapi, asyncio_mongo.Pool), True)
         rapi.close()
\ No newline at end of file