Remove obsolete testing function

Author: opieter-aws

packages/core/src/testE2E/codewhisperer/referenceTracker.test.ts

@@ -6,11 +6,12 @@
 import assert from 'assert'
 import * as codewhispererClient from '../../codewhisperer/client/codewhisperer'
 import { ConfigurationEntry } from '../../codewhisperer/models/model'
-import { setValidConnection, skipTestIfNoValidConn } from '../util/connection'
+import { skipTestIfNoValidConn } from '../util/connection'
 import { RecommendationHandler } from '../../codewhisperer/service/recommendationHandler'
 import { createMockTextEditor, resetCodeWhispererGlobalVariables } from '../../test/codewhisperer/testUtil'
 import { invokeRecommendation } from '../../codewhisperer/commands/invokeRecommendation'
 import { session } from '../../codewhisperer/util/codeWhispererSession'
+import { AuthUtil } from '../../codewhisperer/util/authUtil'
 
 /*
 New model deployment may impact references returned. 
@@ -49,7 +50,7 @@ describe('CodeWhisperer service invocation', async function () {
     }
 
     before(async function () {
-        validConnection = await setValidConnection()
+        validConnection = AuthUtil.instance.isConnected()
     })
 
     beforeEach(function () {

packages/core/src/testE2E/codewhisperer/securityScan.test.ts

@@ -9,7 +9,7 @@ import * as codewhispererClient from '../../codewhisperer/client/codewhisperer'
 import * as CodeWhispererConstants from '../../codewhisperer/models/constants'
 import * as path from 'path'
 import * as testutil from '../../test/testUtil'
-import { setValidConnection, skipTestIfNoValidConn } from '../util/connection'
+import { skipTestIfNoValidConn } from '../util/connection'
 import { resetCodeWhispererGlobalVariables } from '../../test/codewhisperer/testUtil'
 import { getTestWorkspaceFolder } from '../../testInteg/integrationTestsUtilities'
 import { closeAllEditors } from '../../test/testUtil'
@@ -23,6 +23,7 @@ import { makeTemporaryToolkitFolder } from '../../shared/filesystemUtilities'
 import fs from '../../shared/fs/fs'
 import { ZipUtil } from '../../codewhisperer/util/zipUtil'
 import { randomUUID } from '../../shared/crypto'
+import { AuthUtil } from '../../codewhisperer'
 
 const filePromptWithSecurityIssues = `from flask import app
 
@@ -53,7 +54,7 @@ describe('CodeWhisperer security scan', async function () {
     const workspaceFolder = getTestWorkspaceFolder()
 
     before(async function () {
-        validConnection = await setValidConnection()
+        validConnection = AuthUtil.instance.isConnected()
     })
 
     beforeEach(function () {

packages/core/src/testE2E/codewhisperer/serviceInvocations.test.ts

@@ -6,7 +6,7 @@
 import assert from 'assert'
 import * as vscode from 'vscode'
 import * as path from 'path'
-import { setValidConnection, skipTestIfNoValidConn } from '../util/connection'
+import { skipTestIfNoValidConn } from '../util/connection'
 import { ConfigurationEntry } from '../../codewhisperer/models/model'
 import * as codewhispererClient from '../../codewhisperer/client/codewhisperer'
 import { RecommendationHandler } from '../../codewhisperer/service/recommendationHandler'
@@ -20,6 +20,7 @@ import { sleep } from '../../shared/utilities/timeoutUtils'
 import { invokeRecommendation } from '../../codewhisperer/commands/invokeRecommendation'
 import { getTestWorkspaceFolder } from '../../testInteg/integrationTestsUtilities'
 import { session } from '../../codewhisperer/util/codeWhispererSession'
+import { AuthUtil } from '../../codewhisperer/util/authUtil'
 
 describe('CodeWhisperer service invocation', async function () {
     let validConnection: boolean
@@ -32,7 +33,7 @@ describe('CodeWhisperer service invocation', async function () {
     }
 
     before(async function () {
-        validConnection = await setValidConnection()
+        validConnection = AuthUtil.instance.isConnected()
     })
 
     beforeEach(function () {

packages/core/src/testE2E/util/connection.ts

@@ -3,8 +3,6 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import { AuthUtil } from '../../codewhisperer'
-
 /*
 In order to run codewhisperer or gumby integration tests user must:
    
@@ -15,10 +13,6 @@ Test cases will skip if the above criteria are not met.
 If user has an expired connection they must reauthenticate prior to running tests.
 */
 
-export async function setValidConnection() {
-    return AuthUtil.instance.isConnected()
-}
-
 export function skipTestIfNoValidConn(validConnection: boolean, ctx: Mocha.Context) {
     if (!validConnection && ctx.currentTest) {
         ctx.currentTest.title += ` (skipped - no valid connection)`

packages/core/src/testInteg/amazonQTransform/transformByQ.test.ts

@@ -13,18 +13,16 @@ import request from '../../shared/request'
 import { transformByQState, ZipManifest } from '../../codewhisperer/models/model'
 import globals from '../../shared/extensionGlobals'
 import { fs } from '../../shared'
-import { setValidConnection } from '../../testE2E/util/connection'
+import { AuthUtil } from '../../codewhisperer/util/authUtil'
 
 describe('transformByQ', async function () {
     let tempDir = ''
     let tempFileName = ''
     let tempFilePath = ''
     let zippedCodePath = ''
-    let validConnection: boolean
 
     before(async function () {
-        validConnection = await setValidConnection()
-        if (!validConnection) {
+        if (!AuthUtil.instance.isConnected()) {
             this.skip()
         }
         tempDir = path.join(os.tmpdir(), 'gumby-test')