fix(amazonq): Fix showSsoPrompt unit tests

Author: opieter-aws

packages/amazonq/test/unit/codewhisperer/util/showSsoPrompt.test.ts

@@ -7,16 +7,21 @@ import * as vscode from 'vscode'
 import assert from 'assert'
 import * as sinon from 'sinon'
 import { resetCodeWhispererGlobalVariables } from 'aws-core-vscode/test'
-import { assertTelemetryCurried, getTestWindow, getTestLogger } from 'aws-core-vscode/test'
+import { assertTelemetryCurried, getTestWindow } from 'aws-core-vscode/test'
 import { AuthUtil, awsIdSignIn, showCodeWhispererConnectionPrompt } from 'aws-core-vscode/codewhisperer'
+import { SsoAccessTokenProvider, constants } from 'aws-core-vscode/auth'
 
 describe('showConnectionPrompt', function () {
     let isBuilderIdConnection: sinon.SinonStub
+    let useDeviceFlowStub: sinon.SinonStub
 
     beforeEach(async function () {
         await resetCodeWhispererGlobalVariables()
         isBuilderIdConnection = sinon.stub(AuthUtil.instance, 'isBuilderIdConnection')
         isBuilderIdConnection.resolves()
+
+        // Stub useDeviceFlow so we always use DeviceFlow for auth
+        useDeviceFlowStub = sinon.stub(SsoAccessTokenProvider, 'useDeviceFlow').returns(true)
     })
 
     afterEach(function () {
@@ -36,12 +41,13 @@ describe('showConnectionPrompt', function () {
         assert.ok(isBuilderIdConnection)
     })
 
-    it('connectToAwsBuilderId logs that AWS ID sign in was selected', async function () {
+    it('connectToAwsBuilderId calls AuthUtil login with builderIdStartUrl', async function () {
         sinon.stub(vscode.commands, 'executeCommand')
+        const loginStub = sinon.stub(AuthUtil.instance, 'login').resolves()
 
         await awsIdSignIn()
 
-        const loggedEntries = getTestLogger().getLoggedEntries()
-        assert.ok(loggedEntries.find((entry) => entry === 'selected AWS ID sign in'))
+        assert.strictEqual(loginStub.called, true)
+        assert.strictEqual(loginStub.firstCall.args[0], constants.builderIdStartUrl)
     })
 })

packages/core/src/auth/index.ts

@@ -25,3 +25,4 @@ export { LoginManager } from './deprecated/loginManager'
 export * as constants from './sso/constants'
 export * as authUtils from './utils'
 export * as auth2 from './auth2'
+export * as SsoAccessTokenProvider from './sso/ssoAccessTokenProvider'