Fix unit tests

Author: opieter-aws

packages/amazonq/test/unit/codewhisperer/region/regionProfileManager.test.ts

@@ -37,7 +37,7 @@ describe('RegionProfileManager', async function () {
         }
     }
 
-    beforeEach(async function () {
+    beforeEach(function () {
         regionProfileManager = new RegionProfileManager(AuthUtil.instance)
     })
 

packages/amazonq/test/unit/codewhisperer/service/recommendationHandler.test.ts

@@ -15,6 +15,7 @@ import {
     RecommendationHandler,
     CodeWhispererCodeCoverageTracker,
     supplementalContextUtil,
+    AuthUtil,
 } from 'aws-core-vscode/codewhisperer'
 import {
     assertTelemetryCurried,
@@ -24,8 +25,6 @@ import {
 } from 'aws-core-vscode/test'
 // import * as supplementalContextUtil from 'aws-core-vscode/codewhisperer'
 
-const enterpriseSsoStartUrl = 'https://enterprise.awsapps.com/start'
-
 describe('recommendationHandler', function () {
     const config: ConfigurationEntry = {
         isShowMethodsEnabled: true,
@@ -142,7 +141,7 @@ describe('recommendationHandler', function () {
                 codewhispererLineNumber: 1,
                 codewhispererCursorOffset: 38,
                 codewhispererLanguage: 'python',
-                credentialStartUrl: enterpriseSsoStartUrl,
+                credentialStartUrl: AuthUtil.instance.connection?.startUrl,
                 codewhispererSupplementalContextIsUtg: false,
                 codewhispererSupplementalContextTimeout: false,
                 codewhispererSupplementalContextLatency: 0,

packages/amazonq/test/unit/codewhisperer/tracker/codewhispererTracker.test.ts

@@ -5,7 +5,7 @@
 
 import assert from 'assert'
 import * as sinon from 'sinon'
-import { assertTelemetryCurried, createTestAuthUtil } from 'aws-core-vscode/test'
+import { assertTelemetryCurried } from 'aws-core-vscode/test'
 import { AuthUtil, CodeWhispererTracker } from 'aws-core-vscode/codewhisperer'
 import { resetCodeWhispererGlobalVariables, createAcceptedSuggestionEntry } from 'aws-core-vscode/test'
 import { globals } from 'aws-core-vscode/shared'
@@ -82,9 +82,6 @@ describe('codewhispererTracker', function () {
 
     describe('emitTelemetryOnSuggestion', function () {
         it('Should call recordCodewhispererUserModification with suggestion event', async function () {
-            await createTestAuthUtil()
-            const testStartUrl = 'testStartUrl'
-            sinon.stub(AuthUtil.instance.connection!, 'startUrl').value(testStartUrl)
             const suggestion = createAcceptedSuggestionEntry()
             const assertTelemetry = assertTelemetryCurried('codewhisperer_userModification')
             await CodeWhispererTracker.getTracker().emitTelemetryOnSuggestion(suggestion)
@@ -96,7 +93,7 @@ describe('codewhispererTracker', function () {
                 codewhispererModificationPercentage: 1,
                 codewhispererCompletionType: 'Line',
                 codewhispererLanguage: 'java',
-                credentialStartUrl: testStartUrl,
+                credentialStartUrl: AuthUtil.instance.connection?.startUrl,
                 codewhispererCharactersAccepted: suggestion.originalString.length,
                 codewhispererCharactersModified: 0,
             })