Merge pull request #15 from andreicioban/maxSockets

Added maxSockets support.

Author: Matt Bernier

lib/client.js

@@ -27,9 +27,16 @@ var response = {
 }
 
 // Client allows for quick and easy access any REST or REST-like API.
-function Client (globalRequest) {
+function Client (globalRequest, maxSockets) {
   var emptyResponse = JSON.parse(JSON.stringify(response))
   var body = ''
+  var httpAgent = null;
+  var httpsAgent = null;
+
+  if(maxSockets) {
+    httpAgent = new http.Agent({ maxSockets: maxSockets });
+    httpsAgent = new https.Agent({ maxSockets: maxSockets });
+  }
 
   // utility function to create an empty request object
   this.emptyRequest = function () {
@@ -81,6 +88,10 @@ function Client (globalRequest) {
       request.headers['Content-Type'] = 'application/json'
     }
 
+    //if maxsockets where provided use the apropriate agent
+    if(maxSockets) {
+      request.agent = endpointRequest.test == true ? httpAgent : httpsAgent
+    }
     return request
   }
 

test/test.js

@@ -209,4 +209,35 @@ describe('Client', function () {
       })
     })
   })
+
+  // limit maxSockets
+  describe('#API()', function () {
+    it('should limit maxSockets', function (done) {
+      var expectedSockets = 10
+      var maxSocketsClient = new Client(globalRequest, expectedSockets)
+      
+      nock(TEST_HOST)
+        .get('/testMax')
+        .reply(200)
+      
+      // monkey patch the http.request
+      var http = require('http')
+      var originalRequest = http.request
+      http.request = function(options, callback){
+        assert.isDefined(options.agent, 'the request should use a custom agent');
+        assert.equal(options.agent.maxSockets, expectedSockets, 'agent.maxSockets should equal expectedSockets')
+        return originalRequest(options, callback)
+      }
+
+      var requestGet = maxSocketsClient.emptyRequest()
+      requestGet.method = 'GET'
+      requestGet.test = true
+      requestGet.path = '/testMax'
+      maxSocketsClient.API(requestGet, function (response) {
+        //restore the opriginal request
+        http.request = originalRequest
+        done()
+      })
+    })
+  })
 })