tests: fix snippet tests in sandbox env (#2199)

googleapis · Apr 11, 2017 · a66975e · a66975e
1 parent 12a3d10
commit a66975e
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/packages/google-cloud-vision/src/index.js b/packages/google-cloud-vision/src/index.js
@@ -1712,6 +1712,15 @@ Vision.prototype.readDocument = function(images, options, callback) {
  * @private
  */
 Vision.findImages_ = function(images, callback) {
+  if (global.GCLOUD_SANDBOX_ENV) {
+    callback(null, [
+      {
+        content: new Buffer('')
+      }
+    ]);
+    return;
+  }
+
   var MAX_PARALLEL_LIMIT = 5;
   images = arrify(images);
 

diff --git a/packages/google-cloud-vision/test/index.js b/packages/google-cloud-vision/test/index.js
@@ -1008,6 +1008,23 @@ describe('Vision', function() {
   });
 
   describe('findImages_', function() {
+    it('should return buffer for snippet sandbox', function(done) {
+      global.GCLOUD_SANDBOX_ENV = true;
+
+      Vision.findImages_({}, function(err, images) {
+        delete global.GCLOUD_SANDBOX_ENV;
+        assert.ifError(err);
+
+        assert.deepEqual(images, [
+          {
+            content: new Buffer('')
+          }
+        ]);
+
+        done();
+      });
+    });
+
     it('should convert a File object', function(done) {
       var file = {
         name: 'file-name',