feat: add custom retry options (#1088)

steffnay · web-flow · commit b7f29ed4f748 · 2022-05-18T11:48:32.000-07:00
* feat: add custom retry options

* lint

* update test
diff --git a/src/bigquery.ts b/src/bigquery.ts
@@ -326,6 +326,8 @@ export class BigQuery extends Service {
       baseUrl,
       scopes: ['https://www.googleapis.com/auth/bigquery'],
       packageJson: require('../../package.json'),
+      autoRetry: options.autoRetry,
+      maxRetries: options.maxRetries,
     };
 
     if (options.scopes) {
diff --git a/test/bigquery.ts b/test/bigquery.ts
@@ -283,6 +283,26 @@ describe('BigQuery', () => {
       assert.deepStrictEqual(calledWith.scopes, expectedScopes);
     });
 
+    it('should pass autoRetry from options', () => {
+      const retry = false;
+      const bq = new BigQuery({
+        autoRetry: retry,
+      });
+
+      const calledWith = bq.calledWith_[0];
+      assert.deepStrictEqual(calledWith.autoRetry, retry);
+    });
+
+    it('should pass maxRetries from options', () => {
+      const retryVal = 1;
+      const bq = new BigQuery({
+        maxRetries: retryVal,
+      });
+
+      const calledWith = bq.calledWith_[0];
+      assert.deepStrictEqual(calledWith.maxRetries, retryVal);
+    });
+
     it('should not modify options argument', () => {
       const options = {
         projectId: PROJECT_ID,
