Set socket timeout when checking test mongod.

mongodb · Dec 3, 2016 · 34eea7f · 34eea7f
1 parent 2c2e1a4
commit 34eea7f
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/test/test_environment.py b/test/test_environment.py
@@ -143,10 +143,12 @@ def setup_sync_cx(self):
         port = int(os.environ.get("DB_PORT", 27017))
         connectTimeoutMS = 100
         serverSelectionTimeoutMS = 100
+        socketTimeoutMS = 10000
         try:
             client = connected(pymongo.MongoClient(
                 host, port,
                 connectTimeoutMS=connectTimeoutMS,
+                socketTimeoutMS=socketTimeoutMS,
                 serverSelectionTimeoutMS=serverSelectionTimeoutMS,
                 ssl_ca_certs=CA_PEM,
                 ssl=True))
@@ -157,14 +159,19 @@ def setup_sync_cx(self):
                 client = connected(pymongo.MongoClient(
                     host, port,
                     connectTimeoutMS=connectTimeoutMS,
+                    socketTimeoutMS=socketTimeoutMS,
                     serverSelectionTimeoutMS=serverSelectionTimeoutMS,
                     ssl_ca_certs=CA_PEM,
                     ssl_certfile=CLIENT_PEM))
 
                 self.mongod_started_with_ssl = True
                 self.mongod_validates_client_cert = True
             except pymongo.errors.ServerSelectionTimeoutError:
-                client = connected(pymongo.MongoClient(host, port))
+                client = connected(pymongo.MongoClient(
+                    host, port,
+                    connectTimeoutMS=connectTimeoutMS,
+                    socketTimeoutMS=socketTimeoutMS,
+                    serverSelectionTimeoutMS=serverSelectionTimeoutMS))
 
         response = client.admin.command('ismaster')
         if 'setName' in response: