datastore: allow running queries in a transaction

googleapis · Nov 23, 2015 · 4ad9931 · 4ad9931
1 parent 85ca38f
commit 4ad9931
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 1 deletion.
diff --git a/lib/datastore/request.js b/lib/datastore/request.js
@@ -760,7 +760,7 @@ DatastoreRequest.prototype.makeReq_ = function(method, body, callback) {
     body.transaction = this.id;
   }
 
-  if (method === 'lookup' && this.id) {
+  if (this.id && (method === 'lookup' || method === 'runQuery')) {
     body.read_options = body.read_options || {};
     body.read_options.transaction = this.id;
   }

diff --git a/test/datastore/request.js b/test/datastore/request.js
@@ -1107,6 +1107,31 @@ describe('Request', function() {
           request.makeReq_('lookup', util.noop);
         });
       });
+
+      describe('runQuery', function() {
+        it('should attach transactional properties', function(done) {
+          request.id = 'EeMXCSGvwcSWGkkABRmGMTWdbi_pa66VflNhQAGblQFMXf9HrmNGa' +
+            'GugEsO1M2_2x7wZvLencG51uwaDOTZCjTkkRh7bw_oyKUgTmtJ0iWJwath7';
+          var expected = new pb.RunQueryRequest({
+            read_options: {
+              transaction: request.id
+            }
+          }).toBuffer();
+          requestOverride = function(req) {
+            assert.deepEqual(req.body, expected);
+            done();
+          };
+          request.makeReq_('runQuery', util.noop);
+        });
+
+        it('should not attach transactional properties', function(done) {
+          requestOverride = function(req) {
+            assert.strictEqual(req.body, '');
+            done();
+          };
+          request.makeReq_('runQuery', util.noop);
+        });
+      });
     });
   });
 });