fix(run/logging): make tests more robust

Author: ace-n

run/logging-manual/package.json

@@ -20,6 +20,7 @@
   },
   "devDependencies": {
     "@google-cloud/logging": "^10.0.0",
+    "child-process-promise": "^2.2.1",
     "google-auth-library": "^7.0.0",
     "mocha": "^9.0.0"
   }

run/logging-manual/test/system.test.js

@@ -15,7 +15,7 @@
 const assert = require('assert');
 const request = require('got');
 const {Logging} = require('@google-cloud/logging');
-const {execSync} = require('child_process');
+const {exec} = require('child-process-promise');
 const {GoogleAuth} = require('google-auth-library');
 const auth = new GoogleAuth();
 
@@ -66,6 +66,30 @@ function sleep(ms) {
   return new Promise(resolve => setTimeout(resolve, ms));
 }
 
+async function runShellCmd(cmd, opts = {}) {
+  const MAX_ATTEMPTS = 10;
+  let attempt = 0;
+  while (attempt < MAX_ATTEMPTS) {
+    try {
+      console.log('Running command:', cmd);
+      const result = await exec(cmd, opts);
+      return result;
+    } catch (err) {
+      console.log('Shell command failed!');
+      console.log('\tCommand:', cmd);
+      console.log('\tError:', err);
+      attempt += 1;
+
+      if (attempt < MAX_ATTEMPTS) {
+        // Exponential backoff
+        await sleep(attempt ** 2 * 1000);
+      } else {
+        throw err;
+      }
+    }
+  }
+}
+
 describe('Logging', () => {
   let requestLog;
   let sampleLog;
@@ -95,15 +119,21 @@ describe('Logging', () => {
         `--substitutions _SERVICE=${SERVICE_NAME},_PLATFORM=${PLATFORM},_REGION=${REGION}`;
       if (SAMPLE_VERSION) buildCmd += `,_VERSION=${SAMPLE_VERSION}`;
 
-      console.log('Starting Cloud Build...');
-      execSync(buildCmd, {timeout: 240000}); // timeout at 4 mins
-      console.log('Cloud Build completed.');
+      console.log('Starting initial Cloud Build...');
+      try {
+        runShellCmd(buildCmd, {timeout: 240000}); // timeout at 4 mins
+        console.log('Initial Cloud Build completed.');
+      } catch (err) {
+        console.log('Initial Cloud Build failed:', err);
+        throw err;
+      }
 
       // Retrieve URL of Cloud Run service
-      const url = execSync(
+      const url = await runShellCmd(
         `gcloud run services describe ${SERVICE_NAME} --project=${GOOGLE_CLOUD_PROJECT} ` +
           `--platform=${PLATFORM} --region=${REGION} --format='value(status.url)'`
-      );
+      ).stdout;
+      console.log('Read URL:', url);
 
       BASE_URL = url.toString('utf-8').trim();
       if (!BASE_URL) throw Error('Cloud Run service URL not found');
@@ -122,7 +152,14 @@ describe('Logging', () => {
         `--substitutions _SERVICE=${SERVICE_NAME},_PLATFORM=${PLATFORM},_REGION=${REGION}`;
       if (SAMPLE_VERSION) cleanUpCmd += `,_VERSION=${SAMPLE_VERSION}`;
 
-      execSync(cleanUpCmd);
+      console.log('Cleaning up via Cloud Build.');
+      try {
+        runShellCmd(cleanUpCmd);
+        console.log('Cleanup Cloud Build completed.');
+      } catch (err) {
+        console.log('Cleanup Cloud Build failed:', err);
+        throw err;
+      }
     });
 
     it('can be reached by an HTTP request', async () => {
@@ -152,6 +189,7 @@ describe('Logging', () => {
       let attempt = 0;
       const maxAttempts = 5;
       while ((!requestLog || !sampleLog) && attempt < maxAttempts) {
+        console.log(`Fetching logs, attempt #${attempt}`);
         await sleep(attempt * 30000); // Linear backoff between retry attempts
         // Filter by service name over the last 5 minutes
         const filter = `resource.labels.service_name="${service_name}" timestamp>="${dateMinutesAgo(