reduce code dupe, merge settings

Author: Hweinstock

packages/core/src/test/amazonqFeatureDev/prepareRepoData.test.ts

@@ -14,99 +14,60 @@ type resultType = {
     zipFileChecksum: string
 }
 
-describe('prepareRepoData', function () {
-    describe('Performance Tests', function () {
-        function verifyResult(result: resultType, telemetry: TelemetryHelper, expectedSize: number): void {
-            assert.ok(result)
-            assert.strictEqual(Buffer.isBuffer(result.zipFileBuffer), true)
-            assert.strictEqual(telemetry.repositorySize, expectedSize)
-            assert.strictEqual(result.zipFileChecksum.length, 44)
-        }
-        /**
-         * Tests 250 files w/ 10 bytes each.
-         * Running more files can lead to flaky test from timeout.
-         */
-        performanceTest(
-            {
-                testRuns: 10,
-                linux: {
-                    userCpuUsage: 90,
-                    systemCpuUsage: 35,
-                    heapTotal: 4,
-                },
-                darwin: {
-                    userCpuUsage: 90,
-                    systemCpuUsage: 35,
-                    heapTotal: 4,
-                },
-                win32: {
-                    userCpuUsage: 90,
-                    systemCpuUsage: 35,
-                    heapTotal: 4,
-                },
+function performanceTestWrapper(numFiles: number, fileSize: number) {
+    return performanceTest(
+        {
+            testRuns: 10,
+            linux: {
+                userCpuUsage: 100,
+                systemCpuUsage: 35,
+                heapTotal: 4,
             },
-            'handles many files',
-            function () {
-                const telemetry = new TelemetryHelper()
-                return {
-                    setup: async () => {
-                        return await createTestWorkspace(250, {
-                            fileNamePrefix: 'file',
-                            fileContent: '0123456789',
-                            fileNameSuffix: '.md',
-                        })
-                    },
-                    execute: async (workspace: WorkspaceFolder) => {
-                        return await prepareRepoData([workspace.uri.fsPath], [workspace], telemetry, {
-                            record: () => {},
-                        } as unknown as Metric<AmazonqCreateUpload>)
-                    },
-                    verify: async (_w: WorkspaceFolder, result: resultType) => verifyResult(result, telemetry, 2500),
-                }
-            }
-        )
-        /**
-         * Runs 10 files of size 1000 bytes.
-         */
-        performanceTest(
-            {
-                testRuns: 10,
-                linux: {
-                    userCpuUsage: 65,
-                    systemCpuUsage: 30,
-                    heapTotal: 1,
+            darwin: {
+                userCpuUsage: 100,
+                systemCpuUsage: 35,
+                heapTotal: 4,
+            },
+            win32: {
+                userCpuUsage: 100,
+                systemCpuUsage: 35,
+                heapTotal: 4,
+            },
+        },
+        `handles ${numFiles} files of size ${fileSize} bytes`,
+        function () {
+            const telemetry = new TelemetryHelper()
+            return {
+                setup: async () => {
+                    return await createTestWorkspace(numFiles, {
+                        fileNamePrefix: 'file',
+                        fileContent: getRandomString(fileSize),
+                        fileNameSuffix: '.md',
+                    })
                 },
-                darwin: {
-                    userCpuUsage: 50,
-                    systemCpuUsage: 25,
-                    heapTotal: 1,
+                execute: async (workspace: WorkspaceFolder) => {
+                    return await prepareRepoData([workspace.uri.fsPath], [workspace], telemetry, {
+                        record: () => {},
+                    } as unknown as Metric<AmazonqCreateUpload>)
                 },
-                win32: {
-                    userCpuUsage: 60,
-                    systemCpuUsage: 30,
-                    heapTotal: 1,
+                verify: async (_w: WorkspaceFolder, result: resultType) => {
+                    verifyResult(result, telemetry, numFiles * fileSize)
                 },
-            },
-
-            'handles large files',
-            function () {
-                const telemetry = new TelemetryHelper()
-                return {
-                    setup: async () => {
-                        return await createTestWorkspace(10, {
-                            fileNamePrefix: 'file',
-                            fileContent: getRandomString(1000),
-                            fileNameSuffix: '.md',
-                        })
-                    },
-                    execute: async (workspace: WorkspaceFolder) => {
-                        return await prepareRepoData([workspace.uri.fsPath], [workspace], telemetry, {
-                            record: () => {},
-                        } as unknown as Metric<AmazonqCreateUpload>)
-                    },
-                    verify: async (_w: WorkspaceFolder, result: resultType) => verifyResult(result, telemetry, 10000),
-                }
             }
-        )
+        }
+    )
+}
+
+function verifyResult(result: resultType, telemetry: TelemetryHelper, expectedSize: number): void {
+    assert.ok(result)
+    assert.strictEqual(Buffer.isBuffer(result.zipFileBuffer), true)
+    assert.strictEqual(telemetry.repositorySize, expectedSize)
+    assert.strictEqual(result.zipFileChecksum.length, 44)
+}
+
+describe('prepareRepoData', function () {
+    describe('Performance Tests', function () {
+        performanceTestWrapper(250, 10)
+        performanceTestWrapper(10, 1000)
     })
 })