diff --git a/src/goGenerateTests.ts b/src/goGenerateTests.ts index 7a6325414d..2e4d31d9ea 100644 --- a/src/goGenerateTests.ts +++ b/src/goGenerateTests.ts @@ -227,5 +227,5 @@ function generateTests(conf: Config, goConfig: vscode.WorkspaceConfiguration): P async function getFunctions(doc: vscode.TextDocument): Promise { const documentSymbolProvider = new GoDocumentSymbolProvider(); const symbols = await documentSymbolProvider.provideDocumentSymbols(doc, null); - return symbols[0].children.filter((sym) => sym.kind === vscode.SymbolKind.Function); + return symbols[0].children.filter((sym) => [vscode.SymbolKind.Function, vscode.SymbolKind.Method].includes(sym.kind)); }