add logging

Author: Hweinstock

packages/core/src/applicationcomposer/messageHandlers/generateResourceHandler.ts

@@ -18,14 +18,15 @@ import {
 } from '../types'
 import globals from '../../shared/extensionGlobals'
 import { getLogger } from '../../shared/logger/logger'
-import { AmazonqNotFoundError, getAmazonqApi } from '../../amazonq/extApi'
+import * as extApi from '../../amazonq/extApi'
 
 const TIMEOUT = 30_000
 
 export async function generateResourceHandler(request: GenerateResourceRequestMessage, context: WebviewContext) {
     try {
         const { chatResponse, references, metadata, isSuccess } = await generateResource(request.cfnType)
-
+        // eslint-disable-next-line aws-toolkits/no-console-log
+        console.log('post-generateResource')
         const responseMessage: GenerateResourceResponseMessage = {
             command: Command.GENERATE_RESOURCE,
             messageType: MessageType.RESPONSE,
@@ -58,9 +59,11 @@ async function generateResource(cfnType: string) {
     let startTime = globals.clock.Date.now()
 
     try {
-        const amazonqApi = await getAmazonqApi()
+        const amazonqApi = await extApi.getAmazonqApi()
+        // eslint-disable-next-line aws-toolkits/no-console-log
+        console.log('post-getAmazonQApi')
         if (!amazonqApi) {
-            throw new AmazonqNotFoundError()
+            throw new extApi.AmazonqNotFoundError()
         }
         const request: GenerateAssistantResponseRequest = {
             conversationState: {
@@ -81,6 +84,8 @@ async function generateResource(cfnType: string) {
         let references: Reference[] = []
 
         await amazonqApi.authApi.reauthIfNeeded()
+        // eslint-disable-next-line aws-toolkits/no-console-log
+        console.log('post-reauthIfNeeded')
 
         startTime = globals.clock.Date.now()
         // TODO-STARLING - Revisit to see if timeout still needed prior to launch

packages/core/src/test/applicationcomposer/messageHandlers/generateResourceHandler.test.ts

@@ -8,29 +8,35 @@ import sinon from 'sinon'
 import { createTemplate, createWebviewContext } from '../utils'
 import { generateResourceHandler } from '../../../applicationcomposer/messageHandlers/generateResourceHandler'
 import { Command, MessageType } from '../../../applicationcomposer/types'
+import * as extApi from '../../../amazonq/extApi'
 
 describe('generateResourceHandler', function () {
     afterEach(() => {
         sinon.restore()
     })
-
-    it('amazon q is not installed', async () => {
-        const panel = await createTemplate()
-        const postMessageSpy = sinon.spy(panel.webview, 'postMessage')
-        const context = await createWebviewContext({
-            panel,
+    for (const _ of Array.from({ length: 20 }, (i) => i)) {
+        it('amazon q is not installed', async () => {
+            sinon.stub(extApi, 'getAmazonqApi')
+            const panel = await createTemplate()
+            console.log('post-createTemplate')
+            const postMessageSpy = sinon.spy(panel.webview, 'postMessage')
+            const context = await createWebviewContext({
+                panel,
+            })
+            console.log('post-createWebviewContext')
+            await generateResourceHandler(
+                {
+                    command: Command.GENERATE_RESOURCE,
+                    messageType: MessageType.REQUEST,
+                    cfnType: '',
+                    prompt: '',
+                    traceId: '0',
+                },
+                context
+            )
+            console.log('post-generateResourceHandler')
+            assert.ok(postMessageSpy.calledOnce)
+            assert.deepStrictEqual(postMessageSpy.getCall(0).args[0].isSuccess, false)
         })
-        await generateResourceHandler(
-            {
-                command: Command.GENERATE_RESOURCE,
-                messageType: MessageType.REQUEST,
-                cfnType: '',
-                prompt: '',
-                traceId: '0',
-            },
-            context
-        )
-        assert.ok(postMessageSpy.calledOnce)
-        assert.deepStrictEqual(postMessageSpy.getCall(0).args[0].isSuccess, false)
-    })
+    }
 })