bigtable: support local emulator

stephenplusplus · stephenplusplus · commit 1cc8dbad394e · 2016-09-06T09:53:24.000-04:00
diff --git a/packages/bigtable/src/index.js b/packages/bigtable/src/index.js
@@ -49,6 +49,9 @@ var PKG = require('../package.json');
  * @resource [Google Cloud Bigtable Concepts Overview]{@link https://cloud.google.com/bigtable/docs/concepts}
  *
  * @param {object=} options - [Configuration object](#/docs).
+ * @param {string=} options.apiEndpoint - Override the default API endpoint used
+ *     to reach Bigtable. This is useful for connecting to your local Bigtable
+ *     emulator.
  *
  * @example
  * //-
@@ -291,10 +294,20 @@ function Bigtable(options) {
 
   var adminBaseUrl = 'bigtableadmin.googleapis.com';
 
+  var customEndpoint = is.defined(options.apiEndpoint) ||
+    is.defined(process.env.BIGTABLE_EMULATOR_HOST);
+
+  var baseUrl = 'bigtable.googleapis.com';
+
+  if (customEndpoint) {
+    baseUrl = options.apiEndpoint || process.env.BIGTABLE_EMULATOR_HOST;
+  }
+
   var config = {
-    baseUrl: 'bigtable.googleapis.com',
+    baseUrl: baseUrl,
     service: 'bigtable',
     apiVersion: 'v2',
+    customEndpoint: customEndpoint,
     protoServices: {
       Bigtable: googleProtoFiles.bigtable.v2,
       BigtableTableAdmin: {
diff --git a/packages/bigtable/test/index.js b/packages/bigtable/test/index.js
@@ -73,6 +73,7 @@ describe('Bigtable', function() {
   });
 
   beforeEach(function() {
+    delete process.env.BIGTABLE_EMULATOR_HOST;
     bigtable = new Bigtable({ projectId: PROJECT_ID });
   });
 
@@ -113,6 +114,7 @@ describe('Bigtable', function() {
       assert.strictEqual(calledWith.baseUrl, 'bigtable.googleapis.com');
       assert.strictEqual(calledWith.service, 'bigtable');
       assert.strictEqual(calledWith.apiVersion, 'v2');
+      assert.strictEqual(calledWith.customEndpoint, false);
 
       assert.deepEqual(calledWith.protoServices, {
         Bigtable: googleProtoFiles('bigtable/v2/bigtable.proto'),
@@ -146,6 +148,30 @@ describe('Bigtable', function() {
       assert.strictEqual(calledWith.userAgent, PKG.name + '/' + PKG.version);
     });
 
+    it('should work with the emulator', function() {
+      var endpoint = 'http://emulator:8288';
+      process.env.BIGTABLE_EMULATOR_HOST = endpoint;
+
+      var bigtable = new Bigtable({ projectId: PROJECT_ID });
+
+      var calledWith = bigtable.calledWith_[0];
+      assert.strictEqual(calledWith.baseUrl, endpoint);
+      assert.strictEqual(calledWith.customEndpoint, true);
+    });
+
+    it('should work with a custom apiEndpoint', function() {
+      var options = {
+        projectId: PROJECT_ID,
+        apiEndpoint: 'http://local:3888'
+      };
+
+      var bigtable = new Bigtable(options);
+
+      var calledWith = bigtable.calledWith_[0];
+      assert.strictEqual(calledWith.baseUrl, options.apiEndpoint);
+      assert.strictEqual(calledWith.customEndpoint, true);
+    });
+
     it('should set the projectName', function() {
       assert.strictEqual(bigtable.projectName, 'projects/' + PROJECT_ID);
     });
