Fix testify not being detected anymore (#1969)

microsoft · Oct 7, 2018 · d8cc00f · d8cc00f
1 parent f8d9b57
commit d8cc00f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/testUtils.ts b/src/testUtils.ts
@@ -24,7 +24,7 @@ statusBarItem.text = '$(x) Cancel Running Tests';
  */
 const runningTestProcesses: cp.ChildProcess[] = [];
 
-const testFuncRegex = /^Test.+|Example.+/;
+const testFuncRegex = /^Test.+|^Example.+/;
 const testMethodRegex = /^\(([^)]+)\)\.(Test.*)$/;
 const benchmarkRegex = /^Benchmark.+/;
 
@@ -108,7 +108,7 @@ export function getTestFunctions(doc: vscode.TextDocument, token: vscode.Cancell
 	return documentSymbolProvider
 		.provideDocumentSymbols(doc, token)
 		.then(symbols => {
-			const testify = symbols.some(sym => sym.kind === vscode.SymbolKind.Module && sym.name === 'github.com/stretchr/testify/suite');
+			const testify = symbols.some(sym => sym.kind === vscode.SymbolKind.Namespace && sym.name === '"github.com/stretchr/testify/suite"');
 			return symbols.filter(sym =>
 				sym.kind === vscode.SymbolKind.Function
 				&& (testFuncRegex.test(sym.name) || (testify && testMethodRegex.test(sym.name)))