common: add test for connection with keyFilePath.

lib/common/connection.js

@@ -21,7 +21,6 @@
 var fs = require('fs');
 var GAPIToken = require('gapitoken');
 var req = require('request');
-var path = require('path');
 var pkg = require('../../package.json');
 var util = require('./util');
 
@@ -67,7 +66,6 @@ function Connection(opts) {
 
 /**
  * Retrieves a token to authorize the requests.
-
  * @param  {Function} callback  Callback.
  */
 Connection.prototype.connect = function(callback) {
@@ -93,8 +91,7 @@ Connection.prototype.connect = function(callback) {
 Connection.prototype.fetchToken = function(callback) {
   var that = this;
   if (!this.opts.keyFilename) {
-    // We should be on GCE, try to retrieve token from
-    // the metadata from server.
+    // We should be on GCE, try to retrieve token from the metadata server.
     req({
       method: 'get',
       uri: METADATA_TOKEN_URL,
@@ -113,10 +110,10 @@ Connection.prototype.fetchToken = function(callback) {
     return;
   }
   if (!this.credentials) {
-    // read key file for once and cache the contents.
+    // read key file once and cache the contents.
     fs.readFile(this.opts.keyFilename, function(err, data) {
       if (err) {
-        callback(err); return;
+        return callback(err);
       }
       that.credentials = JSON.parse(data);
       that.fetchServiceAccountToken_(callback);

test/common/connection.js

@@ -20,69 +20,95 @@
 
 var assert = require('assert');
 var async = require('async');
-var conn = require('../../lib/common/connection.js');
+var fs = require('fs');
+var path = require('path');
+
+var connection = require('../../lib/common/connection.js');
 
 describe('Connection', function() {
-  var tokenNeverExpires = new conn.Token('token', new Date(3000, 0, 0));
-  var tokenExpired = new conn.Token('token', new Date(2011, 0, 0));
-  it('should fetch a new token if token expires', function(done) {
-    var c = new conn.Connection({
-      email: 'x@provider',
-      privateKey: '/some/path',
-      scopes: ['scope1', 'scope2']
+  var conn;
+
+  beforeEach(function() {
+    conn = new connection.Connection({
+      keyFilename: path.join(__dirname, '../testdata/privateKeyFile.json')
     });
-    c.token = tokenExpired;
-    c.fetchToken = function() {
-      done();
-    };
-    c.requester = function(opts, callback) {
-      callback(null);
-    };
-    c.req({ uri: 'https://someuri' }, function(){});
   });
 
-  it('should make other requests wait while connecting', function(done) {
-    var numTokenFetches = 0;
-    var c = new conn.Connection({
-      email: 'x@provider',
-      privateKey: '/some/path',
-      scopes: ['scope1', 'scope2']
-    });
-    c.fetchToken = function(cb) {
-      numTokenFetches++;
-      setTimeout(function() {
-        cb(null, tokenNeverExpires);
-      }, 100);
-    };
-    c.requester = function(opts, callback) {
+  it('should use a private key json file', function(done) {
+    var privateKeyFileJson = require('../testdata/privateKeyFile.json');
+    conn.fetchServiceAccountToken_ = function(callback) {
       callback(null);
     };
-
-    async.parallel([
-      function(done) { c.req({ uri: 'https://someuri' }, done); },
-      function(done) { c.req({ uri: 'https://someuri' }, done); },
-      function(done) { c.req({ uri: 'https://someuri' }, done); }
-    ], function(err) {
-      assert.equal(err, null);
-      assert.equal(numTokenFetches, 1);
-      assert.equal(c.token, tokenNeverExpires);
+    conn.fetchToken(function(err) {
+      assert.ifError(err);
+      assert.deepEqual(conn.credentials, privateKeyFileJson);
       done();
     });
   });
 
-  it('should fetch a new token if token is invalid', function(done) {
-    var c = new conn.Connection({
-      email: 'x@provider',
-      privateKey: '/some/path',
-      scopes: ['scope1', 'scope2']
+  describe('Token', function() {
+    var tokenNeverExpires = new connection.Token('token', new Date(3000, 0, 0));
+    var tokenExpired = new connection.Token('token', new Date(2011, 0, 0));
+
+    it('should fetch a new token if token expires', function(done) {
+      var c = new connection.Connection({
+        email: 'x@provider',
+        privateKey: '/some/path',
+        scopes: ['scope1', 'scope2']
+      });
+      c.token = tokenExpired;
+      c.fetchToken = function() {
+        done();
+      };
+      c.requester = function(opts, callback) {
+        callback(null);
+      };
+      c.req({ uri: 'https://someuri' }, function(){});
+    });
+
+    it('should make other requests wait while connecting', function(done) {
+      var numTokenFetches = 0;
+      var c = new connection.Connection({
+        email: 'x@provider',
+        privateKey: '/some/path',
+        scopes: ['scope1', 'scope2']
+      });
+      c.fetchToken = function(cb) {
+        numTokenFetches++;
+        setTimeout(function() {
+          cb(null, tokenNeverExpires);
+        }, 100);
+      };
+      c.requester = function(opts, callback) {
+        callback(null);
+      };
+
+      async.parallel([
+        function(done) { c.req({ uri: 'https://someuri' }, done); },
+        function(done) { c.req({ uri: 'https://someuri' }, done); },
+        function(done) { c.req({ uri: 'https://someuri' }, done); }
+      ], function(err) {
+        assert.equal(err, null);
+        assert.equal(numTokenFetches, 1);
+        assert.equal(c.token, tokenNeverExpires);
+        done();
+      });
+    });
+
+    it('should fetch a new token if token is invalid', function(done) {
+      var c = new connection.Connection({
+        email: 'x@provider',
+        privateKey: '/some/path',
+        scopes: ['scope1', 'scope2']
+      });
+      c.token = new connection.Token();
+      c.fetchToken = function() {
+        done();
+      };
+      c.requester = function(opts, callback) {
+        callback(null);
+      };
+      c.req({ uri: 'https://someuri' }, function(){});
     });
-    c.token = new conn.Token();
-    c.fetchToken = function() {
-      done();
-    };
-    c.requester = function(opts, callback) {
-      callback(null);
-    };
-    c.req({ uri: 'https://someuri' }, function(){});
   });
 });

test/testdata/privateKeyFile.json

@@ -0,0 +1,7 @@
+{
+  "private_key_id": "7",
+  "private_key": "-----BEGIN PRIVATE KEY-----\n\n-----END PRIVATE KEY-----\n",
+  "client_email": "[email protected]",
+  "client_id": "8",
+  "type": "service_account"
+}