Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add: support of js function finder #66

Merged
merged 2 commits into from
Oct 7, 2024
Merged

add: support of js function finder #66

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
36d8644
add: support of js function finder
shivamsouravjha Sep 27, 2024
57a5f66
fix:adding auto-detect test location for python
Aditya-eddy Oct 3, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
141 changes: 66 additions & 75 deletions src/Utg.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,31 +4,16 @@ import * as path from 'path';
import axios, { AxiosResponse } from 'axios';


async function Utg(context: vscode.ExtensionContext , additional_prompts?:string) {
async function Utg(context: vscode.ExtensionContext , additional_prompts?:string,testFilesPath?: vscode.Uri[] | undefined) {

try {
return new Promise<void>(async (resolve, reject) => {
try {

const token = await context.globalState.get<'string'>('JwtToken');
// console.log("token in the utg" , token);
let apiResponse:string = '';
// vscode.window.showInformationMessage('Attempting to trigger API request...');
// if(token){
// apiResponse = (await makeApiRequest(token)) || 'no response'; // Fallback to empty string if null
// if (apiResponse) {
// vscode.window.showInformationMessage(`Received API Response: ${apiResponse}`);

// context.globalState.update('apiResponse', apiResponse);
// }
// }else{
// console.log("token no defined");
// }
// Create a terminal named "Keploy Terminal"

const token = await context.globalState.get<'string'>('JwtToken');
let apiResponse: string = '';
const terminal = vscode.window.createTerminal("Keploy Terminal");
terminal.show();

const editor = vscode.window.activeTextEditor;
let currentFilePath = "";
if (editor) {
Expand All @@ -39,112 +24,117 @@ async function Utg(context: vscode.ExtensionContext , additional_prompts?:string
vscode.window.showInformationMessage('No file is currently opened.');
return;
}

const scriptPath = path.join(context.extensionPath, 'scripts', 'utg.sh');

const sourceFilePath = currentFilePath;
// ensureTestFileExists(sourceFilePath);


if (!vscode.workspace.workspaceFolders) {
vscode.window.showErrorMessage('No workspace is opened.');
return;
}

const rootDir = vscode.workspace.workspaceFolders[0].uri.fsPath;


const extension = path.extname(sourceFilePath);
let testFilePath: string;
let testFilePaths: string[] = [];
let command: string;
let coverageReportPath: string;
let testFileContent:string;
let testFileContent: string;

if (extension === '.js' || extension === '.ts') {
testFilePath = path.join(path.join(rootDir, 'test'), path.basename(sourceFilePath).replace(extension, `.test${extension}`));
if (!fs.existsSync(testFilePath)) {
vscode.window.showInformationMessage("Test doesn't exist", testFilePath);
fs.writeFileSync(testFilePath, `// Test file for ${testFilePath}`);
if (testFilesPath && testFilesPath.length > 0) {
// Use only the first path from testFilesPath
testFilePaths = [testFilesPath[0].fsPath];
} else {
const defaultTestFilePath = path.join(rootDir, 'test', path.basename(sourceFilePath).replace(extension, `.test${extension}`));
testFilePaths.push(defaultTestFilePath);
if (!fs.existsSync(defaultTestFilePath)) {
vscode.window.showInformationMessage("Test doesn't exist", defaultTestFilePath);
fs.writeFileSync(defaultTestFilePath, `// Test file for ${defaultTestFilePath}`);
}
}
command = `npm test -- --coverage --coverageReporters=text --coverageReporters=cobertura --coverageDirectory=./coverage`;
coverageReportPath = "./coverage/cobertura-coverage.xml";

} else if (extension === '.py') {
const testDir = path.join(rootDir,'test');
testFilePath = path.join(testDir,'test_'+ path.basename(sourceFilePath));
if (!fs.existsSync(testFilePath)) {
vscode.window.showInformationMessage("Test doesn't exist", testFilePath);
fs.writeFileSync(testFilePath, `// Test file for ${testFilePath}`);
if (testFilesPath && testFilesPath.length > 0) {
// Use only the first path from testFilesPath
testFilePaths = [testFilesPath[0].fsPath];
} else {
const testDir = path.join(rootDir, 'test');
const defaultTestFilePath = path.join(testDir, 'test_' + path.basename(sourceFilePath));
testFilePaths.push(defaultTestFilePath);
if (!fs.existsSync(defaultTestFilePath)) {
vscode.window.showInformationMessage("Test doesn't exist", defaultTestFilePath);
fs.writeFileSync(defaultTestFilePath, `# Test file for ${defaultTestFilePath}`);
}
}
console.log(sourceFilePath , testFilePath , "python wala");
command = `pytest --cov=${path.basename(sourceFilePath,'.py')} --cov-report=xml:coverage.xml ${testFilePath}`;
command = `pytest --cov=${path.basename(sourceFilePath, '.py')} --cov-report=xml:coverage.xml ${testFilePaths[0]}`;
coverageReportPath = "./coverage.xml";

}else if (extension === '.java') {

} else if (extension === '.java') {
// Proceed as before for Java
const testDir = path.join(rootDir, 'src', 'test', 'java');
const testFileName = path.basename(sourceFilePath).replace('.java', 'Test.java');
testFilePath = path.join(testDir, testFileName);

const testFilePath = path.join(testDir, testFileName);
if (!fs.existsSync(testFilePath)) {
vscode.window.showInformationMessage("Test doesn't exist", testFilePath);
fs.writeFileSync(testFilePath, `// Test file for ${testFilePath}`);
}
testFilePaths.push(testFilePath);
command = `mvn clean test jacoco:report`;
coverageReportPath = "./target/site/jacoco/jacoco.xml";

} else if (extension === '.go') {
testFilePath = path.join(rootDir, path.basename(sourceFilePath).replace('.go', '_test.go'));
console.log(testFilePath , "in the go block");
if (!fs.existsSync(testFilePath)) {
vscode.window.showInformationMessage("Test doesn't exist", testFilePath);
const uniqueFuncName = path.basename(sourceFilePath).replace('.go', 'Test');
// Proceed as before for Go
const defaultTestFilePath = path.join(rootDir, path.basename(sourceFilePath).replace('.go', '_test.go'));
testFilePaths.push(defaultTestFilePath);
if (!fs.existsSync(defaultTestFilePath)) {
vscode.window.showInformationMessage("Test doesn't exist", defaultTestFilePath);
testFileContent = `package main\n\nimport "testing"`;
fs.writeFileSync(testFilePath, testFileContent); }
fs.writeFileSync(defaultTestFilePath, testFileContent);
}
command = `go test -v ./... -coverprofile=coverage.out && gocov convert coverage.out | gocov-xml > coverage.xml`;
coverageReportPath = "./coverage.xml";
}
else {

} else {
vscode.window.showErrorMessage(`Unsupported file type: ${extension}`);
return;
}

if(!additional_prompts){
additional_prompts = "";
}


terminal.sendText(`sh "${scriptPath}" "${sourceFilePath}" "${testFilePath}" "${coverageReportPath}" "${command}" "${additional_prompts}";`);

const delay = (ms: number) => new Promise(resolve => setTimeout(resolve, ms));

// Add a 5-second delay before calling the API
await delay(5000);


try {
if(token){
// console.log("token inside the try block in utg.ts" , token);

// Adjust the terminal command to include the test file path
terminal.sendText(`sh "${scriptPath}" "${sourceFilePath}" "${testFilePaths[0]}" "${coverageReportPath}" "${command}" "${additional_prompts}";`);

const delay = (ms: number) => new Promise(resolve => setTimeout(resolve, ms));

// Add a 5-second delay before calling the API
await delay(5000);

try {
if (token) {
apiResponse = await makeApiRequest(token) || 'no response';
const response = JSON.parse(apiResponse);
await context.globalState.update('apiResponse', apiResponse);
if(response.usedCall === response.totalCall ){
await context.globalState.update('SubscriptionEnded' , true);
if (response.usedCall === response.totalCall) {
await context.globalState.update('SubscriptionEnded', true);
}
}else{
} else {
console.log("token not found");
}
} catch (apiError) {
vscode.window.showErrorMessage('Error during API request: ' + apiError);
}

const disposable = vscode.window.onDidCloseTerminal(eventTerminal => {
if (eventTerminal === terminal) {
disposable.dispose();
resolve();
}



});
} catch (error) {
} catch (error) {
console.log(error);
vscode.window.showErrorMessage('Error occurred Keploy utg: ' + error);
reject(error);
Expand All @@ -157,6 +147,7 @@ async function Utg(context: vscode.ExtensionContext , additional_prompts?:string
}
}


// Separate function for making the API request using axios
export async function makeApiRequest(token:string): Promise<string | null> {
const url = 'https://api.keploy.io/ai/call/count ';
Expand Down
Loading
Loading