fix:java sample code (#77)

* fix:java sample code Signed-off-by: Aditya-eddy <[email protected]> * chore: add size of parser Signed-off-by: shivamsouravjha <[email protected]> --------- Signed-off-by: Aditya-eddy <[email protected]> Signed-off-by: shivamsouravjha <[email protected]> Co-authored-by: shivamsouravjha <[email protected]>
keploy · Oct 15, 2024 · aae9c1a · aae9c1a
1 parent 52f6637
commit aae9c1a
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 6 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/src/Utg.ts b/src/Utg.ts
@@ -145,16 +145,20 @@ async function Utg(context: vscode.ExtensionContext , additional_prompts?:string
 
             // Define the test file name by appending 'Test' to the original class name
             const originalClassName = path.basename(sourceFilePath, '.java');
-            const testFileName = `${originalClassName}Test.java`;
+            const testFileName = `${originalClassName}Tests.java`;
             const testFilePath = path.join(fullTestDir, testFileName);
+            const JavaClassName = `${originalClassName}Tests`
 
             testFilePaths.push(testFilePath);
 
             if (!fs.existsSync(testFilePath)) {
                 vscode.window.showInformationMessage("Test doesn't exist", testFilePath);
 
                 // **Create Test File Content with Proper Package and JUnit Imports**
-                const testFileContent = `package ${packageName};`;
+                const testFileContent = `package ${packageName};\n\n
+                    class ${JavaClassName} {\n
+                }\n`;
+
                 fs.writeFileSync(testFilePath, testFileContent);
                 vscode.window.showInformationMessage(`Created test file: ${testFilePath}`);
                 console.log(`🐰 Created test file with package name: ${testFilePath}`);

diff --git a/src/extension.ts b/src/extension.ts
@@ -122,6 +122,7 @@ class KeployCodeLensProvider implements vscode.CodeLensProvider {
         if (
             fileName.endsWith('.test.js') ||
             fileName.endsWith('.test.ts') ||
+            fileName.endsWith('Tests.java') ||  // Check for Java test file ending
             fileName.endsWith('Test.java') ||  // Check for Java test file ending
             fileName.includes('/Test') ||      // Check for Java test file prefix in the path
             fileName.includes('/test/') ||    // Skip files in a "tests" directory
@@ -296,7 +297,11 @@ async function findTestCasesForFunction(functionName: string, fileExtension: str
                 continue;
             }
 
-            const tree = parser.parse(text);
+            const options: TreeSitter.Options = {
+                bufferSize: 1024 * 1024,
+            };
+
+            const tree = parser.parse(text, undefined, options);  // Parse the document text
             const cursor = tree.walk();
             let found = false;
 
@@ -414,8 +419,12 @@ async function getAllFunctionsInFile(
         console.log('🐰 Unsupported file type:', filePath);
         throw new Error("Unsupported file type");
     }
+    const options: TreeSitter.Options = {
+        bufferSize: 1024 * 1024,
+    };
+
+    const tree = parser.parse(text, undefined, options);  // Parse the document text
 
-    const tree = parser.parse(text);
     const cursor = tree.walk();
     const functionNames: string[] = [];