feat(goland): add GoWriteTestService for writing test cases

This commit adds the `GoWriteTestService` class to the `goland` module. This class is responsible for providing functionality to write test cases in Go. It includes methods for determining if the service is applicable, looking up relevant classes, and finding or creating test files. Additionally, it includes a helper method for getting the element for tests. This new service will enhance the testing capabilities of the `goland` module.

Author: phodal

goland/src/main/kotlin/cc/unitmesh/go/provider/testing/GoWriteTestService.kt

@@ -0,0 +1,49 @@
+package cc.unitmesh.go.provider.testing
+
+import cc.unitmesh.devti.context.ClassContext
+import cc.unitmesh.devti.provider.WriteTestService
+import cc.unitmesh.devti.provider.context.TestFileContext
+import com.goide.execution.testing.GoTestRunConfiguration
+import com.goide.psi.GoFile
+import com.goide.psi.GoFunctionOrMethodDeclaration
+import com.intellij.execution.configurations.RunProfile
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.roots.TestSourcesFilter
+import com.intellij.psi.PsiElement
+import com.intellij.psi.PsiFile
+import com.intellij.psi.util.PsiTreeUtil
+import com.intellij.testIntegration.TestFinderHelper
+import com.intellij.util.PlatformUtils
+
+class GoWriteTestService : WriteTestService() {
+    override fun isApplicable(element: PsiElement): Boolean = PlatformUtils.isGoIde()
+    override fun runConfigurationClass(project: Project): Class<out RunProfile> = GoTestRunConfiguration::class.java
+
+    override fun lookupRelevantClass(project: Project, element: PsiElement): List<ClassContext> {
+        TODO("Not yet implemented")
+    }
+
+    override fun findOrCreateTestFile(sourceFile: PsiFile, project: Project, element: PsiElement): TestFileContext? {
+        TODO("Not yet implemented")
+    }
+
+    fun getElementForTests(elementAtCaret: PsiElement): PsiElement? {
+        val parent = PsiTreeUtil.getParentOfType(elementAtCaret, GoFunctionOrMethodDeclaration::class.java, false)
+        if (parent == null) {
+            val goFile: GoFile = elementAtCaret as? GoFile ?: return null
+            return if (goFile.functions.isNotEmpty() || goFile.methods.isNotEmpty()) {
+                goFile
+            } else {
+                null
+            }
+        }
+
+        val virtualFile = elementAtCaret.containingFile?.virtualFile ?: return null
+
+        val project = elementAtCaret.project
+        if (TestSourcesFilter.isTestSources(virtualFile, project) || TestFinderHelper.isTest(parent)) return null
+
+        return parent
+    }
+
+}

goland/src/main/resources/cc.unitmesh.go.xml

@@ -12,6 +12,8 @@
         <methodContextBuilder language="go" implementationClass="cc.unitmesh.go.context.GoMethodContextBuilder"/>
         <variableContextBuilder language="go" implementationClass="cc.unitmesh.go.context.GoVariableContextBuilder"/>
 
+        <testContextProvider language="go" implementation="cc.unitmesh.go.provider.testing.GoWriteTestService"/>
+
         <chatContextProvider implementation="cc.unitmesh.go.provider.GoVersionChatContextProvider"/>
     </extensions>
 </idea-plugin>