Skip to content

Commit 57dab4b

Browse files
committed
add test for many files
1 parent 2b60a35 commit 57dab4b

File tree

1 file changed

+73
-0
lines changed

1 file changed

+73
-0
lines changed
Lines changed: 73 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,73 @@
1+
/*!
2+
* Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
3+
* SPDX-License-Identifier: Apache-2.0
4+
*/
5+
import assert from 'assert'
6+
import { WorkspaceFolder } from 'vscode'
7+
import { performanceTest } from '../../shared/performance/performance'
8+
import { createTestWorkspace } from '../testUtil'
9+
import { prepareRepoData, TelemetryHelper } from '../../amazonqFeatureDev'
10+
import { AmazonqCreateUpload, Metric } from '../../shared'
11+
12+
describe('prepareRepoDataPerformanceTest', function () {
13+
let fileAmount: number
14+
let fileNamePrefix: string
15+
let fileNameSuffix: string
16+
let fileContent: string
17+
let workspace: WorkspaceFolder
18+
19+
beforeEach(async function () {})
20+
21+
afterEach(async function () {})
22+
23+
after(async function () {})
24+
25+
before(async function () {
26+
fileAmount = 1000
27+
fileNamePrefix = 'file'
28+
fileNameSuffix = '.md'
29+
fileContent = 'test content'
30+
31+
workspace = await createTestWorkspace(fileAmount, { fileNamePrefix, fileContent, fileNameSuffix })
32+
})
33+
34+
performanceTest(
35+
{
36+
testRuns: 10,
37+
linux: {
38+
userCpuUsage: 80,
39+
heapTotal: 8,
40+
duration: 0.7,
41+
},
42+
darwin: {
43+
userCpuUsage: 80,
44+
heapTotal: 8,
45+
duration: 0.7,
46+
},
47+
win32: {
48+
userCpuUsage: 80,
49+
heapTotal: 8,
50+
duration: 0.7,
51+
},
52+
},
53+
'handles many files',
54+
function () {
55+
const telemetry = new TelemetryHelper()
56+
let result: any
57+
return {
58+
setup: async () => {},
59+
execute: async () => {
60+
result = await prepareRepoData([workspace.uri.fsPath], [workspace], telemetry, {
61+
record: () => {},
62+
} as unknown as Metric<AmazonqCreateUpload>)
63+
},
64+
verify: async () => {
65+
assert.ok(result)
66+
assert.strictEqual(Buffer.isBuffer(result.zipFileBuffer), true)
67+
assert.strictEqual(telemetry.repositorySize, 12000)
68+
assert.strictEqual(result.zipFileChecksum.length, 44)
69+
},
70+
}
71+
}
72+
)
73+
})

0 commit comments

Comments
 (0)