test: update /dev prepareRepoData unit test to include devfile filtering

Author: willyyhuang

packages/amazonq/test/unit/amazonqFeatureDev/util/files.test.ts

@@ -14,6 +14,7 @@ import { assertTelemetry, createTestWorkspace } from 'aws-core-vscode/test'
 import { fs, AmazonqCreateUpload } from 'aws-core-vscode/shared'
 import { Span } from 'aws-core-vscode/telemetry'
 import sinon from 'sinon'
+import { CodeWhispererSettings } from 'aws-core-vscode/codewhisperer'
 
 describe('file utils', () => {
     describe('prepareRepoData', function () {
@@ -56,6 +57,64 @@ describe('file utils', () => {
             assertTelemetry('amazonq_bundleExtensionIgnored', { filenameExt: 'mp4', count: 1 })
         })
 
+        it('prepareRepoData should ignore devfile.yaml when setting is disabled', async function () {
+            const fileAmount = 2
+            const fileNamePrefix = 'file'
+            const fileNameSuffix = '.md'
+            const fileContent = 'test content'
+
+            const workspace = await createTestWorkspace(fileAmount, {
+                fileNamePrefix,
+                fileContent,
+                fileNameSuffix,
+                additionalFiles: [
+                    {
+                        fileContent: 'test',
+                        fileName: 'devfile.yaml',
+                    },
+                ],
+            })
+
+            const telemetry = new TelemetryHelper()
+            const result = await prepareRepoData([workspace.uri.fsPath], [workspace], telemetry, {
+                record: () => {},
+            } as unknown as Span<AmazonqCreateUpload>)
+            assert.strictEqual(Buffer.isBuffer(result.zipFileBuffer), true)
+            // checksum is not the same across different test executions because some unique random folder names are generated
+            assert.strictEqual(result.zipFileChecksum.length, 44)
+            assert.strictEqual(telemetry.repositorySize, 24)
+        })
+
+        it('prepareRepoData should ignore devfile.yaml when setting is enabled', async function () {
+            const fileAmount = 2
+            const fileNamePrefix = 'file'
+            const fileNameSuffix = '.md'
+            const fileContent = 'test content'
+
+            const workspace = await createTestWorkspace(fileAmount, {
+                fileNamePrefix,
+                fileContent,
+                fileNameSuffix,
+                additionalFiles: [
+                    {
+                        fileContent: 'test',
+                        fileName: 'devfile.yaml',
+                    },
+                ],
+            })
+
+            await CodeWhispererSettings.instance.updateDevCommandWorkspaceConfigurations(workspace.uri.fsPath, true)
+            const telemetry = new TelemetryHelper()
+            console.log(workspace.uri.fsPath)
+            const result = await prepareRepoData([workspace.uri.fsPath], [workspace], telemetry, {
+                record: () => {},
+            } as unknown as Span<AmazonqCreateUpload>)
+            assert.strictEqual(Buffer.isBuffer(result.zipFileBuffer), true)
+            // checksum is not the same across different test executions because some unique random folder names are generated
+            assert.strictEqual(result.zipFileChecksum.length, 44)
+            assert.strictEqual(telemetry.repositorySize, 28)
+        })
+
         // Test the logic that allows the customer to modify root source folder
         it('prepareRepoData throws a ContentLengthError code when repo is too big', async function () {
             const workspace = await createTestWorkspace(1, {})

packages/core/src/test/testUtil.ts

@@ -185,6 +185,10 @@ export async function createTestWorkspace(
          * optional file name suffix
          */
         fileNameSuffix?: string
+        /**
+         * optional additional files
+         */
+        additionalFiles?: Array<{ fileName: string; fileContent: string }>
     }
 ): Promise<vscode.WorkspaceFolder> {
     const workspace = await createTestWorkspaceFolder(opts.workspaceName, opts.subDir)
@@ -202,6 +206,12 @@ export async function createTestWorkspace(
         await fs.writeFile(tempFilePath, fileContent)
     } while (--n > 0)
 
+    if (opts.additionalFiles) {
+        opts.additionalFiles.forEach(async (file) => {
+            await fs.writeFile(path.join(workspace.uri.fsPath, file.fileName), file.fileContent)
+        })
+    }
+
     return workspace
 }