Mark test methods as testify suite methods only if testify framework …

…is used #1911
microsoft · Sep 10, 2018 · a11c5aa · a11c5aa
1 parent 3477829
commit a11c5aa
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/testUtils.ts b/src/testUtils.ts
@@ -96,12 +96,13 @@ export function getTestFunctions(doc: vscode.TextDocument, token: vscode.Cancell
 	let documentSymbolProvider = new GoDocumentSymbolProvider();
 	return documentSymbolProvider
 		.provideDocumentSymbols(doc, token)
-		.then(symbols =>
-			symbols.filter(sym =>
+		.then(symbols => {
+			const testify = symbols.some(sym => sym.kind === vscode.SymbolKind.Module && sym.name === "github.com/stretchr/testify/suite");
+			return symbols.filter(sym =>
 				sym.kind === vscode.SymbolKind.Function
-				&& (sym.name.startsWith('Test') || sym.name.startsWith('Example') || testSuiteMethodRegex.test(sym.name))
+				&& (sym.name.startsWith('Test') || sym.name.startsWith('Example') || (testify && testSuiteMethodRegex.test(sym.name)))
 			)
-		);
+		});
 }
 
 /**