update controller tests

Author: kelvin-klchu

packages/core/src/test/amazonqFeatureDev/controllers/chat/controller.test.ts

@@ -258,6 +258,14 @@ describe('Controller', () => {
 
             assertTelemetry('amazonq_endChat', { amazonqConversationId: conversationID, result: 'Succeeded' })
         })
+
+        it('disableFileList is called', async () => {
+            const disableFileListSpy = sinon.stub(session, 'disableFileList').resolves()
+
+            await newTaskClicked()
+
+            assert.ok(disableFileListSpy.calledOnce)
+        })
     })
 
     describe('fileClicked', () => {
@@ -437,11 +445,13 @@ describe('Controller', () => {
                 const sendAnswerSpy = sinon.stub(controllerSetup.messenger, 'sendAnswer')
                 const sendErrorMessageSpy = sinon.stub(controllerSetup.messenger, 'sendErrorMessage')
                 const sendMonthlyLimitErrorSpy = sinon.stub(controllerSetup.messenger, 'sendMonthlyLimitError')
+                const disableFileListSpy = sinon.stub(session, 'disableFileList').resolves()
 
                 await fireChatMessage()
 
                 switch (error.constructor.name) {
                     case ContentLengthError.name:
+                        assert.ok(disableFileListSpy.calledOnce)
                         assert.ok(
                             sendAnswerSpy.calledWith({
                                 type: 'answer',
@@ -452,13 +462,15 @@ describe('Controller', () => {
                         )
                         break
                     case MonthlyConversationLimitError.name:
+                        assert.ok(disableFileListSpy.calledOnce)
                         assert.ok(sendMonthlyLimitErrorSpy.calledWith(tabID))
                         break
                     case FeatureDevServiceError.name:
                     case UploadCodeError.name:
                     case UserMessageNotFoundError.name:
                     case TabIdNotFoundError.name:
                     case PrepareRepoFailedError.name:
+                        assert.ok(disableFileListSpy.calledOnce)
                         assert.ok(
                             sendErrorMessageSpy.calledWith(
                                 createTestErrorMessage(error.message),
@@ -470,6 +482,7 @@ describe('Controller', () => {
                         break
                     case PromptRefusalException.name:
                     case ZipFileError.name:
+                        assert.ok(disableFileListSpy.calledOnce)
                         assert.ok(
                             sendErrorMessageSpy.calledWith(
                                 createTestErrorMessage(error.message),
@@ -481,8 +494,19 @@ describe('Controller', () => {
                         )
                         break
                     case NoChangeRequiredException.name:
+                        assert.ok(disableFileListSpy.calledTwice)
+                        assert.ok(
+                            sendAnswerSpy.calledWith({
+                                type: 'answer',
+                                tabID,
+                                message: error.message,
+                                canBeVoted: true,
+                            })
+                        )
+                        break
                     case CodeIterationLimitError.name:
                     case UploadURLExpired.name:
+                        assert.ok(disableFileListSpy.calledOnce)
                         assert.ok(
                             sendAnswerSpy.calledWith({
                                 type: 'answer',
@@ -493,6 +517,7 @@ describe('Controller', () => {
                         )
                         break
                     default:
+                        assert.ok(disableFileListSpy.calledOnce)
                         assert.ok(
                             sendErrorMessageSpy.calledWith(
                                 i18n('AWS.amazonq.featureDev.error.codeGen.default'),
@@ -524,4 +549,36 @@ describe('Controller', () => {
             assertTelemetry('ui_click', { elementId: 'amazonq_stopCodeGeneration' })
         })
     })
+
+    describe('closeSession', async () => {
+        async function closeSessionClicked() {
+            const getSessionStub = sinon.stub(controllerSetup.sessionStorage, 'getSession').resolves(session)
+
+            controllerSetup.emitters.followUpClicked.fire({
+                tabID,
+                followUp: {
+                    type: FollowUpTypes.CloseSession,
+                },
+            })
+
+            // Wait until the controller has time to process the event
+            await waitUntil(() => {
+                return Promise.resolve(getSessionStub.callCount > 0)
+            }, {})
+        }
+
+        it('end chat telemetry is sent', async () => {
+            await closeSessionClicked()
+
+            assertTelemetry('amazonq_endChat', { amazonqConversationId: conversationID, result: 'Succeeded' })
+        })
+
+        it('disableFileList is called', async () => {
+            const disableFileListSpy = sinon.stub(session, 'disableFileList').resolves()
+
+            await closeSessionClicked()
+
+            assert.ok(disableFileListSpy.calledOnce)
+        })
+    })
 })