From eae3e31e681939d70f8e646cfb446e2bf02fe88d Mon Sep 17 00:00:00 2001
From: Justin Wilaby <jwilaby@salesforce.com>
Date: Fri, 15 Mar 2024 12:02:08 -0700
Subject: [PATCH] chore(W-15263489): improve unit test performance

---
 packages/cli/src/global_telemetry.ts                  | 11 +++++++----
 .../cli/src/hooks/postrun/performance_analytics.ts    | 10 ++++++----
 packages/cli/src/hooks/prerun/analytics.ts            |  4 ++++
 packages/cli/test/acceptance/smoke.acceptance.test.ts |  1 -
 packages/cli/test/helpers/init.js                     |  8 ++++++++
 5 files changed, 25 insertions(+), 9 deletions(-)

diff --git a/packages/cli/src/global_telemetry.ts b/packages/cli/src/global_telemetry.ts
index a0bb0362fb..c047d81afd 100644
--- a/packages/cli/src/global_telemetry.ts
+++ b/packages/cli/src/global_telemetry.ts
@@ -13,9 +13,12 @@ const {version} = require('../package.json')
 
 const root = path.resolve(__dirname, '../package.json')
 const isDev = process.env.IS_DEV_ENVIRONMENT === 'true'
-const config = new Config({root})
-const heroku = new APIClient(config)
-const token = heroku.auth
+
+function getToken() {
+  const config = new Config({root})
+  const heroku = new APIClient(config)
+  return heroku.auth
+}
 
 const debug = require('debug')('global_telemetry')
 
@@ -44,7 +47,7 @@ const provider = new NodeTracerProvider({
   resource,
 })
 
-const headers = {Authorization: `Bearer ${token}`}
+const headers = {Authorization: `Bearer ${process.env.IS_HEROKU_TEST_ENV !== 'true' ? getToken() : ''}`}
 
 const exporter = new OTLPTraceExporter({
   url: isDev ? 'https://backboard-staging.herokuapp.com/otel/v1/traces' : 'https://backboard.heroku.com/otel/v1/traces',
diff --git a/packages/cli/src/hooks/postrun/performance_analytics.ts b/packages/cli/src/hooks/postrun/performance_analytics.ts
index d178c50413..97f925d62d 100644
--- a/packages/cli/src/hooks/postrun/performance_analytics.ts
+++ b/packages/cli/src/hooks/postrun/performance_analytics.ts
@@ -5,11 +5,13 @@ import * as telemetry from '../../global_telemetry'
 declare const global: telemetry.TelemetryGlobal
 
 const performance_analytics: Hook<'postrun'> = async function () {
-  if (global.cliTelemetry) {
-    const cmdStartTime = global.cliTelemetry.commandRunDuration
-    global.cliTelemetry.commandRunDuration = telemetry.computeDuration(cmdStartTime)
-    global.cliTelemetry.lifecycleHookCompletion.postrun = true
+  if (process.env.IS_HEROKU_TEST_ENV === 'true' || !global.cliTelemetry) {
+    return
   }
+
+  const cmdStartTime = global.cliTelemetry.commandRunDuration
+  global.cliTelemetry.commandRunDuration = telemetry.computeDuration(cmdStartTime)
+  global.cliTelemetry.lifecycleHookCompletion.postrun = true
 }
 
 export default performance_analytics
diff --git a/packages/cli/src/hooks/prerun/analytics.ts b/packages/cli/src/hooks/prerun/analytics.ts
index 2bd7dbaea3..50a585bf7b 100644
--- a/packages/cli/src/hooks/prerun/analytics.ts
+++ b/packages/cli/src/hooks/prerun/analytics.ts
@@ -6,6 +6,10 @@ import * as telemetry from '../../global_telemetry'
 declare const global: telemetry.TelemetryGlobal
 
 const analytics: Hook<'prerun'> = async function (options) {
+  if (process.env.IS_HEROKU_TEST_ENV === 'true') {
+    return
+  }
+
   global.cliTelemetry = telemetry.setupTelemetry(this.config, options)
   const analytics = new Analytics(this.config)
   await analytics.record(options)
diff --git a/packages/cli/test/acceptance/smoke.acceptance.test.ts b/packages/cli/test/acceptance/smoke.acceptance.test.ts
index 7dbdb206c6..d6dd1290c7 100755
--- a/packages/cli/test/acceptance/smoke.acceptance.test.ts
+++ b/packages/cli/test/acceptance/smoke.acceptance.test.ts
@@ -1,5 +1,4 @@
 // tslint:disable no-console
-
 import * as fs from 'fs-extra'
 import {expect} from 'chai'
 import * as path from 'path'
diff --git a/packages/cli/test/helpers/init.js b/packages/cli/test/helpers/init.js
index f32a43dae7..8ae1cd0f38 100644
--- a/packages/cli/test/helpers/init.js
+++ b/packages/cli/test/helpers/init.js
@@ -1,5 +1,13 @@
 const path = require('path')
+
+globalThis.setInterval = () => ({unref: () => {}})
+const tm = globalThis.setTimeout
+globalThis.setTimeout = cb => {
+  return tm(cb)
+}
+
 process.env.TS_NODE_PROJECT = path.resolve('test/tsconfig.json')
+process.env.IS_HEROKU_TEST_ENV = 'true'
 
 let nock = require('nock')
 nock.disableNetConnect()