tst

Author: Will-ShaoHua

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

@@ -7,7 +7,7 @@ import assert from 'assert'
 import * as vscode from 'vscode'
 import * as sinon from 'sinon'
 import * as crossFile from 'aws-core-vscode/codewhisperer'
-import { createMockTextEditor } from 'aws-core-vscode/test'
+import { aStringWithLineCount, createMockTextEditor } from 'aws-core-vscode/test'
 import { CodeWhispererUserGroupSettings, UserGroup, crossFileContextConfig } from 'aws-core-vscode/codewhisperer'
 import {
     assertTabCount,
@@ -36,19 +36,19 @@ describe('crossFileContextUtil', function () {
             tempFolder = (await createTestWorkspaceFolder()).uri.fsPath
         })
 
-        describe('should fetch 3 chunks and each chunk should contains 10 lines', function () {
+        describe('should fetch 3 chunks and each chunk should contains 50 lines', function () {
             async function assertCorrectCodeChunk() {
-                await toTextEditor(sampleFileOf60Lines, 'CrossFile.java', tempFolder, { preview: false })
+                await toTextEditor(aStringWithLineCount(200), 'CrossFile.java', tempFolder, { preview: false })
                 const myCurrentEditor = await toTextEditor('', 'TargetFile.java', tempFolder, {
                     preview: false,
                 })
                 const actual = await crossFile.fetchSupplementalContextForSrc(myCurrentEditor, fakeCancellationToken)
                 assert.ok(actual)
                 assert.ok(actual.supplementalContextItems.length === 3)
 
-                assert.strictEqual(actual.supplementalContextItems[0].content.split('\n').length, 10)
-                assert.strictEqual(actual.supplementalContextItems[1].content.split('\n').length, 10)
-                assert.strictEqual(actual.supplementalContextItems[2].content.split('\n').length, 10)
+                assert.strictEqual(actual.supplementalContextItems[0].content.split('\n').length, 50)
+                assert.strictEqual(actual.supplementalContextItems[1].content.split('\n').length, 50)
+                assert.strictEqual(actual.supplementalContextItems[2].content.split('\n').length, 50)
             }
 
             it('control group', async function () {
@@ -265,73 +265,59 @@ describe('crossFileContextUtil', function () {
             assert.strictEqual(chunks[1].content, 'line_6\nline_7')
         })
 
-        it('codewhisperer crossfile config should use 10 lines', async function () {
+        it('codewhisperer crossfile config should use 50 lines', async function () {
             const filePath = path.join(tempFolder, 'file.txt')
-            await toFile(sampleFileOf60Lines, filePath)
+            await toFile(aStringWithLineCount(210), filePath)
 
             const chunks = await crossFile.splitFileToChunks(filePath, crossFileContextConfig.numberOfLinesEachChunk)
-            assert.strictEqual(chunks.length, 6)
+
+            // (210 / 50) + 1
+            assert.strictEqual(chunks.length, 5)
+            // line0 -> line49
+            assert.strictEqual(chunks[0].content, aStringWithLineCount(50, 0))
+            // line50 -> line99
+            assert.strictEqual(chunks[1].content, aStringWithLineCount(50, 50))
+            // line100 -> line149
+            assert.strictEqual(chunks[2].content, aStringWithLineCount(50, 100))
+            // line150 -> line199
+            assert.strictEqual(chunks[3].content, aStringWithLineCount(50, 150))
+            // line 200 -> line209
+            assert.strictEqual(chunks[4].content, aStringWithLineCount(10, 200))
+        })
+
+        it('link', async function () {
+            const filePath = path.join(tempFolder, 'file.txt')
+            await toFile(aStringWithLineCount(210), filePath)
+
+            const chunks = await crossFile.splitFileToChunks(filePath, crossFileContextConfig.numberOfLinesEachChunk)
+            const linkedChunks = await crossFile.linkChunks(chunks)
+
+            // 210 / 50 + 2
+            assert.strictEqual(linkedChunks.length, 6)
+
+            // 0th
+            assert.strictEqual(linkedChunks[0].content, aStringWithLineCount(3, 0))
+            assert.strictEqual(linkedChunks[0].nextContent, aStringWithLineCount(50, 0))
+
+            // 1st
+            assert.strictEqual(linkedChunks[1].content, aStringWithLineCount(50, 0))
+            assert.strictEqual(linkedChunks[1].nextContent, aStringWithLineCount(50, 50))
+
+            // 2nd
+            assert.strictEqual(linkedChunks[2].content, aStringWithLineCount(50, 50))
+            assert.strictEqual(linkedChunks[2].nextContent, aStringWithLineCount(50, 100))
+
+            // 3rd
+            assert.strictEqual(linkedChunks[3].content, aStringWithLineCount(50, 100))
+            assert.strictEqual(linkedChunks[3].nextContent, aStringWithLineCount(50, 150))
+
+            // 4th
+            assert.strictEqual(linkedChunks[4].content, aStringWithLineCount(50, 150))
+            assert.strictEqual(linkedChunks[4].nextContent, aStringWithLineCount(10, 200))
+
+            // 5th
+            assert.strictEqual(linkedChunks[5].content, aStringWithLineCount(10, 200))
+            assert.strictEqual(linkedChunks[5].nextContent, aStringWithLineCount(10, 200))
         })
     })
 })
-
-const sampleFileOf60Lines = `import java.util.List;
-// we need this comment on purpose because chunk will be trimed right, adding this to avoid trimRight and make assertion easier
-/**
- * 
- * 
- * 
- * 
- * 
- **/
-class Main {
-    public static void main(String[] args) {
-        Calculator calculator = new Calculator();
-        calculator.add(1, 2);
-        calculator.subtract(1, 2);
-        calculator.multiply(1, 2);
-        calculator.divide(1, 2);
-        calculator.remainder(1, 2);
-    }
-}
-//
-class Calculator {
-    public Calculator() {
-        System.out.println("constructor");
-    }
-//
-    public add(int num1, int num2) {
-        System.out.println("add");
-        return num1 + num2;
-    }
-//
-    public subtract(int num1, int num2) {
-        System.out.println("subtract");
-        return num1 - num2;
-    }
-//
-    public multiply(int num1, int num2) {
-        System.out.println("multiply");
-        return num1 * num2;    
-    }
-//
-    public divide(int num1, int num2) {
-        System.out.println("divide");
-        return num1 / num2;
-    }
-//
-    public remainder(int num1, int num2) {
-        System.out.println("remainder");
-        return num1 % num2;
-    }
-//
-    public power(int num1, int num2) {
-        System.out.println("power");
-        return (int) Math.pow(num1, num2);
-    }
-//
-    public squareRoot(int num1) {
-        System.out.println("squareRoot");
-        return (int) Math.sqrt(num1);
-    }
-}`

packages/core/src/codewhisperer/util/supplementalContext/crossFileContextUtil.ts

@@ -171,7 +171,7 @@ function shouldFetchCrossFileContext(
  * when a given chunk context passes the match in BM25.
  * Special handling is needed for last(its next points to its own) and first chunk
  */
-function linkChunks(chunks: Chunk[]) {
+export function linkChunks(chunks: Chunk[]) {
     const updatedChunks: Chunk[] = []
 
     // This additional chunk is needed to create a next pointer to chunk 0.

packages/core/src/test/codewhisperer/testUtil.ts

@@ -207,3 +207,12 @@ export function createMockDirentFile(fileName: string): fs.Dirent {
     dirent.name = fileName
     return dirent
 }
+
+export function aStringWithLineCount(lineCount: number, start: number = 0): string {
+    let s = ''
+    for (let i = start; i < start + lineCount; i++) {
+        s += `line${i}\n`
+    }
+
+    return s.trimEnd()
+}