feat(go): add method context builder test

Add a new test case for the GoMethodContextBuilder class. The test verifies that the method signature is correctly parsed and the context is built with the expected values.

Author: phodal

goland/src/main/kotlin/cc/unitmesh/go/context/GoMethodContextBuilder.kt

@@ -2,7 +2,9 @@ package cc.unitmesh.go.context
 
 import cc.unitmesh.devti.context.MethodContext
 import cc.unitmesh.devti.context.builder.MethodContextBuilder
+import com.goide.psi.GoFunctionDeclaration
 import com.goide.psi.GoFunctionOrMethodDeclaration
+import com.goide.psi.GoMethodDeclaration
 import com.intellij.psi.PsiElement
 
 class GoMethodContextBuilder : MethodContextBuilder {
@@ -16,7 +18,21 @@ class GoMethodContextBuilder : MethodContextBuilder {
             return null
         }
 
-        val functionSignature = psiElement.signature?.text
+
+        var funcName = ""
+        val functionSignature: String = when (psiElement) {
+            is GoMethodDeclaration -> {
+                funcName = psiElement.receiver?.text ?: ""
+                psiElement.signature?.text ?: ""
+            }
+
+            is GoFunctionDeclaration -> {
+                funcName = psiElement.name ?: ""
+                psiElement.signature?.text ?: ""
+            }
+
+            else -> ""
+        }
         val returnType = psiElement.signature?.resultType?.text
         val languages = psiElement.language.displayName
         val returnTypeText = returnType
@@ -25,7 +41,7 @@ class GoMethodContextBuilder : MethodContextBuilder {
         }.orEmpty()
 
         return MethodContext(
-            psiElement, psiElement.text, psiElement.name!!, functionSignature, null, languages,
+            psiElement, psiElement.text, funcName, functionSignature, null, languages,
             returnTypeText, parameterNames, includeClassContext, emptyList()
         )
 

goland/src/main/kotlin/cc/unitmesh/go/context/GoVariableContextBuilder.kt

@@ -16,7 +16,6 @@ class GoVariableContextBuilder : VariableContextBuilder {
             return null
         }
 
-
         val name = psiElement.name
 
         return VariableContext(

goland/src/test/kotlin/cc/unitmesh/go/context/GoMethodContextBuilderTest.kt

@@ -0,0 +1,33 @@
+package cc.unitmesh.go.context;
+
+import com.goide.psi.GoFunctionOrMethodDeclaration
+import com.goide.psi.GoTypeDeclaration
+import com.intellij.psi.PsiElement
+import com.intellij.psi.util.PsiTreeUtil
+import com.intellij.testFramework.fixtures.BasePlatformTestCase
+import org.junit.Test
+
+class GoMethodContextBuilderTest : BasePlatformTestCase() {
+
+    fun testShouldHandleMethodSignature() {
+        // given
+        val file = myFixture.configureByText(
+            "test.go", """
+            package main
+
+            func f3(float64, float64, float64) {}
+            """.trimIndent()
+        )
+
+        val struct = PsiTreeUtil.getChildrenOfTypeAsList(file, GoFunctionOrMethodDeclaration::class.java).first()
+        val context = GoMethodContextBuilder().getMethodContext(struct, false, false)
+
+        assertEquals(context?.name, "f3")
+        assertEquals(context?.format(), """
+            path: /src/test.go
+            language: Go
+            fun name: f3
+            fun signature: (float64, float64, float64)
+            """.trimIndent())
+    }
+}