Add unit tests for retries

googleapis · Dec 13, 2017 · e561e4f · e561e4f
1 parent 5fa4aa6
commit e561e4f
Showing 1 changed file with 63 additions and 0 deletions.
diff --git a/test/table.js b/test/table.js
@@ -1122,6 +1122,69 @@ describe('Bigtable/Table', function() {
         table.mutate(entries, done);
       });
     });
+
+    describe('retries', function() {
+      var fakeStatuses;
+      var entryRequests;
+
+      beforeEach(function() {
+        entryRequests = [];
+        fakeStatuses = [
+          [
+            {
+              index: 0,
+              status: {
+                code: 0,
+              },
+            },
+            {
+              index: 1,
+              status: {
+                code: 4,
+              },
+            },
+          ],
+          [
+            {
+              index: 0,
+              status: {
+                code: 0,
+              },
+            },
+
+          ]
+        ];
+        FakeGrpcService.decorateStatus_ = function(status) {
+          return {};
+        };
+        table.requestStream = function(_, reqOpts) {
+          entryRequests.push(reqOpts.entries);
+          var stream = new Stream({
+            objectMode: true,
+          });
+
+          setImmediate(function() {
+            stream.emit('request');
+            stream.end({entries: fakeStatuses.shift()});
+          });
+
+          return stream;
+        };
+      });
+
+      it('should succeed after a retry', function(done) {
+        table.maxRetries = 1;
+        table.mutate(entries, done);
+      });
+
+      it('should retry the same failed entry', function(done) {
+        table.maxRetries = 1;
+        table.mutate(entries, function() {
+          assert.strictEqual(entryRequests[0][1], entryRequests[1][0]);
+          done();
+        });
+      });
+    });
   });
 
   describe('row', function() {