fix:java sample code

src/Utg.ts

@@ -145,16 +145,20 @@
 
             // 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}`);

src/extension.ts

@@ -122,6 +122,7 @@
         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 @@
                 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 @@
         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[] = [];
 
@@ -704,7 +713,7 @@
         ExtentionName = FileExtentionName;
         FunctionFilePath = filePath;
         vscode.commands.executeCommand('workbench.view.extension.Keploy-Sidebar');
         sidebarProvider.postMessage("KeployChatBot")
     });
 
     context.subscriptions.push(showSidebarDisposable);
@@ -712,7 +721,7 @@
     let runAdditionalPrompts = vscode.commands.registerCommand('keploy.runAdditionalPrompts', async (additionalPrompts: string) => {
         console.log("value inside the function: ", functionName, ExtentionName, additionalPrompts);
         await vscode.commands.executeCommand('keploy.utg', FunctionFilePath, functionName, ExtentionName, additionalPrompts);
     })
 
     context.subscriptions.push(runAdditionalPrompts);