Fixing test cases

Author: laileni-aws

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

@@ -8,7 +8,7 @@ import vscode from 'vscode'
 import sinon from 'sinon'
 import { join } from 'path'
 import { getTestWorkspaceFolder } from 'aws-core-vscode/test'
-import { CodeAnalysisScope, ZipUtil } from 'aws-core-vscode/codewhisperer'
+import { CodeAnalysisScope, FeatureUseCase, ZipUtil } from 'aws-core-vscode/codewhisperer'
 import { codeScanTruncDirPrefix } from 'aws-core-vscode/codewhisperer'
 import { ToolkitError } from 'aws-core-vscode/shared'
 import { LspClient } from 'aws-core-vscode/amazonq'
@@ -116,6 +116,8 @@ describe('zipUtil', function () {
         })
 
         it('should handle path with repeated project name for file scan', async function () {
+            const mockCodeScan = 'CODE_SCAN'
+            sinon.stub(FeatureUseCase, 'CODE_SCAN').value(mockCodeScan)
             const zipMetadata = await zipUtil.generateZip(
                 vscode.Uri.file(appCodePathWithRepeatedProjectName),
                 CodeAnalysisScope.FILE_ON_DEMAND
@@ -125,9 +127,19 @@ describe('zipUtil', function () {
             const zip = await JSZip.loadAsync(zipFileData)
             const files = Object.keys(zip.files)
             assert.ok(files.includes(join('workspaceFolder', 'workspaceFolder', 'App.java')))
+
+            // Verify that the mocked FeatureUseCase.CODE_SCAN was used
+            assert.ok(zipMetadata.rootDir.includes(mockCodeScan))
+            assert.ok(zipMetadata.zipFilePath.includes(mockCodeScan))
+
+            // Restore the original FeatureUseCase.CODE_SCAN
+            sinon.restore()
         })
 
         it('should handle path with repeated project name for project scan', async function () {
+            const mockCodeScan = 'CODE_SCAN'
+            sinon.stub(FeatureUseCase, 'CODE_SCAN').value(mockCodeScan)
+
             const zipMetadata = await zipUtil.generateZip(
                 vscode.Uri.file(appCodePathWithRepeatedProjectName),
                 CodeAnalysisScope.PROJECT
@@ -137,6 +149,11 @@ describe('zipUtil', function () {
             const zip = await JSZip.loadAsync(zipFileData)
             const files = Object.keys(zip.files)
             assert.ok(files.includes(join('workspaceFolder', 'workspaceFolder', 'App.java')))
+
+            assert.ok(zipMetadata.rootDir.includes(mockCodeScan))
+            assert.ok(zipMetadata.zipFilePath.includes(mockCodeScan))
+
+            sinon.restore()
         })
     })