implementing mocha retries

googleapis · Oct 23, 2023 · 674dd37 · 674dd37
1 parent 53781db
commit 674dd37
Show file tree

Hide file tree

Showing 10 changed files with 40 additions and 11 deletions.
diff --git a/samples/test/auth.test.js b/samples/test/auth.test.js
@@ -53,6 +53,7 @@ describe('authUserFlow()', () => {
     const output = await authUserFlow.main.exchangeCode('abc123');
     assert.strictEqual(output, 'tokens');
     sinon.assert.calledWith(tokenStub, 'abc123');
+    retries(3);
   });
 
   it('should return project id and credentials', async () => {

diff --git a/samples/test/authViewTutorial.test.js b/samples/test/authViewTutorial.test.js
@@ -36,7 +36,8 @@ const sharedViewId = generateUuid();
 
 const bigquery = new BigQuery();
 
-describe('Authorized View Tutorial', () => {
+describe('Authorized View Tutorial', function () {
+  this.retries(3);
   after(async () => {
     await bigquery.dataset(datasetId).delete({force: true}).catch(console.warn);
     await bigquery

diff --git a/samples/test/datasets.test.js b/samples/test/datasets.test.js
@@ -29,7 +29,8 @@ const datasetId = `${GCLOUD_TESTS_PREFIX}_datasets_${uuid.v4()}`.replace(
 
 const bigquery = new BigQuery();
 
-describe('Datasets', () => {
+describe('Datasets', function () {
+  this.retries(3);
   before(async () => {
     // Delete any stale datasets from samples tests
     await deleteDatasets();

diff --git a/samples/test/helpers.ts b/samples/test/helpers.ts
@@ -0,0 +1,16 @@
+const retry = (asyncMethod: Function, counter?: number | undefined) => {
+    let retryCounter: number;
+    counter ? retryCounter = counter : retryCounter = 0;
+
+    try {
+        const res = asyncMethod;
+        return res;
+    } catch (err) {
+        if (retryCounter <= 3) {
+            asyncMethod;
+            retry(asyncMethod, retryCounter++);
+        } else {
+            throw err;
+        }
+    }
+}
diff --git a/samples/test/jobs.test.js b/samples/test/jobs.test.js
@@ -24,7 +24,8 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 const bigquery = new BigQuery();
 let jobId;
 
-describe('Jobs', () => {
+describe('Jobs', function () {
+  this.retries(3);
   before(async () => {
     const query = `SELECT name
         FROM \`bigquery-public-data.usa_names.usa_1910_2013\`

diff --git a/samples/test/models.test.js b/samples/test/models.test.js
@@ -27,6 +27,7 @@ const GCLOUD_TESTS_PREFIX = 'nodejs_samples_tests_models';
 const bigquery = new BigQuery();
 
 describe('Models', function () {
+  this.retries(3);
   // Increase timeout to accommodate model creation.
   this.timeout(300000);
   const datasetId = `${GCLOUD_TESTS_PREFIX}_${uuid.v4()}`.replace(/-/gi, '_');
@@ -91,7 +92,8 @@ describe('Models', function () {
   });
 });
 
-describe('Create/Delete Model', () => {
+describe('Create/Delete Model', function () {
+  this.retries(3);
   const datasetId = `${GCLOUD_TESTS_PREFIX}_delete_${uuid.v4()}`.replace(
     /-/gi,
     '_'

diff --git a/samples/test/queries.test.js b/samples/test/queries.test.js
@@ -36,7 +36,8 @@ let projectId;
 
 const bigquery = new BigQuery();
 
-describe('Queries', () => {
+describe('Queries', function () {
+  this.retries(3);
   before(async () => {
     const schema = [{name: 'age', type: 'STRING', mode: 'REQUIRED'}];
     const options = {

diff --git a/samples/test/quickstart.test.js b/samples/test/quickstart.test.js
@@ -24,7 +24,8 @@ const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 
 const bigquery = new BigQuery();
 
-describe('Quickstart', () => {
+describe('Quickstart', function () {
+  this.retries(3);
   const datasetName = `nodejs_samples_tests_quickstart_${uuid.v4()}`.replace(
     /-/gi,
     '_'

diff --git a/samples/test/routines.test.js b/samples/test/routines.test.js
@@ -34,7 +34,8 @@ const newRoutineId = generateUuid();
 
 const bigquery = new BigQuery();
 
-describe('Routines', () => {
+describe('Routines', function () {
+  this.retries(3);
   after(async () => {
     await bigquery.dataset(datasetId).delete({force: true}).catch(console.warn);
   });
@@ -84,7 +85,8 @@ describe('Routines', () => {
     assert.include(output, 'Routine description: New description');
   });
 
-  describe('Delete Routine', () => {
+  describe('Delete Routine', function () {
+    this.retries(3);
     const datasetId = `gcloud_tests_${uuid.v4()}`.replace(/-/gi, '_');
     const routineId = `gcloud_tests_${uuid.v4()}`.replace(/-/gi, '_');
 

diff --git a/samples/test/tables.test.js b/samples/test/tables.test.js
@@ -61,7 +61,8 @@ const partialDataFilePath = path.join(
 );
 const bigquery = new BigQuery();
 
-describe('Tables', () => {
+describe('Tables', function () {
+  this.retries(3);
   before(async () => {
     const [bucket] = await storage.createBucket(bucketName);
     await Promise.all([
@@ -618,7 +619,8 @@ describe('Tables', () => {
     assert.include(output, 'color: green');
   });
 
-  describe('Views', () => {
+  describe('Views', function () {
+    this.retries(3);
     it('should create a view', async () => {
       const output = execSync(`node createView.js ${datasetId} ${viewId}`);
       assert.include(output, `View ${viewId} created.`);
@@ -640,7 +642,8 @@ describe('Tables', () => {
     });
   });
 
-  describe('Delete Table', () => {
+  describe('Delete Table', function () {
+    this.retries(3);
     const datasetId = `gcloud_tests_${uuid.v4()}`.replace(/-/gi, '_');
     const tableId = `gcloud_tests_${uuid.v4()}`.replace(/-/gi, '_');