fix(test): add check for now writing test service #78

Add WriteTestService to AutoTestThisIntention to check if language support is available for the current file. This ensures that the intention is only available when the necessary plugin is installed.

Author: phodal

src/main/kotlin/cc/unitmesh/devti/intentions/action/AutoTestThisIntention.kt

@@ -5,6 +5,7 @@ import cc.unitmesh.devti.intentions.action.base.AbstractChatIntention
 import com.intellij.temporary.getElementToAction
 import cc.unitmesh.devti.intentions.action.task.TestCodeGenRequest
 import cc.unitmesh.devti.intentions.action.task.TestCodeGenTask
+import cc.unitmesh.devti.provider.WriteTestService
 import com.intellij.openapi.editor.Editor
 import com.intellij.openapi.progress.ProgressManager
 import com.intellij.openapi.progress.impl.BackgroundableProcessIndicator
@@ -16,6 +17,12 @@ class AutoTestThisIntention : AbstractChatIntention() {
     override fun getText(): String = AutoDevBundle.message("intentions.chat.code.test.name")
     override fun getFamilyName(): String = AutoDevBundle.message("intentions.chat.code.test.family.name")
 
+    override fun isAvailable(project: Project, editor: Editor?, file: PsiFile?): Boolean {
+        val psiElement = file?.originalElement ?: return false
+        val service = WriteTestService.context(psiElement)
+        return service != null
+    }
+
     override fun invoke(project: Project, editor: Editor?, file: PsiFile?) {
         if (editor == null || file == null) return
 

src/main/kotlin/cc/unitmesh/devti/provider/WriteTestService.kt

@@ -111,7 +111,6 @@ abstract class WriteTestService : LazyExtensionInstance<WriteTestService>() {
             }
 
             if (writeTestService == null) {
-                log.warn("Could not find language support for: ${psiElement.language}, make you have the plugin installed.")
                 return null
             }