fix: fix test issues

Author: phodal

java/src/main/kotlin/cc/unitmesh/idea/provider/JavaTestContextProvider.kt

@@ -10,6 +10,7 @@ import com.intellij.openapi.project.Project
 import com.intellij.psi.PsiClass
 import com.intellij.psi.PsiElement
 import com.intellij.psi.PsiMethod
+import org.jetbrains.kotlin.idea.util.application.runReadAction
 
 open class JavaTestContextProvider : ChatContextProvider {
     override fun isApplicable(project: Project, creationContext: ChatCreationContext): Boolean {
@@ -26,7 +27,7 @@ open class JavaTestContextProvider : ChatContextProvider {
         val isController = fileName?.let { MvcUtil.isController(it, langFileSuffix()) } ?: false
         val isService = fileName?.let { MvcUtil.isService(it, langFileSuffix()) } ?: false
 
-        val isSpringRelated = creationContext.element?.let { isSpringRelated(it) } ?: false
+        val isSpringRelated = runReadAction { creationContext.element?.let { isSpringRelated(it) } ?: false }
 
         var baseTestPrompt = """
             |You MUST use should_xx_xx style for test method name.

java/src/main/kotlin/cc/unitmesh/idea/service/JavaRelatedContext.kt

@@ -63,8 +63,12 @@ object JavaRelatedContext {
     }
 
     private fun findSuperClasses(psiClass: PsiClass): List<PsiClass> {
-        if (!isProjectContent(psiClass)) return emptyList()
-        return generateSequence(psiClass) { it.superClass }.toList()
+        val superClass = psiClass.superClass ?: return emptyList()
+        if (isProjectContent(superClass)) {
+            return listOf(psiClass.superClass!!)
+        }
+
+        return emptyList()
     }
 
     private fun canBeRemoved(member: PsiMember): Boolean {
@@ -90,7 +94,7 @@ object JavaRelatedContext {
         return projectContentClasses.toList()
     }
 
-    private fun isProjectContent(element: PsiElement): Boolean {
+    fun isProjectContent(element: PsiElement): Boolean {
         val virtualFile = PsiUtil.getVirtualFile(element)
         return virtualFile == null || ProjectFileIndex.getInstance(element.project).isInContent(virtualFile)
     }

java/src/main/kotlin/cc/unitmesh/idea/service/JavaTypeUtil.kt

@@ -47,6 +47,10 @@ object JavaTypeUtil {
                 it.type is PsiClassReferenceType
             }.map {
                 val resolve = (it.type as PsiClassReferenceType).resolve() ?: return@map null
+                if (!JavaRelatedContext.isProjectContent(resolve)) {
+                    return@map null
+                }
+
                 resolvedClasses[it.name] = JavaRelatedContext.cleanUp(resolve)
             }