changed command ID; implemented menu and config.

implemented menu context option to create/open the test;
config to define where to open the file

Author: wesleyegberto

package.json

@@ -1,9 +1,13 @@
 {
 	"name": "vscode-java-tests",
 	"publisher": "wesleyegberto",
-	"displayName": "vscode-java-tests",
+	"displayName": "VSCode Java Tests",
 	"description": "Extension to help write Java tests",
 	"version": "0.0.2",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/wesleyegberto/vscode-java-tests.git"
+  },
 	"engines": {
 		"vscode": "^1.47.0"
 	},
@@ -12,16 +16,37 @@
 		"Snippets"
 	],
 	"activationEvents": [
-		"onCommand:vscode-java-tests.createTestClass"
+		"onLanguage:java",
+		"onCommand:java.tests.createTestClass"
 	],
 	"main": "./out/extension.js",
 	"contributes": {
 		"commands": [
 			{
-				"command": "vscode-java-tests.createTestClass",
-				"title": "Java Tests: Create Test Class"
+				"command": "java.tests.createTestClass",
+				"title": "Java Tests: Create/Open Test Class"
 			}
-		]
+		],
+		"menus": {
+			"explorer/context": [
+				{
+					"when": "resourceLangId == java",
+					"command": "java.tests.createTestClass",
+					"group": "vscodeJavaTests@1"
+				}
+			]
+		},
+		"configuration": {
+			"title": "Java Tests",
+			"properties": {
+				"javaTests.file.openLocation": {
+					"type": "string",
+					"default": "beside",
+					"enum": ["beside", "currentGroup"],
+					"description": "Controls where test class should be opened."
+				}
+			}
+		}
 	},
 	"scripts": {
 		"vscode:prepublish": "npm run compile",

src/extension.ts

@@ -4,52 +4,63 @@ import { posix } from 'path';
 import { generateTestClassFileContent } from './file-content-generator';
 
 export function activate(context: vscode.ExtensionContext) {
-	let disposable = vscode.commands.registerCommand('vscode-java-tests.createTestClass', createTestClass);
+  console.debug('Java Tests - Extension loaded');
 
-	context.subscriptions.push(disposable);
+  context.subscriptions.push(
+    vscode.commands.registerCommand('java.tests.createTestClass', createTestClass)
+  );
 }
 
 export function deactivate() {}
 
-async function createTestClass() {
-	const activeEditor = vscode.window.activeTextEditor;
+async function createTestClass(args: any) {
+  let javaFileUri: vscode.Uri;
 
-	if (!activeEditor) {
-		vscode.window.showWarningMessage("Please, open a Java file.");
-		return;
-	}
+  if (args && args.scheme === 'file' && args.path) {
+    javaFileUri = args as vscode.Uri;
 
-	if (!activeEditor.document.fileName.endsWith('.java')) {
-		vscode.window.showWarningMessage('Only allowed to create test from a Java file');
-		return;
-	}
+  } else {
+    const activeEditor = vscode.window.activeTextEditor;
 
-	const javaFileUri = activeEditor.document.uri;
-	const javaClassName = posix.basename(javaFileUri.path, '.java');
+    if (!activeEditor) {
+      vscode.window.showWarningMessage('Please, open a Java file.');
+      return;
+    }
 
-	const testClassName = `${javaClassName}Test`;
-	const testFileUri = getTestFileUri(javaFileUri, testClassName);
+    javaFileUri = activeEditor.document.uri;
+  }
 
-	try {
-		await vscode.workspace.fs.stat(testFileUri);
-		showTestFile(testFileUri);
+  if (!javaFileUri || !javaFileUri.path.endsWith('.java')) {
+    vscode.window.showWarningMessage('Only allowed to create test from a Java file');
+    return;
+  }
 
-	} catch {
-		const fileContent = generateTestClassFileContent(javaFileUri, javaClassName, testFileUri, testClassName);
-		await vscode.workspace.fs.writeFile(testFileUri, fileContent);
+  const javaClassName = posix.basename(javaFileUri.path, '.java');
 
-		showTestFile(testFileUri);
-	}
+  const testClassName = `${javaClassName}Test`;
+  const testFileUri = getTestFileUri(javaFileUri, testClassName);
+
+  try {
+    await vscode.workspace.fs.stat(testFileUri);
+    showTestFile(testFileUri);
+  } catch {
+    const fileContent = generateTestClassFileContent(javaFileUri, javaClassName, testFileUri, testClassName);
+    await vscode.workspace.fs.writeFile(testFileUri, fileContent);
+
+    showTestFile(testFileUri);
+  }
 }
 
 function showTestFile(testFileUri: vscode.Uri) {
-  // TODO: extract configuration to define where to open it: beside (new editor group) or current editor group
-  vscode.window.showTextDocument(testFileUri, { viewColumn: vscode.ViewColumn.Beside });
+  const config = vscode.workspace.getConfiguration('javaTests');
+  const configOpenLocationValue = config.get('file.openLocation', 'beside');
+  vscode.window.showTextDocument(testFileUri, {
+    viewColumn: configOpenLocationValue === 'beside' ? vscode.ViewColumn.Beside : vscode.ViewColumn.Active
+  });
 }
 
 function getTestFileUri(javaFileUri: vscode.Uri, testClassName: string) {
-	const testPath = javaFileUri.path.replace('/src/main/java', '/src/test/java');
-	const testFilePath = posix.join(testPath, '..', `${testClassName}.java`);
-	return javaFileUri.with({ path: testFilePath });
+  const testPath = javaFileUri.path.replace('/src/main/java', '/src/test/java');
+  const testFilePath = posix.join(testPath, '..', `${testClassName}.java`);
+  return javaFileUri.with({ path: testFilePath });
 }
-

src/file-content-generator.ts

@@ -41,25 +41,25 @@ public class ${testClassName} {
 function generateTestClassPackageDeclaration(testFileUri: vscode.Uri, testClassName: string) {
   const packageName = extractPackageName(testFileUri, testClassName, true);
   if (!packageName.length) {
-	  return '';
+    return '';
   }
   return `package ${packageName};`;
 }
 
 function generateTargetTestClassPackageImport(javaFileUri: vscode.Uri, javaClassName: string) {
   const packageName = extractPackageName(javaFileUri, javaClassName, false);
   if (!packageName.length) {
-	  return '';
+    return '';
   }
   return `import ${packageName}.${javaClassName};`;
 }
 
 function extractPackageName(fileUri: vscode.Uri, fileClassName: string, isTest: boolean): string {
-	const pathPrefix = isTest ? '/src/test/java' : '/src/main/java';
-	const startIndex = fileUri.fsPath.indexOf(pathPrefix) + 15; // '/src/test/java/'.length
-	const endIndex = fileUri.fsPath.indexOf(fileClassName) - 1;
-	if (startIndex >= endIndex) {
-		return '';
-	}
-	return fileUri.fsPath.substring(startIndex, endIndex).replace(/\//g, '.');
+  const pathPrefix = isTest ? '/src/test/java' : '/src/main/java';
+  const startIndex = fileUri.fsPath.indexOf(pathPrefix) + 15; // '/src/test/java/'.length
+  const endIndex = fileUri.fsPath.indexOf(fileClassName) - 1;
+  if (startIndex >= endIndex) {
+    return '';
+  }
+  return fileUri.fsPath.substring(startIndex, endIndex).replace(/\//g, '.');
 }