Add timeout

README.md

@@ -27,6 +27,7 @@ var clearbit = new Client({key: 'api_key'});
   * `webhook_id` *String*: Custom identifier for the webhook request
   * `subscribe` *Boolean*: Set to `true` to subscribe to the changes
   * `stream` *Boolean*: Set to `true` to use the [streaming API](https://clearbit.com/docs?shell#streaming) instead of webhooks
+  * `timeout` *Integer*: The timeout in milliseconds after which a socket closed error will be thrown.
 
 ```js
 var Person = clearbit.Person;
@@ -51,6 +52,7 @@ Person.find({email: 'email@domain.com'})
   * `domain` *String*: The company domain to look up **(required)**
   * `webhook_id` *String*: Custom identifier for the webhook request
   * `stream` *Boolean*: Set to `true` to use the [streaming API](https://clearbit.com/docs?shell#streaming) instead of webhooks
+  * `timeout` *Integer*: The timeout in milliseconds after which a socket closed error will be thrown.
 
 ```js
 var Company = clearbit.Company;

package.json

 {
   "name": "clearbit",
-  "version": "1.2.3",
+  "version": "1.3.0",
   "description": "Client for Clearbit.co business intelligence APIs",
   "main": "./src",
   "scripts": {

src/client.js

@@ -59,6 +59,12 @@ ClearbitClient.prototype.request = function (options) {
     method: 'get'
   });
 
+  if (options.timeout) {
+    var timeout = options.timeout;
+  } else {
+    var timeout = options.stream ? 60000 : 10000;
+  }
+
   return needle.requestAsync(
     options.method,
     this.url(options),
@@ -66,7 +72,7 @@ ClearbitClient.prototype.request = function (options) {
     {
       json: options.json,
       headers: options.headers,
-      timeout: options.stream ? 60000 : 10000,
+      timeout: timeout,
       username: this.key,
       password: '',
       user_agent: 'ClearbitNode/v' + pkg.version

test/client.js

@@ -100,18 +100,51 @@ describe('Client', function () {
       });
     });
 
-    it('uses a timeout of 60 seconds for streaming requests', function () {
+    function requestWithOptions(clientRequestOptions) {
       sinon.stub(needle, 'request').yieldsAsync(null, {}, undefined);
-      return client.request({
+
+      return client
+        .request(clientRequestOptions)
+        .then(() => needle.request.firstCall.args[3])
+        .finally(() => needle.request.restore());
+    }
+
+    it('can specify a custom timeout for a request', function () {
+      return requestWithOptions({
+        api: 'person',
+        timeout: 30000
+      })
+      .then((needleOptions) => {
+        expect(needleOptions).to.have.property('timeout', 30000);
+      });
+    });
+
+    it('sets the default timeout to 10 seconds', function () {
+      return requestWithOptions({ api: 'person' })
+      .then((needleOptions) => {
+        expect(needleOptions).to.have.property('timeout', 10000);
+      });
+    });
+
+
+    it('uses a default timeout of 60 seconds for streaming requests', function () {
+      return requestWithOptions({
         api: 'person',
         stream: true
       })
-      .then(function () {
-        expect(needle.request.firstCall.args[3])
-          .to.have.property('timeout', 60000);
+      .then((needleOptions) => {
+        expect(needleOptions).to.have.property('timeout', 60000);
+      });
+    });
+
+    it('can specify a custom timeout for streaming requests', function () {
+      return requestWithOptions({
+        api: 'person',
+        stream: true,
+        timeout: 30000
       })
-      .finally(function () {
-        needle.request.restore();
+      .then((needleOptions) => {
+        expect(needleOptions).to.have.property('timeout', 30000);
       });
     });