fix(amazonq): Fix showSsoPrompt unit tests (aws#7236)

## Problem
`showConnectionPrompt` unit tests were failing

## Solution
* Ensure DeviceFlow is used for auth
* Stub login
* Assert whether the login was called with builder ID instead of if the
log line was emitted

---

- Treat all work as PUBLIC. Private `feature/x` branches will not be
squash-merged at release time.
- Your code changes must meet the guidelines in
[CONTRIBUTING.md](https://github.com/aws/aws-toolkit-vscode/blob/master/CONTRIBUTING.md#guidelines).
- License: I confirm that my contribution is made under the terms of the
Apache 2.0 license.

Author: opieter-aws

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

@@ -7,8 +7,9 @@ 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
@@ -17,13 +18,18 @@ describe('showConnectionPrompt', function () {
         await resetCodeWhispererGlobalVariables()
         isBuilderIdConnection = sinon.stub(AuthUtil.instance, 'isBuilderIdConnection')
         isBuilderIdConnection.resolves()
+
+        // Stub useDeviceFlow so we always use DeviceFlow for auth
+        sinon.stub(SsoAccessTokenProvider, 'useDeviceFlow').returns(true)
     })
 
     afterEach(function () {
         sinon.restore()
     })
 
     it('can select connect to AwsBuilderId', async function () {
+        sinon.stub(AuthUtil.instance, 'login').resolves()
+
         getTestWindow().onDidShowQuickPick(async (picker) => {
             await picker.untilReady()
             picker.acceptItem(picker.items[0])
@@ -36,12 +42,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'