fix(goTest): handler for debugging suite tests

Author: Alysson Ribeiro

extension/src/goTest.ts

@@ -44,7 +44,7 @@ async function _testAtCursor(
 	goCtx: GoExtensionContext,
 	goConfig: vscode.WorkspaceConfiguration,
 	cmd: TestAtCursorCmd,
-	args: any
+	args?: SubTestAtCursorArgs
 ) {
 	const editor = vscode.window.activeTextEditor;
 	if (!editor) {
@@ -72,7 +72,7 @@ async function _testAtCursor(
 	await editor.document.save();
 
 	if (cmd === 'debug') {
-		return debugTestAtCursor(editor, testFunctionName, testFunctions, suiteToTest, goConfig);
+		return debugTestAtCursor(editor, testFunctionName, testFunctions, suiteToTest, goConfig, undefined, args);
 	} else if (cmd === 'benchmark' || cmd === 'test') {
 		return runTestAtCursor(editor, testFunctionName, testFunctions, suiteToTest, goConfig, cmd, args);
 	} else {
@@ -165,7 +165,7 @@ async function _subTestAtCursor(
 	const escapedName = escapeSubTestName(testFunctionName, subTestName);
 
 	if (cmd === 'debug') {
-		return debugTestAtCursor(editor, escapedName, testFunctions, suiteToTest, goConfig);
+		return debugTestAtCursor(editor, escapedName, testFunctions, suiteToTest, goConfig, undefined, args);
 	} else if (cmd === 'test') {
 		return runTestAtCursor(editor, escapedName, testFunctions, suiteToTest, goConfig, cmd, args);
 	} else {
@@ -308,10 +308,11 @@ export async function debugTestAtCursor(
 	testFunctions: vscode.DocumentSymbol[],
 	suiteToFunc: SuiteToTestMap,
 	goConfig: vscode.WorkspaceConfiguration,
-	sessionID?: string
+	sessionID?: string,
+	testArgs?: { isTestSuite?: boolean },
 ) {
 	const doc = 'document' in editorOrDocument ? editorOrDocument.document : editorOrDocument;
-	const args = getTestFunctionDebugArgs(doc, testFunctionName, testFunctions, suiteToFunc);
+	const args = getTestFunctionDebugArgs(doc, testFunctionName, testFunctions, suiteToFunc, testArgs?.isTestSuite);
 	const tags = getTestTags(goConfig);
 	const buildFlags = tags ? ['-tags', tags] : [];
 	const flagsFromConfig = getTestFlags(goConfig);

extension/src/testUtils.ts

@@ -250,12 +250,14 @@ export function extractInstanceTestName(symbolName: string): string {
  * @param document  The document containing the tests
  * @param testFunctionName The test function to get the debug args
  * @param testFunctions The test functions found in the document
+ * @param isTestSuite Indicator if the test function is part of a test suite
  */
 export function getTestFunctionDebugArgs(
 	document: vscode.TextDocument,
 	testFunctionName: string,
 	testFunctions: vscode.DocumentSymbol[],
-	suiteToFunc: SuiteToTestMap
+	suiteToFunc: SuiteToTestMap,
+	isTestSuite ?: boolean,
 ): string[] {
 	if (benchmarkRegex.test(testFunctionName)) {
 		return ['-test.bench', '^' + testFunctionName + '$', '-test.run', 'a^'];
@@ -265,7 +267,7 @@ export function getTestFunctionDebugArgs(
 		const testFns = findAllTestSuiteRuns(document, testFunctions, suiteToFunc);
 		return ['-test.run', `^${testFns.map((t) => t.name).join('|')}$/^${instanceMethod}$`];
 	} else {
-		return ['-test.run', `^${testFunctionName}$`];
+		return ['-test.run', `${!!isTestSuite ? '/': ''}^${testFunctionName}$`];
 	}
 }
 /**