Codelens for benchmarks #1522

src/goRunTestCodelens.ts

@@ -14,6 +14,7 @@ import { getCurrentGoPath } from './util';
 import { GoBaseCodeLensProvider } from './goBaseCodelens';
 
 export class GoRunTestCodeLensProvider extends GoBaseCodeLensProvider {
+	private readonly benchmarkRegex = /^Benchmark.+/;
 	private readonly debugConfig: any = {
 		'name': 'Launch',
 		'type': 'go',
@@ -53,21 +54,35 @@ export class GoRunTestCodeLensProvider extends GoBaseCodeLensProvider {
 	private getCodeLensForPackage(document: TextDocument, token: CancellationToken): Thenable<CodeLens[]> {
 		let documentSymbolProvider = new GoDocumentSymbolProvider();
 		return documentSymbolProvider.provideDocumentSymbols(document, token)
-			.then(symbols => symbols.find(sym => sym.kind === vscode.SymbolKind.Package && !!sym.name))
-			.then(pkg => {
-				if (pkg) {
-					const range = pkg.location.range;
-					return [
+			.then(symbols => {
+				const pkg = symbols.find(sym => sym.kind === vscode.SymbolKind.Package && !!sym.name);
+				if (!pkg) {
+					return;
+				}
+				const range = pkg.location.range;
+				const packageCodeLens = [
+					new CodeLens(range, {
+						title: 'run package tests',
+						command: 'go.test.package'
+					}),
+					new CodeLens(range, {
+						title: 'run file tests',
+						command: 'go.test.file'
+					})
+				];
+				if (symbols.some(sym => sym.kind === vscode.SymbolKind.Function && this.benchmarkRegex.test(sym.name))) {
+					packageCodeLens.push(
 						new CodeLens(range, {
-							title: 'run package tests',
-							command: 'go.test.package'
+							title: 'run package benchmarks',
+							command: 'go.benchmark.package'
 						}),
 						new CodeLens(range, {
-							title: 'run file tests',
-							command: 'go.test.file'
+							title: 'run file benchmarks',
+							command: 'go.benchmark.file'
 						})
-					];
+					);
 				}
+				return packageCodeLens;
 			});
 	}
 

src/testUtils.ts

@@ -25,7 +25,7 @@ const runningTestProcesses: cp.ChildProcess[] = [];
 
 const testFuncRegex = /^Test.+|Example.+/;
 const testMethodRegex = /^\(([^)]+)\)\.(Test.*)$/;
-const benchmarkRegex = /^Benchmark$/;
+const benchmarkRegex = /^Benchmark.+/;
 
 /**
  * Input to goTest.