commit 41591e39eb1f6e935162380ddb7a663ccad7f544
author marcel-basel <59963477+marcel-basel@users.noreply.github.com> Sun Jan 26 08:12:41 2020 +0100
committer Ramya Rao <ramyar@microsoft.com> Sat Jan 25 23:12:41 2020 -0800
tree ede1949e6ce805ebcf56ed3ac0a077d1de0be588
parent 97a7fc63ebbdfb1a22a34b4c7e7c56cb28f285b6

Considering receiverType on function test creation issue-2282 (#2987)

src/goGenerateTests.ts

diff --git a/src/goGenerateTests.ts b/src/goGenerateTests.ts
index 7e8df75..1fbd717 100644
--- a/src/goGenerateTests.ts
+++ b/src/goGenerateTests.ts
@@ -98,9 +98,13 @@
 		return Promise.resolve(false);
 	}
 	let funcName = currentFunction.name;
-	if (funcName.includes('.')) {
-		funcName = funcName.split('.')[1];
+	const funcNameParts = funcName.match(/^\(\*?(.*)\)\.(.*)$/);
+	if (funcNameParts != null && funcNameParts.length === 3) {  // receiver type specified
+		const rType = funcNameParts[1].replace(/^\w/, (c) => c.toUpperCase());
+		const fName = funcNameParts[2].replace(/^\w/, (c) => c.toUpperCase());
+		funcName = rType + fName;
 	}
+
 	return generateTests({ dir: editor.document.uri.fsPath, func: funcName }, getGoConfig(editor.document.uri));
 }